Update app.py

app.py

@@ -13,12 +13,30 @@ def get_wikipedia_summary(name):
     page = wiki_wiki.page(name)
     return page.summary if page.exists() else "Sorry, I couldn't find information on that historical figure."
 
-def generate_response(person_name, user_input, context):
-    """Generate a dynamic response based on the Wikipedia summary and user input."""
-    if any(word in user_input.lower() for word in ["who", "what", "when", "where", "why", "how"]):
-        return f"{person_name}: {context}"
+def generate_persona_response(person_name, user_input, context):
+    """Generate a persona-based response based on the Wikipedia summary."""
+    persona_responses = {
+        "Genghis Khan": [
+            "Fishing? A peaceful pursuit, but my mind was always on conquest.",
+            "The rivers of the steppe were not for leisure but for strategy.",
+            "A warrior has little time for such pastimes. Tell me, do you seek power?"
+        ],
+        "Thomas Jefferson": [
+            "Ah, the beauty of nature! I often found solace in Monticello’s gardens.",
+            "A fine question. One must balance duty with pleasure.",
+            "Indeed, the world is full of wonders. How do you spend your leisure time?"
+        ],
+        "Marie Curie": [
+            "Fishing? No, my passion lay in the pursuit of knowledge.",
+            "The only elements I fished for were those unseen by the human eye.",
+            "Curiosity, much like science, leads us to unexpected places. What interests you?"
+        ]
+    }
+    
+    if person_name in persona_responses:
+        return random.choice(persona_responses[person_name])
     else:
-        return f"{person_name}: That is an interesting thought. Could you elaborate?"
+        return f"As {person_name}, I must say—that is an interesting question. History has recorded much about my life. What else do you wish to know?"
 
 st.title("Chat with a Historical Figure!")
 
@@ -30,5 +48,5 @@ if person_name:
     
     user_input = st.text_input("Ask a question:")
     if user_input:
-        response = generate_response(person_name, user_input, summary)
+        response = generate_persona_response(person_name, user_input, summary)
         st.write(f"**{person_name}:** {response}")