import streamlit as st
import requests
import os
import time

# =============================
# CONFIG
# =============================

BACKEND_URL = "https://mohith96-researchagentbackend.hf.space/research"
REQUEST_TIMEOUT = 300  # seconds

# Hugging Face token for private backend access
HF_BACKEND_TOKEN = os.getenv("HF_BACKEND_TOKEN")

# =============================
# PAGE SETUP
# =============================

st.set_page_config(
    page_title="Research Agent",
    page_icon="📚",
    layout="wide"
)

st.title("📚 Research Agent")
st.caption("A general purpose research agent that will break down the input query into sub queries and perform iterative search in order to generate an ideal output")

# =============================
# STATE
# =============================

if "result" not in st.session_state:
    st.session_state.result = None

if "title" not in st.session_state:
    st.session_state.title = None

# =============================
# INPUT
# =============================

query = st.text_area(
    label="Research Query",
    placeholder="e.g. Compatibility issues between LangChain, Python 3.12, and Groq LLMs",
    height=120
)

col1, col2 = st.columns([1, 5])

with col1:
    run_clicked = st.button("🚀 Run Research", use_container_width=True)

# =============================
# ACTION
# =============================

if run_clicked:
    if not query.strip():
        st.warning("You need to enter a research query. This is not optional.")
    elif not HF_BACKEND_TOKEN:
        st.error("Backend token missing. Set HF_BACKEND_TOKEN in Space secrets.")
    else:
        with st.spinner("Running research pipeline..."):
            try:
                headers = {
                    "Authorization": f"Bearer {HF_BACKEND_TOKEN}"
                }

                response = requests.post(
                    BACKEND_URL,
                    json={"query": query},
                    headers=headers,
                    timeout=REQUEST_TIMEOUT
                )

                if response.status_code != 200:
                    st.error(
                        f"Backend error ({response.status_code}). "
                        f"Response: {response.text}"
                    )
                else:
                    payload = response.json()
                    st.session_state.title = payload.get("title", "Untitled")
                    st.session_state.result = payload.get("report", "")

            except requests.exceptions.Timeout:
                st.error("Backend timed out. Either it's slow or asleep.")
            except Exception as e:
                st.error(f"Unexpected failure: {str(e)}")

# =============================
# OUTPUT
# =============================

if st.session_state.result:
    st.divider()

    st.subheader(st.session_state.title)
    st.markdown(st.session_state.result)

    st.download_button(
        label="⬇️ Download Markdown",
        data=st.session_state.result,
        file_name=f"{st.session_state.title}.md",
        mime="text/markdown"
    )

# =============================
# FOOTER
# =============================

st.divider()
st.caption("Frontend-only Space. Backend is private and authenticated. As it should be.")