Update app.py

app.py

@@ -48,9 +48,9 @@ def augment_multiple_query(query):
         messages=[
             {
                 "role": "system",
-                "content": f"""You are a helpful expert in {topic}. Your users are asking questions about {topic}. 
-                Suggest up to five additional related questions to help them find the information they need for the provided question. 
-                Suggest only short questions without compound sentences. Suggest a variety of questions that cover different aspects of the topic. 
+                "content": f"""You are a helpful expert in {topic}. Your users are asking questions about {topic}.
+                Suggest up to five additional related questions to help them find the information they need for the provided question.
+                Suggest only short questions without compound sentences. Suggest a variety of questions that cover different aspects of the topic.
                 Make sure they are complete questions, and that they are related to the original question."""
             },
             {
@@ -62,37 +62,10 @@ def augment_multiple_query(query):
     )
     return content.choices[0].message.content.split("\n")
 
-# Custom function to handle fuzzy keyword searches
-def load_wikipedia_page(topic):
-    try:
-        # Attempt to load the page directly
-        page = wikipedia.page(topic)
-        return page.content
-    
-    except wikipedia.exceptions.DisambiguationError as e:
-        # Handle disambiguation page case
-        st.write(f"The keyword '{topic}' returned multiple possible pages.")
-        st.write("Please refine your search or select one of the following options:")
-        
-        # List disambiguation options
-        options = e.options
-        selected_option = st.selectbox("Select the closest match:", options)
-        
-        # Load the selected option
-        page = wikipedia.page(selected_option)
-        return page.content
-    
-    except wikipedia.exceptions.PageError:
-        # Handle the case where no page is found
-        st.write(f"No page found for '{topic}'. Please try a different keyword.")
-        return None
-
 # Function to handle RAG-based question answering
 def rag_advanced(user_query):
     # Document Loading
-    docs_content = load_wikipedia_page(topic)
-    if docs_content:
-        docs = [docs_content]  # Wrap the content in a list for consistency
+    docs = WikipediaLoader(query=topic).load()
 
     # Text Splitting
     character_splitter = RecursiveCharacterTextSplitter(separators=["\n\n", "\n", ". ", " ", ""], chunk_size=1000, chunk_overlap=0)
@@ -134,8 +107,8 @@ def rag_advanced(user_query):
         messages=[
             {
                 "role": "system",
-                "content": f"""You are a helpful expert in {topic}. Your users are asking questions about {topic}. 
-                You will be shown the user's questions, and the relevant information from the documents related to {topic}. 
+                "content": f"""You are a helpful expert in {topic}. Your users are asking questions about {topic}.
+                You will be shown the user's questions, and the relevant information from the documents related to {topic}.
                 Answer the user's question using only this information."""
             },
             {