import streamlit as st
import wikipediaapi
import random

# Initialize Wikipedia API
wiki_wiki = wikipediaapi.Wikipedia(
    user_agent='HistoricalFiguresApp/1.0 (your_email@example.com)',
    language='en'
)

def get_wikipedia_summary(name):
    """Retrieve and store the Wikipedia page summary for a historical figure."""
    page = wiki_wiki.page(name)
    return page.summary if page.exists() else "Sorry, I couldn't find information on that historical figure."

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."
    
    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)
    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)
    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_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}")