Update app.py

app.py

@@ -9,20 +9,31 @@ wiki_wiki = wikipediaapi.Wikipedia(
 )
 
 def get_wikipedia_summary(name):
-    """Retrieve and store the Wikipedia page summary for a historical figure."""
+    """Retrieve and clean the Wikipedia summary for a historical figure, avoiding disambiguation pages."""
     page = wiki_wiki.page(name)
-    return page.summary if page.exists() else "Sorry, I couldn't find information on that historical figure."
+    if not page.exists():
+        return "Sorry, I couldn't find information on that historical figure."
+    
+    # Remove disambiguation-related content
+    if 'may refer to' in page.summary.lower():
+        return "This name refers to multiple topics. Can you provide more details?"
+    
+    return page.summary
 
 def generate_dynamic_response(person_name, user_input, context):
     """Dynamically generate a response based on the historical figure's persona and context."""
     user_input = user_input.lower()
-    intro = f"I am {person_name}. Let us discuss."
+    intro = f"Hello! I am {person_name}. Do you have any questions for me?"
     
     sentences = [s.strip() for s in context.split('.') if len(s) > 10]
     relevant_sentences = [s for s in sentences if any(word in s.lower() for word in user_input.split())]
     
     if relevant_sentences:
         response = random.choice(relevant_sentences)
+    elif "fishing" in user_input:
+        response = "Ah, fishing! A pastime I did not indulge in, but I was always fascinated by the natural world and its forces."
+    elif "hobbies" in user_input or "free time" in user_input:
+        response = "In my free time, I was consumed by my experiments, always seeking the next great discovery."
     else:
         response = random.choice([
             "That is an intriguing question. What makes you ask?",
@@ -44,9 +55,12 @@ person_name = st.text_input("Enter the name of a historical figure:")
 if person_name and not st.session_state.person_name:
     st.session_state.person_name = person_name
     st.session_state.summary = get_wikipedia_summary(person_name)
-    st.session_state.chat_history.append((person_name, "I am here. Let us talk."))
+    if "multiple topics" in st.session_state.summary:
+        st.write(st.session_state.summary)
+    else:
+        st.session_state.chat_history.append((person_name, "I am here. Let us talk."))
 
-if st.session_state.person_name:
+if st.session_state.person_name and "multiple topics" not in st.session_state.summary:
     user_input = st.text_input("Ask a question:")
     if user_input:
         response = generate_dynamic_response(st.session_state.person_name, user_input, st.session_state.summary)