import streamlit as st
import os
from dotenv import load_dotenv
from Agents.multi_agent import ResearchAgents
from data_loaders import DataLoader

load_dotenv()

st.title("📚 Virtual Research Assistant")

groq_api_key = os.getenv("GROQ_API_KEY")
if not groq_api_key:
    st.error("GROQ_API_KEY is missing. Please set it in your environment variables.")
    st.stop()

agents = ResearchAgents(groq_api_key)
data_loader = DataLoader()

query = st.text_input("Enter a research topic:")

if st.button("Search"):
    with st.spinner("Fetching research papers..."): 
        arxiv_papers = data_loader.fetch_arxiv_papers(query)
        all_papers = arxiv_papers

        if not all_papers:
            st.error("Failed to fetch papers. Try again!")
        else:
            processed_papers = []

            for paper in all_papers:
                summary = agents.summarize_paper(paper['summary'])  
                adv_dis = agents.analyze_advantages_disadvantages(summary)  

                processed_papers.append({
                    "title": paper["title"],
                    "link": paper["link"],
                    "summary": summary,
                    "advantages_disadvantages": adv_dis,
                })
            st.subheader("Top Research Papers:")
            for i, paper in enumerate(processed_papers, 1):
                st.markdown(f"### {i}. {paper['title']}") 
                st.markdown(f"🔗 [Read Paper]({paper['link']})") 
                st.write(f"**Summary:** {paper['summary']}") 
                st.write(f"{paper['advantages_disadvantages']}")  
                st.markdown("---")