Update app.py

app.py

@@ -6,13 +6,26 @@ import requests
 import pandas as pd
 
 from smolagents import LiteLLMModel, CodeAgent, DuckDuckGoSearchTool
+from gaia_tools import ReverseTextTool, RunPythonFileTool, download_server
 
 # System prompt for the agent
 SYSTEM_PROMPT = """You are a general AI assistant. I will ask you a question.
 Report your thoughts, and finish your answer with just the answer — no prefixes like "FINAL ANSWER:".
 Your answer should be a number OR as few words as possible OR a comma-separated list of numbers and/or strings.
 If you're asked for a number, don’t use commas or units like $ or %, unless specified.
-If you're asked for a string, don’t use articles or abbreviations (e.g. for cities), and write digits in plain text unless told otherwise."""
+If you're asked for a string, don’t use articles or abbreviations (e.g. for cities), and write digits in plain text unless told otherwise.
+
+Tool Use Guidelines:
+1. Do **not** use any tools outside of the provided tools list.
+2. Always use **only one tool at a time** in each step of your execution.
+3. If the question refers to a `.py` file or uploaded Python script, use **RunPythonFileTool** to execute it and base your answer on its output.
+4. If the question looks reversed (starts with a period or reads backward), first use **ReverseTextTool** to reverse it, then process the question.
+5. For logic or word puzzles, solve them directly unless they are reversed — in which case, decode first using **ReverseTextTool**.
+6. When dealing with Excel files, prioritize using the **excel** tool over writing code in **terminal-controller**.
+7. If you need to download a file, always use the **download_server** tool and save it to the correct path.
+8. Even for complex tasks, assume a solution exists. If one method fails, try another approach using different tools.
+9. Due to context length limits, keep browser-based tasks (e.g., searches) as short and efficient as possible.
+"""
 
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
@@ -23,16 +36,19 @@ class MyAgent:
         if not gemini_api_key:
             raise ValueError("GEMINI_API_KEY not set in environment variables.")
         
-        # Instantiate LiteLLMModel with Gemini API key and model id
         self.model = LiteLLMModel(
             model_id="gemini/gemini-2.0-flash-lite",
             api_key=gemini_api_key,
             system_prompt=SYSTEM_PROMPT
         )
         
-        # Create the CodeAgent with optional base tools and DuckDuckGo search
         self.agent = CodeAgent(
-            tools=[DuckDuckGoSearchTool()],
+            tools=[
+                DuckDuckGoSearchTool(),
+                ReverseTextTool(),
+                RunPythonFileTool(),
+                download_server()
+            ],
             model=self.model,
             add_base_tools=True,
         )
@@ -132,4 +148,3 @@ if __name__ == "__main__":
 
 
 
-