Update app.py

app.py

@@ -3,13 +3,31 @@ import gradio as gr
 import requests
 import inspect
 import pandas as pd
-
+from langchain_core.prompts import ChatPromptTemplate
 from smolagents import CodeAgent, DuckDuckGoSearchTool, OpenAIServerModel, WikipediaSearchTool
 
 # (Keep Constants as is)
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
+SYSTEM_PROMPT = """
+You are GAIA-Bot, a fully-autonomous problem-solver.
+• For each question, THINK step-by-step before answering.  
+• When outside knowledge or calculations are needed, CALL an appropriate tool.  
+• Explain your reasoning only to yourself (in “Thought” messages); do NOT reveal it to the user.  
+• After finishing all reasoning, RESPOND with **only** the final answer string—no pre-amble, no labels, no extra text.
+You have access to these tools:
+- gaia_file_fetch    – download the file attached to the current task_id.
+- duckduckgo_search  – look up current facts on the public web.
+If you require a file attached to the current task, download it with
+`requests_get("https://agents-course-unit4-scoring.hf.space/files/{task_id}")`
+and parse it inside Python.
+Adhere strictly to these rules:
+1. Use tools whenever they can reduce hallucination.  
+2. Never mention tool names or JSON in your final answer.  
+3. If you cannot find an answer, return “unknown”.
+Begin.
+"""
 # --- Basic Agent Definition ---
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
 class BasicAgent:
@@ -18,12 +36,15 @@ class BasicAgent:
 
         search_tool = DuckDuckGoSearchTool()
         wiki_search = WikipediaSearchTool()
-
+        prompt = ChatPromptTemplate.from_messages([
+            ("system", SYSTEM_PROMPT),
+            ("user", "{input}")
+        ])
         self.agent = CodeAgent(
             model = model,
-            tools=[
-                search_tool,
-            ]
+            tools=[search_tool,wiki_search],
+            prompt=prompt,
+
         )
 
     def __call__(self, question: str) -> str: