Update app.py

for final project

app.py

@@ -4,12 +4,22 @@ import requests
 import inspect
 import pandas as pd
 
+
+from duckduckgo_search import ddg
+from transformers import pipeline
+from newspaper import Article
+import hashlib, datetime
+
+
 # (Keep Constants as is)
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
 # --- Basic Agent Definition ---
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
+# Summarization pipeline (load once)
+summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
+
 class BasicAgent:
     def __init__(self):
         print("BasicAgent initialized.")
@@ -19,6 +29,59 @@ class BasicAgent:
         print(f"Agent returning fixed answer: {fixed_answer}")
         return fixed_answer
 
+
+class SmartResearchAgent:
+    def __init__(self):
+        print("SmartResearchAgent initialized.")
+
+    def search_web(self, query: str) -> str:
+        results = ddg(query, max_results=3)
+        if not results:
+            return "No results found."
+        return "\n".join(f"{i+1}. {r['title']}: {r['href']}" for i, r in enumerate(results))
+
+    def summarize(self, input_text: str) -> str:
+        if input_text.startswith("http"):
+            try:
+                article = Article(input_text)
+                article.download()
+                article.parse()
+                input_text = article.text
+            except Exception as e:
+                return f"Error processing article: {e}"
+        summary = summarizer(input_text, max_length=130, min_length=30, do_sample=False)
+        return summary[0]['summary_text']
+
+    def generate_citation(self, source: str) -> str:
+        citation_id = hashlib.md5(source.encode()).hexdigest()[:6]
+        year = datetime.datetime.now().year
+        return f"@article{{cite{citation_id}, title={{\"Research Article\"}}, author={{Unknown}}, journal={{Online Source}}, year={{ {year} }}, url={{ {source} }} }}"
+
+    def __call__(self, question: str) -> str:
+        print(f"Received question: {question}")
+
+        if "summarize" in question.lower():
+            return self.summarize(question.replace("summarize", "", 1).strip())
+
+        elif "cite" in question.lower() or "citation" in question.lower():
+            return self.generate_citation(question.split()[-1])
+
+        elif "search" in question.lower():
+            return self.search_web(question.replace("search", "", 1).strip())
+
+        else:
+            # Default: web search + summarization
+            web_result = self.search_web(question)
+            if "http" in web_result:
+                first_url = next((line.split(": ", 1)[-1] for line in web_result.splitlines() if line.startswith("1.")), None)
+                if first_url:
+                    summary = self.summarize(first_url)
+                    citation = self.generate_citation(first_url)
+                    return f"Summary:\n{summary}\n\nCitation:\n{citation}"
+            return web_result
+
+            
+
 def run_and_submit_all( profile: gr.OAuthProfile | None):
     """
     Fetches all questions, runs the BasicAgent on them, submits all answers,
@@ -40,7 +103,9 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
 
     # 1. Instantiate Agent ( modify this part to create your agent)
     try:
-        agent = BasicAgent()
+        #agent = BasicAgent()
+        agent = SmartResearchAgent()
+        
     except Exception as e:
         print(f"Error instantiating agent: {e}")
         return f"Error initializing agent: {e}", None