Update prompt; fix requirements.

app.py

@@ -152,7 +152,7 @@ def fetch_and_run_single(selected_id, questions_data, profile: gr.OAuthProfile |
         print(f"User logged in: {profile.username}")
     else:
         print("User not logged in.")
-        return "Please Login to Hugging Face with the button.", None, None
+        return "Please Login to Hugging Face with the button.", None, None, []
 
     try:
         index = int(selected_id)
@@ -160,9 +160,9 @@ def fetch_and_run_single(selected_id, questions_data, profile: gr.OAuthProfile |
         task_id = question_item.get("task_id")
         question_text = question_item.get("question")
         if not task_id or question_text is None:
-            return "Invalid question format received.", None, None
+            return "Invalid question format received.", None, None, []
     except Exception as e:
-        return f"Error selecting question: {e}", None, None
+        return f"Error selecting question: {e}", None, None, []
 
     agent = init_agent()
     generated_answer = agent(question_item)
@@ -171,7 +171,8 @@ def fetch_and_run_single(selected_id, questions_data, profile: gr.OAuthProfile |
         "Question": question_text,
         "Generated Answer": generated_answer
     }])
-    return "Fetched and ran agent on selected question.", result_df, question_item
+    logs = agent.get_logs()
+    return "Fetched and ran agent on selected question.", result_df, question_item, logs
 
 
 # --- Build Gradio Interface using Blocks ---
@@ -230,11 +231,12 @@ with (gr.Blocks() as demo):
     single_question_json = gr.JSON(label="Raw Question JSON")
     single_status = gr.Textbox(label="Single Question Status", lines=2, interactive=False)
     single_result_table = gr.DataFrame(label="Single Question and Answer", wrap=True)
+    single_question_logs = gr.JSON(label="Agent logs")
 
     single_run_button.click(
         fn=fetch_and_run_single,
         inputs=[question_id_dropdown, questions_data_state],
-        outputs=[single_status, single_result_table, single_question_json]
+        outputs=[single_status, single_result_table, single_question_json, single_question_logs]
     )
 
     # All questions for submission run

basic_agent.py

@@ -71,19 +71,20 @@ class BasicSmolAgent:
         if not model:
             model = HfApiModel()
         search_tool = DuckDuckGoSearchToolWH()
-        self.agent = CodeAgent(tools=[search_tool], model=model)
-        self.prompt = ("You are a general AI assistant. I will ask you a question."
-                             " Return only 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."
-                             " make sure you don’t include the text “FINAL ANSWER” in your submission, just reply with the answer and nothing else."
-                            " The question is the following: {}")
+        self.agent = CodeAgent(tools=[search_tool], model=model, max_steps=10)
+        self.prompt = ("The question is the following:\n ```{}```"
+                       " 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."
+                       )
         # Load the Whisper pipeline
         self.mp3_pipe = pipeline("automatic-speech-recognition", model="openai/whisper-base")
 
+    def get_logs(self):
+        return self.agent.memory.steps
+
     def __call__(self, question_item: dict) -> str:
         task_id = question_item.get("task_id")
         question_text = question_item.get("question")

requirements.txt

@@ -5,4 +5,5 @@ python-dotenv
 pandas
 duckduckgo_search
 smolagents
-transformers
+transformers
+torch