Update app.py

app.py

@@ -3,6 +3,7 @@ import gradio as gr
 import requests
 import inspect
 import pandas as pd
+from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel, load_tool, tool
 
 # (Keep Constants as is)
 # --- Constants ---
@@ -12,10 +13,47 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
 class BasicAgent:
     def __init__(self):
+
+        self.model = HfApiModel(
+            max_tokens=3000,
+            temperature=0.5,
+            model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
+            #model_id='deepseek-ai/DeepSeek-R1-Distill-Qwen-32B',
+            #model_id='mistralai/Mistral-7B-Instruct-v0.3',
+            #model_id='https://wxknx1kg971u7k1n.us-east-1.aws.endpoints.huggingface.cloud',
+            #model_id="google/gemma-2-9b-it",
+            custom_role_conversions=None,
+            token=os.getenv('hf_token'),
+        )
+
+        self.agent = CodeAgent(
+            model=self.model,
+            tools=[
+                final_answer, 
+                #image_generation_tool, 
+                #get_stock_data_baostock, 
+                #weather_data_tool
+            ], ## add your tools here (don't remove final answer)
+            max_steps=4,
+            verbosity_level=1,
+            grammar=None,
+            planning_interval=3,
+            name=None,
+            description=None,
+            #prompt_templates=prompt_templates,
+            add_base_tools=True,
+            additional_authorized_imports=["time", "numpy", "pandas","datetime"],
+        )
+        
         print("BasicAgent initialized.")
+
+        
     def __call__(self, question: str) -> str:
         print(f"Agent received question (first 50 chars): {question[:50]}...")
-        fixed_answer = "This is a default answer."
+        
+        fixed_answer = agent.run(question)
+        
+        #fixed_answer = "This is a default answer."
         print(f"Agent returning fixed answer: {fixed_answer}")
         return fixed_answer