import uuid
import traceback
import streamlit as st
from deep_research import run_deep_research_system

st.set_page_config(page_title="Deep Research AI", layout="centered")

if "question" not in st.session_state:
    st.session_state.question = ""
if "show_result" not in st.session_state:
    st.session_state.show_result = False
if "input_key" not in st.session_state:
    st.session_state.input_key = str(uuid.uuid4())

st.title("Deep Research AI Agentic System")
st.write("Enter a question below to get the latest insights from web research. Click 'Reset' to start over.")

st.session_state.question = st.text_input(
    "Your Question",
    value=st.session_state.question,
    placeholder="e.g., What are the latest advancements in quantum computing?",
    key=st.session_state.input_key
)

if st.button("Get Answer"):
    if st.session_state.question.strip() != "":
        st.session_state.show_result = True
        st.write(f"Research Agent: Searching for '{st.session_state.question}'...")
        try:
            with st.spinner("Gathering research data..."):
                answer = run_deep_research_system(st.session_state.question)
            st.success("Research complete! Answer Drafter Agent: Drafted the final answer.")
            st.markdown("**Final Answer:**")
            st.write(answer)
        except Exception as e:
            st.error(f"An error occurred: {str(e)}\n{traceback.format_exc()}")
    else:
        st.warning("Please enter a question!")

if st.session_state.show_result:
    if st.button("Reset"):
        st.session_state.question = ""
        st.session_state.show_result = False
        st.session_state.input_key = str(uuid.uuid4())
        st.rerun()