Update app.py

app.py

@@ -13,26 +13,24 @@ 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_persona_response(person_name, user_input, context):
-    """Generate a conversational response based on user input and historical context."""
+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."
     
-    hobby_keywords = ["hobby", "free time", "fun", "leisure", "relax"]
-    politics_keywords = ["government", "law", "president", "policy", "rights"]
-    war_keywords = ["battle", "war", "fight", "military", "enemy"]
+    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 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?"
+    if relevant_sentences:
+        response = random.choice(relevant_sentences)
     else:
-        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?"
+        response = random.choice([
+            "That is an intriguing question. What makes you ask?",
+            "A curious thought. Tell me more of your own views.",
+            "History is shaped by questions such as these. What do you believe?"
+        ])
+    
+    return f"{intro} {response}"
 
 st.title("Chat with a Historical Figure!")
 
@@ -46,12 +44,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. Ask what you wish."))
+    st.session_state.chat_history.append((person_name, "I am here. Let us talk."))
 
 if st.session_state.person_name:
     user_input = st.text_input("Ask a question:")
     if user_input:
-        response = generate_persona_response(st.session_state.person_name, user_input, st.session_state.summary)
+        response = generate_dynamic_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))