Update app.py

app.py

@@ -9,23 +9,30 @@ wiki_wiki = wikipediaapi.Wikipedia(
 )
 
 def get_wikipedia_summary(name):
-    """Retrieve and summarize the Wikipedia page for a given historical figure."""
+    """Retrieve and store the Wikipedia page summary for a historical figure."""
     page = wiki_wiki.page(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 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())]
+    """Generate a conversational response based on user input and historical context."""
+    user_input = user_input.lower()
     
-    if relevant_info:
-        return random.choice(relevant_info) + '.'
+    hobby_keywords = ["hobby", "free time", "fun", "leisure", "relax"]
+    politics_keywords = ["government", "law", "president", "policy", "rights"]
+    war_keywords = ["battle", "war", "fight", "military", "enemy"]
+    
+    if any(word in user_input for word in hobby_keywords):
+        return f"Ah, leisure! In my time, I enjoyed reading, studying science, and tending to my gardens. What do you do for leisure?"
+    elif any(word in user_input for word in politics_keywords):
+        return f"Politics is a fascinating matter. I strongly believed in individual liberty and self-governance. What are your thoughts on democracy?"
+    elif any(word in user_input for word in war_keywords):
+        return f"Conflict is often necessary to secure freedom. I played my role in the Revolution to help shape a new nation. What do you know of it?"
     else:
-        return random.choice([
-            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."
-        ])
+        relevant_sentences = [s.strip() for s in context.split('.') if any(word in s.lower() for word in user_input.split())]
+        if relevant_sentences:
+            return random.choice(relevant_sentences) + '.'
+        else:
+            return f"That is an interesting question. Could you elaborate on what you mean?"
 
 st.title("Chat with a Historical Figure!")