Update app.py

app.py

@@ -1,49 +1,34 @@
 import streamlit as st
 import wikipediaapi
+import random
+
+# Initialize Wikipedia API
+wiki_wiki = wikipediaapi.Wikipedia(
+    user_agent='HistoricalFiguresApp/1.0 ([email protected])',
+    language='en'
+)
 
 def get_wikipedia_summary(name):
-    wiki = wikipediaapi.Wikipedia(
-        user_agent='HistoricalFiguresBot/1.0 (educational-app)',
-        language='en'
-    )
-    page = wiki.page(name)
-    return page.summary if page.exists() else None
+    """Retrieve and summarize the Wikipedia page for a given historical figure."""
+    page = wiki_wiki.page(name)
+    return page.summary if page.exists() else "Sorry, I couldn't find information on that historical figure."
 
-# Added simple character response system
-def get_historical_response(context, user_input):
-    # Basic response templates based on context
-    if "born" in context.lower():
-        return f"Indeed, as someone who experienced those times, I can tell you that {user_input}"
-    elif "died" in context.lower():
-        return f"In my time, we would have approached this by {user_input}"
+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}"
     else:
-        return f"Based on my experiences, I would say that {user_input}"
-
-st.title("Historical Figures Chat")
-st.write("Chat with historical figures using Wikipedia information!")
-
-person_name = st.text_input("Enter historical figure name:")
-
-if "messages" not in st.session_state:
-    st.session_state.messages = []
+        return f"{person_name}: That is an interesting thought. Could you elaborate?"
 
-for msg in st.session_state.messages:
-    st.chat_message(msg["role"]).write(msg["content"])
+st.title("Chat with a Historical Figure!")
 
-if user_input := st.chat_input("Message..."):
-    if not person_name:
-        st.error("Please enter a historical figure's name")
-        st.stop()
-        
-    st.session_state.messages.append({"role": "user", "content": user_input})
-    st.chat_message("user").write(user_input)
+# User inputs a historical figure
+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}")
     
-    with st.spinner("Researching..."):
-        context = get_wikipedia_summary(person_name)
-        if context:
-            st.write("Historical Context:", context)
-            response = get_historical_response(context, user_input)
-            st.session_state.messages.append({"role": "assistant", "content": response})
-            st.chat_message("assistant").write(response)
-        else:
-            st.error(f"Could not find information about {person_name}")
+    user_input = st.text_input("Ask a question:")
+    if user_input:
+        response = generate_response(person_name, user_input, summary)
+        st.write(f"**{person_name}:** {response}")