Update agent.py

agent.py

@@ -4,7 +4,11 @@ import requests
 import tempfile
 import pandas as pd
 from openai import OpenAI
-from duckduckgo_search import DDGS
+
+try:
+    from duckduckgo_search import DDGS
+except ImportError:
+    DDGS = None  # In case of install problems
 
 PROMPT = (
     "You are a general AI assistant. I will ask you a question. "
@@ -21,6 +25,8 @@ class BasicAgent:
         print("BasicAgent initialized.")
 
     def web_search(self, query: str, max_results: int = 5) -> str:
+        if not DDGS:
+            return ""
         try:
             with DDGS() as ddgs:
                 results = list(ddgs.text(query, max_results=max_results))
@@ -54,6 +60,40 @@ class BasicAgent:
         except Exception as e:
             return ""
 
+    def transcribe_audio(self, file_url: str) -> str:
+        import openai
+        openai.api_key = os.getenv("OPENAI_API_KEY")
+        try:
+            r = requests.get(file_url, timeout=20)
+            r.raise_for_status()
+            # Guess extension from url or response
+            ext = ".mp3"
+            if file_url.endswith(".wav"):
+                ext = ".wav"
+            with tempfile.NamedTemporaryFile(suffix=ext, delete=False) as f:
+                f.write(r.content)
+                f.flush()
+                audio_path = f.name
+            transcript = openai.Audio.transcribe("whisper-1", open(audio_path, "rb"))
+            return transcript.get("text", "")
+        except Exception as e:
+            return ""
+
+    def execute_python(self, file_url: str) -> str:
+        try:
+            r = requests.get(file_url, timeout=20)
+            r.raise_for_status()
+            code = r.content.decode("utf-8")
+            import io, contextlib
+            buf = io.StringIO()
+            with contextlib.redirect_stdout(buf):
+                exec(code, {})
+            output = buf.getvalue().strip().split('\n')[-1]
+            numbers = re.findall(r'[-+]?\d*\.\d+|\d+', output)
+            return numbers[-1] if numbers else output
+        except Exception as e:
+            return ""
+
     def fetch_file_url(self, task_id):
         DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
         try:
@@ -68,16 +108,31 @@ class BasicAgent:
     def __call__(self, question: str, task_id: str = None) -> str:
         file_url = self.fetch_file_url(task_id) if task_id else None
         file_result = None
-        # --- Always try Excel tool if file exists ---
+        ext = file_url.split('.')[-1].lower() if file_url else ""
+        # --- Try all known file tools by extension ---
         if file_url:
-            try:
+            # Excel (any file: always try)
+            if ext in ["xlsx", "xls"] or "excel" in question.lower() or "spreadsheet" in question.lower():
+                file_result = self.excel_tool(file_url)
+                if file_result and re.match(r'^\d+(\.\d+)?$', file_result):
+                    return file_result
+            # Audio
+            elif ext in ["mp3", "wav"] or "audio" in question.lower() or "transcribe" in question.lower():
+                file_result = self.transcribe_audio(file_url)
+                if file_result and file_result.strip():
+                    return file_result
+            # Python code
+            elif ext == "py":
+                file_result = self.execute_python(file_url)
+                if file_result and file_result.strip():
+                    return file_result
+            # Fallback: try Excel anyway
+            if not file_result:
                 file_result = self.excel_tool(file_url)
-                # Only return if it actually found a number
                 if file_result and re.match(r'^\d+(\.\d+)?$', file_result):
                     return file_result
-            except Exception:
-                pass  # fallback below
 
+        # --- Web search and LLM as before ---
         search_snippet = self.web_search(question)
         prompt = PROMPT + f"\n\nWeb search results:\n{search_snippet}\n\nQuestion: {question}"
         response = self.llm.chat.completions.create(
@@ -104,16 +159,13 @@ class BasicAgent:
         ]
         norm_final = (final_line or "").lower()
         if norm_final in bads or norm_final.startswith("unable") or norm_final.startswith("i'm sorry") or norm_final.startswith("i apologize"):
-            # --- Try to extract a plausible answer from web or file ---
-            # Try: numbers from search
+            # Try to extract a plausible answer from web or file
             numbers = re.findall(r'\b\d{2,}\b', search_snippet)
             if numbers:
                 return numbers[0]
-            # Try: possible names from search
             words = re.findall(r'\b[A-Z][a-z]{2,}\b', search_snippet)
             if words:
                 return words[0]
-            # Try: numbers or words from file_result
             if file_result:
                 file_numbers = re.findall(r'\b\d{2,}\b', str(file_result))
                 if file_numbers: