init

app.py

@@ -10,15 +10,50 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
 # --- Basic Agent Definition ---
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
+from smolagents import CodeAgent, OpenAIModel
+from tool import web_search_tool, web_fetch_tool
+
+#model used for CodeAgent, need to be multi-modal
+llm= OpenAIModel(
+    model="gpt-4o",  # or gpt-4.1, gpt-4-turbo
+    api_key="YOUR_API_KEY",
+)
+
 class BasicAgent:
     def __init__(self):
+        
         print("BasicAgent initialized.")
+        self.smart_agent = CodeAgent(
+            tools=[web_search_tool, web_fetch_tool],
+            model=llm,
+            max_steps = 5,
+        )
+
     def __call__(self, question: str) -> str:
         print(f"Agent received question (first 50 chars): {question[:50]}...")
-        fixed_answer = "This is a default answer."
+        
+        help_prompt = (
+            "You are a general AI assistant. "
+            "Answer the following question and respond ONLY using this template:\n"
+            "FINAL ANSWER: [ANSWER]\n\n"
+            "Rules:\n"
+            "- The answer should be a number, a few words, or a comma-separated list.\n"
+            "- Do NOT include explanations or reasoning.\n"
+            "- Do NOT use units unless explicitly requested.\n"
+            "- Do NOT use articles (a, an, the).\n"
+            "- Write digits in plain text unless specified otherwise.\n\n"
+            "Question:\n"
+        )
+        fixed_answer = self.smart_agent.run(help_prompt + question)
+
+        
         print(f"Agent returning fixed answer: {fixed_answer}")
+        
         return fixed_answer
 
+
+
+
 def run_and_submit_all( profile: gr.OAuthProfile | None):
     """
     Fetches all questions, runs the BasicAgent on them, submits all answers,

tool.py

@@ -0,0 +1,123 @@
+from smolagents import DuckDuckGoSearchTool
+
+# Initialize the DuckDuckGo search tool
+web_search_tool = DuckDuckGoSearchTool()
+
+from smolagents import Tool
+import json
+import os
+
+# PDF support
+import pdfplumber
+
+# CSV support
+import csv
+
+
+class FileReader(Tool):
+    name = "file_reader"
+    description = (
+        "Read local files and return extracted text. "
+        "Supports PDF, JSON, TXT, and CSV."
+    )
+
+    inputs = {
+        "path": {
+            "type": "string",
+            "description": "Path to the file on disk"
+        }
+    }
+
+    output_type = "string"
+
+    def forward(self, path: str) -> str:
+        if not os.path.exists(path):
+            return f"Error: file not found at {path}"
+
+        ext = os.path.splitext(path)[1].lower()
+
+        try:
+            if ext == ".pdf":
+                return self._read_pdf(path)
+
+            elif ext == ".json":
+                return self._read_json(path)
+
+            elif ext == ".txt":
+                return self._read_txt(path)
+
+            elif ext == ".csv":
+                return self._read_csv(path)
+
+            else:
+                return f"Unsupported file type: {ext}"
+
+        except Exception as e:
+            return f"Error reading file: {str(e)}"
+
+    def _read_pdf(self, path: str) -> str:
+        text = []
+        with pdfplumber.open(path) as pdf:
+            for page in pdf.pages:
+                page_text = page.extract_text()
+                if page_text:
+                    text.append(page_text)
+        return "\n".join(text)
+
+    def _read_json(self, path: str) -> str:
+        with open(path, "r", encoding="utf-8") as f:
+            data = json.load(f)
+        return json.dumps(data, indent=2)
+
+    def _read_txt(self, path: str) -> str:
+        with open(path, "r", encoding="utf-8") as f:
+            return f.read()
+
+    def _read_csv(self, path: str) -> str:
+        rows = []
+        with open(path, newline="", encoding="utf-8") as f:
+            reader = csv.reader(f)
+            for row in reader:
+                rows.append(", ".join(row))
+        return "\n".join(rows)
+
+
+file_reader_tool = FileReader()
+
+
+from smolagents import Tool
+import httpx
+from bs4 import BeautifulSoup
+
+class WebFetch(Tool):
+    name = "web_fetch"
+    description = "Fetch and read webpage content from a URL."
+    inputs = {
+        "url": {
+            "type": "string",
+            "description": "URL of the webpage to read"
+        }
+    }
+    output_type = "string"
+
+    def forward(self, url: str) -> str:
+        try:
+            with httpx.Client(follow_redirects=True, timeout=20) as client:
+                r = client.get(url)
+                r.raise_for_status()
+
+            soup = BeautifulSoup(r.text, "html.parser")
+
+            # Remove scripts, styles, nav, footer
+            for tag in soup(["script", "style", "noscript", "nav", "footer", "header", "aside"]):
+                tag.decompose()
+
+            text = soup.get_text(separator="\n")
+            lines = [line.strip() for line in text.splitlines() if line.strip()]
+
+            return "\n".join(lines[:5000])  # cap length for LLM
+
+        except Exception as e:
+            return f"Error fetching page: {str(e)}"
+
+web_fetch_tool = WebFetch()