Upload 6 files

agent.py

@@ -30,6 +30,15 @@ from tools import (
     SimpleOCRTool,
 )
 
+ 
+# ---------------------------------------------------------------------------
+# Load the system prompt from system_prompt.txt (located in the same directory)
+# ---------------------------------------------------------------------------
+SYSTEM_PROMPT_PATH = os.path.join(os.path.dirname(__file__), "system_prompt.txt")
+with open(SYSTEM_PROMPT_PATH, "r", encoding="utf-8") as f:
+    SYSTEM_PROMPT = f.read().strip()
+
+
 # ---------------------------------------------------------------------------
 # Model selection helper
 # ---------------------------------------------------------------------------
@@ -45,13 +54,21 @@ def _select_model():
         from smolagents import InferenceClientModel
         hf_model_id = os.getenv("HF_MODEL", "HuggingFaceH4/zephyr-7b-beta")
         hf_token = os.getenv("HF_API_KEY")
-        return InferenceClientModel(model_id=hf_model_id, token=hf_token)
+        return InferenceClientModel(
+            model_id=hf_model_id,
+            token=hf_token,
+            system_prompt=SYSTEM_PROMPT
+        )
 
     if provider == "openai":
         from smolagents import OpenAIServerModel
         openai_model_id = os.getenv("OPENAI_MODEL", "gpt-3.5-turbo")
         openai_token = os.getenv("OPENAI_API_KEY")
-        return OpenAIServerModel(model_id=openai_model_id, api_key=openai_token)
+        return OpenAIServerModel(
+            model_id=openai_model_id,
+            api_key=openai_token,
+            system_prompt=SYSTEM_PROMPT
+        )
 
     raise ValueError(
         f"Unsupported MODEL_PROVIDER: {provider!r}. "
@@ -78,16 +95,16 @@ class GAIAAgent(CodeAgent):
     # Convenience so the object itself can be *called* directly
     def __call__(self, question: str, **kwargs: Any) -> str:
         steps = self.run(question, **kwargs)
-        if isinstance(steps, str):
-            print("DEBUG: steps is a string:", repr(steps))
-            return steps.strip()
-        print("DEBUG: steps is a", type(steps))
+        # If steps is a primitive, just return it
+        if isinstance(steps, (int, float, str)):
+            return str(steps).strip()
         last_step = None
         for step in steps:
             last_step = step
-        print("DEBUG: last_step =", repr(last_step))
+        # Defensive: handle int/float/str directly
+        if isinstance(last_step, (int, float, str)):
+            return str(last_step).strip()
         answer = getattr(last_step, "answer", None)
-        print("DEBUG: answer =", repr(answer))
         if answer is not None:
             return str(answer).strip()
         return str(last_step).strip()

app.py

@@ -6,6 +6,9 @@ import pandas as pd
 # --- Our Agent ---
 from agent import gaia_agent
 
+# Debugging level. If DEBUG=0 then DEBUG will be False. If DEBUG=1 then DEBUG will be True.
+DEBUG = os.getenv("DEBUG", "0") == "1"
+
 # (Keep Constants as is)
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
@@ -76,7 +79,11 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
         try:
             submitted_answer = agent(question_text)
             # --- DEBUG LOGGING ---
-            print(f"[DEBUG] Task {task_id}: Answer type: {type(submitted_answer)}, Value: {repr(submitted_answer)}")
+            if DEBUG:
+                print(f"[DEBUG] Task {task_id}: Answer type: {type(submitted_answer)}, Value: {repr(submitted_answer)}")
+            else:
+                print(f"[{task_id}] {question_text[:50]}... → {submitted_answer[:40]}")
+
             # Force string type here just in case (defensive)
             submitted_answer = str(submitted_answer).strip()
             answers_payload.append({"task_id": task_id, "submitted_answer": submitted_answer})

system_prompt.txt

@@ -0,0 +1,5 @@
+You are a helpful assistant tasked with answering questions using a set of tools. 
+Now, I will ask you a question. Report your thoughts, and finish your answer with the following template: 
+FINAL ANSWER: [YOUR FINAL ANSWER]. 
+YOUR FINAL ANSWER should be a number OR as few words as possible OR a comma separated list of numbers and/or strings. If you are asked for a number, don't use comma to write your number neither use units such as $ or percent sign unless specified otherwise. If you are asked for a string, don't use articles, neither abbreviations (e.g. for cities), and write the digits in plain text unless specified otherwise. If you are asked for a comma separated list, apply the above rules depending of whether the element to be put in the list is a number or a string.
+Your answer should only start with "FINAL ANSWER: ", then follows with the answer.