File size: 1,673 Bytes
065cda0 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 |
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("---")
|