Update app.py

app.py

@@ -15,38 +15,39 @@ def get_wikipedia_summary(name):
 
 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])
+    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?"
     else:
-        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?"
+        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?"
+        ])
 
 st.title("Chat with a Historical Figure!")
 
 # User inputs a historical figure
+if "chat_history" not in st.session_state:
+    st.session_state.chat_history = []
+    st.session_state.summary = ""
+    st.session_state.person_name = ""
+
 person_name = st.text_input("Enter the name of a historical figure:")
-if person_name:
-    summary = get_wikipedia_summary(person_name)
-    st.write(f"**Historical Context:** {summary}")
-    
+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. Ask what you wish."))
+
+if st.session_state.person_name:
     user_input = st.text_input("Ask a question:")
     if user_input:
-        response = generate_persona_response(person_name, user_input, summary)
-        st.write(f"**{person_name}:** {response}")
+        response = generate_persona_response(st.session_state.person_name, user_input, st.session_state.summary)
+        st.session_state.chat_history.append(("You", user_input))
+        st.session_state.chat_history.append((st.session_state.person_name, response))
+
+    for speaker, message in st.session_state.chat_history:
+        st.write(f"**{speaker}:** {message}")