Update app.py

app.py

@@ -1,9 +1,10 @@
 import os
 import re
+import ast
 import json
-import gradio as gr
 import requests
 import pandas as pd
+import gradio as gr
 
 from groq import Groq
 from duckduckgo_search import DDGS
@@ -11,7 +12,7 @@ from duckduckgo_search import DDGS
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
 # ---------------------------------------------------
-# WEB SEARCH TOOL
+# SEARCH TOOL
 # ---------------------------------------------------
 class WebSearchTool:
 
@@ -30,11 +31,13 @@ class WebSearchTool:
             snippets = []
 
             for r in results:
-                body = r.get("body", "")
+
                 title = r.get("title", "")
+                body = r.get("body", "")
+
                 snippets.append(f"{title}: {body}")
 
-            return "\n".join(snippets[:5])
+            return "\n".join(snippets[:max_results])
 
         except Exception as e:
 
@@ -56,7 +59,7 @@ class BasicAgent:
 
         self.search_tool = WebSearchTool()
 
-        print("Agent initialized.")
+        print("Lightweight GAIA agent initialized.")
 
     # ---------------------------------------------------
     # CLEAN ANSWER
@@ -68,9 +71,16 @@ class BasicAgent:
 
         text = str(text)
 
-        text = text.replace("```", "")
-        text = text.replace("FINAL ANSWER:", "")
-        text = text.replace("Answer:", "")
+        bad_phrases = [
+            "FINAL ANSWER:",
+            "Answer:",
+            "answer:",
+            "```",
+            "`"
+        ]
+
+        for b in bad_phrases:
+            text = text.replace(b, "")
 
         text = re.sub(r"\s+", " ", text)
 
@@ -81,39 +91,98 @@ class BasicAgent:
 
         return text[:300]
 
+    # ---------------------------------------------------
+    # REVERSE STRING TASK
+    # ---------------------------------------------------
+    def handle_reverse_text(self, question):
+
+        reversed_text = question[::-1]
+
+        return reversed_text[:300]
+
     # ---------------------------------------------------
     # DETECT SEARCH NEED
     # ---------------------------------------------------
     def needs_search(self, question):
 
+        q = question.lower()
+
         keywords = [
             "who",
             "when",
             "where",
-            "latest",
-            "current",
-            "search",
-            "find",
-            "look up",
-            "verify",
-            "paper",
-            "award",
+            "which",
+            "youtube",
+            "wikipedia",
+            "movie",
             "actor",
+            "award",
+            "paper",
             "country",
             "city",
             "population",
-            "website",
-            "nasa",
-            "movie",
+            "published",
+            "album",
             "song",
-            "audio",
-            "youtube",
-            "wikipedia"
+            "species",
+            "nasa",
+            "video"
         ]
 
+        return any(k in q for k in keywords)
+
+    # ---------------------------------------------------
+    # BUILD CONTEXT
+    # ---------------------------------------------------
+    def build_context(self, question):
+
         q = question.lower()
 
-        return any(k in q for k in keywords)
+        context = []
+
+        # ---------------------------------------------------
+        # SEARCH
+        # ---------------------------------------------------
+        if self.needs_search(question):
+
+            print("\nRunning web search...")
+
+            web = self.search_tool.search(question)
+
+            context.append(
+                f"WEB SEARCH RESULTS:\n{web}"
+            )
+
+        # ---------------------------------------------------
+        # CHESS
+        # ---------------------------------------------------
+        if "chess" in q:
+
+            context.append(
+                "This is a chess puzzle. "
+                "Return the best move only."
+            )
+
+        # ---------------------------------------------------
+        # CODE
+        # ---------------------------------------------------
+        if "python code" in q:
+
+            context.append(
+                "Infer likely numeric output."
+            )
+
+        # ---------------------------------------------------
+        # YOUTUBE
+        # ---------------------------------------------------
+        if "youtube.com" in q or "youtu.be" in q:
+
+            context.append(
+                "Use search results and inference "
+                "to answer the YouTube question."
+            )
+
+        return "\n\n".join(context)
 
     # ---------------------------------------------------
     # MAIN CALL
@@ -122,46 +191,49 @@ class BasicAgent:
 
         try:
 
-            web_context = ""
-
             # ---------------------------------------------------
-            # WEB SEARCH
+            # REVERSE STRING
             # ---------------------------------------------------
-            if self.needs_search(question):
+            if question.strip().startswith("."):
+
+                print("Reverse text detected.")
 
-                print(f"\nSearching web for: {question}")
+                return self.handle_reverse_text(question)
 
-                web_context = self.search_tool.search(question)
+            # ---------------------------------------------------
+            # CONTEXT
+            # ---------------------------------------------------
+            context = self.build_context(question)
 
             # ---------------------------------------------------
             # PROMPT
             # ---------------------------------------------------
             system_prompt = """
-You are an advanced GAIA benchmark solving assistant.
+You are a lightweight GAIA benchmark solving assistant.
 
-Rules:
+STRICT RULES:
 - Return ONLY the final answer.
-- No reasoning.
-- No markdown.
 - No explanations.
-- Be concise.
-- If the answer is a list, format correctly.
-- If unsure, still provide your best answer.
+- No markdown.
+- No bullet points.
+- No reasoning traces.
+- Keep answers concise.
+- Use context carefully.
 """
 
             user_prompt = f"""
 QUESTION:
 {question}
 
-WEB SEARCH RESULTS:
-{web_context}
+CONTEXT:
+{context}
 """
 
             # ---------------------------------------------------
-            # GROQ CALL
+            # GROQ
             # ---------------------------------------------------
             completion = self.client.chat.completions.create(
-                model="llama-3.3-70b-versatile",
+                model="llama-3.1-8b-instant",
                 messages=[
                     {
                         "role": "system",
@@ -173,15 +245,19 @@ WEB SEARCH RESULTS:
                     }
                 ],
                 temperature=0,
-                max_tokens=128
+                max_tokens=96
             )
 
             answer = completion.choices[0].message.content
 
             cleaned = self.clean_answer(answer)
 
-            print(f"\nQUESTION:\n{question}")
-            print(f"\nANSWER:\n{cleaned}")
+            print("\n======================")
+            print("QUESTION:")
+            print(question)
+            print("\nANSWER:")
+            print(cleaned)
+            print("======================\n")
 
             return cleaned
 
@@ -238,16 +314,14 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
     results_log = []
 
     # ---------------------------------------------------
-    # RUN QUESTIONS
+    # RUN TASKS
     # ---------------------------------------------------
     for idx, item in enumerate(questions_data):
 
         task_id = item.get("task_id")
         question = item.get("question")
 
-        print(f"\n========================")
-        print(f"TASK {idx+1}")
-        print(f"========================")
+        print(f"\nTASK {idx+1}")
 
         try:
 
@@ -310,12 +384,14 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
 # ---------------------------------------------------
 with gr.Blocks() as demo:
 
-    gr.Markdown("# HF Agents Course Assignment")
+    gr.Markdown("# Lightweight GAIA Agent")
 
     gr.Markdown("""
-Tools enabled:
-- Groq Llama 3.3 70B
+Enabled tools:
+- Groq Llama 3.1 8B Instant
 - DuckDuckGo Search
+- Reverse text handling
+- Lightweight reasoning
 """)
 
     gr.LoginButton()
@@ -340,6 +416,6 @@ Tools enabled:
 # ---------------------------------------------------
 if __name__ == "__main__":
 
-    print("Launching app...")
+    print("Launching lightweight GAIA agent...")
 
     demo.launch()