Update app.py

app.py

@@ -2,6 +2,7 @@ import os
 import streamlit as st
 import requests
 import feedparser
+import datetime
 from dotenv import load_dotenv
 from duckduckgo_search import DDGS
 
@@ -67,6 +68,18 @@ def get_image_urls(query, max_images=3):
     with DDGS() as ddgs:
         return [img["image"] for img in ddgs.images(query, max_results=max_images)]
 
+def generate_apa_citation(title, url, source=""):
+    current_year = datetime.datetime.now().year
+    if source == "arxiv":
+        return f"{title}. ({current_year}). *arXiv*. {url}"
+    elif source == "semantic":
+        return f"{title}. ({current_year}). *Semantic Scholar*. {url}"
+    elif source == "web":
+        return f"{title}. ({current_year}). *Web Source*. {url}"
+    else:
+        return f"{title}. ({current_year}). {url}"
+
+
 # --- Research Agent ---
 def autonomous_research_agent(topic):
     arxiv = get_arxiv_papers(topic)
@@ -74,17 +87,20 @@ def autonomous_research_agent(topic):
     web = search_duckduckgo(topic)
     images = get_image_urls(topic)
 
-    arxiv_md = ""
+    arxiv_md, arxiv_citations = "", []
     for p in arxiv:
         arxiv_md += f"- [{p['title']}]({p['url']})\n> {p['summary'][:300]}...\n\n"
+        arxiv_citations.append(generate_apa_citation(p["title"], p["url"], source="arxiv"))
 
-    scholar_md = ""
+    scholar_md, scholar_citations = "", []
     for p in scholar:
         scholar_md += f"- [{p['title']}]({p['url']})\n> {p['summary'][:300]}...\n\n"
+        scholar_citations.append(generate_apa_citation(p["title"], p["url"], source="semantic"))
 
-    web_md = ""
+    web_md, web_citations = "", []
     for w in web:
         web_md += f"- [{w['title']}]({w['url']})\n> {w['snippet']}\n\n"
+        web_citations.append(generate_apa_citation(w["title"], w["url"], source="web"))
 
     prompt = f"""
 # Research Topic: {topic}
@@ -115,6 +131,12 @@ Now synthesize this information into:
     if web_md:
         response += "**Web:**\n" + web_md
 
+    # APA Citations Section
+    all_citations = arxiv_citations + scholar_citations + web_citations
+    response += "\n---\n### 📚 APA Citations\n"
+    for cite in all_citations:
+        response += f"- {cite}\n"
+
     return response, images
 
 # --- Streamlit UI ---