Update app.py

app.py

@@ -96,22 +96,20 @@ class BasicAgent:
                     print(f"✅ Model loaded on device: {self.model.model.device}")
                 else:
                     print("✅ Model loaded (device info not directly accessible)")
-
-                # Initialize the search tool
-                self.search_tool = DuckDuckGoSearchTool()
                 
                 # Create CodeAgent with DuckDuckGoSearchTool and additional imports
                 self.agent = CodeAgent(
-                    tools=[self.search_tool],
+                    tools=[],
                     model=self.model,
                     max_steps=30,
                     additional_authorized_imports=[
-                        'math', 'statistics', 're',  # Basic computation
+                        'math', 'statistics', 're',   # Basic computation
                         'requests', 'json',           # Web requests and JSON
-                        'pandas', 'numpy',            # Data analysis
+                        'pandas', 'numpy', 'openpyxl',# Data analysis
                         'zipfile', 'os',              # File processing
                         'datetime', 'time'            # Date/time operations
-                    ]
+                    ],
+                    add_base_tools=True,
                 )
 
                 self.tools_available = True
@@ -132,7 +130,7 @@ class BasicAgent:
         """Run question through smolagents CodeAgent with enhanced prompting."""
         try:
             # Use the global CURRENT_PROMPT variable
-            formatted_question = CURRENT_PROMPT.format(question=question)
+            formatted_question = question #CURRENT_PROMPT.format(question=question)
 
             print(f"🔄 Processing question: {question}")
             print(f"🔧 Available tools: {[tool.__class__.__name__ for tool in self.agent.tools]}")