Spaces:
Sleeping
Sleeping
File size: 3,078 Bytes
b97b30a 4140b01 b97b30a eb2a4a7 4140b01 eb2a4a7 ecee870 09be789 318c58a eb2a4a7 ebf07f4 09be789 318c58a 09be789 eb2a4a7 e8d591c 09be789 b97b30a 4140b01 b97b30a 4140b01 4031494 4140b01 4031494 eb2a4a7 4031494 eb2a4a7 4140b01 09be789 4031494 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 |
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):
"""Retrieve and clean the Wikipedia summary for a historical figure, avoiding disambiguation pages."""
page = wiki_wiki.page(name)
if not page.exists():
return "Sorry, I couldn't find information on that historical figure."
# Remove disambiguation-related content
if 'may refer to' in page.summary.lower():
return "This name refers to multiple topics. Can you provide more details?"
return page.summary
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"Hello! I am {person_name}. Do you have any questions for me?"
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 relevant_sentences:
response = random.choice(relevant_sentences)
elif "fishing" in user_input:
response = "Ah, fishing! A pastime I did not indulge in, but I was always fascinated by the natural world and its forces."
elif "hobbies" in user_input or "free time" in user_input:
response = "In my free time, I was consumed by my experiments, always seeking the next great discovery."
else:
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!")
# 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 and not st.session_state.person_name:
st.session_state.person_name = person_name
st.session_state.summary = get_wikipedia_summary(person_name)
if "multiple topics" in st.session_state.summary:
st.write(st.session_state.summary)
else:
st.session_state.chat_history.append((person_name, "I am here. Let us talk."))
if st.session_state.person_name and "multiple topics" not in st.session_state.summary:
user_input = st.text_input("Ask a question:")
if user_input:
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))
for speaker, message in st.session_state.chat_history:
st.write(f"**{speaker}:** {message}")
|