Spaces:
Sleeping
Sleeping
| 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}") | |