Update app.py

app.py

@@ -14,18 +14,17 @@ def get_wikipedia_summary(name):
     return page.summary if page.exists() else "Sorry, I couldn't find information on that historical figure."
 
 def generate_persona_response(person_name, user_input, context):
-    """Generate a persona-based response based on the Wikipedia summary."""
-    if "battle" in user_input.lower() or "war" in user_input.lower():
-        return f"War is the path to power. I conquered vast lands through strategy and strength. What more do you wish to know?"
-    elif "family" in user_input.lower():
-        return f"Family is the foundation of a strong empire. My sons carried forth my legacy. What about your family?"
-    elif "travel" in user_input.lower():
-        return f"My armies rode thousands of miles across mountains and deserts. Do you seek adventure?"
+    """Generate a response based on the Wikipedia summary dynamically."""
+    sentences = context.split('.')
+    relevant_info = [sentence for sentence in sentences if any(word in sentence.lower() for word in user_input.lower().split())]
+    
+    if relevant_info:
+        return random.choice(relevant_info) + '.'
     else:
         return random.choice([
-            f"A curious question. Tell me, why does this interest you?",
-            f"The steppes were harsh, yet they shaped me. Tell me more of your world.",
-            f"I have conquered many lands, yet the mind is the greatest battlefield. What is your next question?"
+            f"An interesting question. I have much knowledge to share—what else do you wish to know?",
+            f"Tell me, why does this interest you? Perhaps we may discover more together.",
+            f"My life was complex and filled with challenges. Ask again, and I shall elaborate."
         ])
 
 st.title("Chat with a Historical Figure!")