Update app.py

app.py

@@ -5,6 +5,8 @@ import inspect
 import pandas as pd
 import json
 
+from smolagents import CodeAgent, DuckDuckGoSearchTool, InferenceClientModel, load_tool, tool
+
 # (Keep Constants as is)
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
@@ -20,46 +22,66 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
 class BasicAgent:
     def __init__(self):
-        
-        
+        final_answer = FinalAnswerTool()      
         print("BasicAgent initialized.")
+
+        model = = OpenAIServerModel(
+        api_key="fw_3ZSrkw3Qv5eeA1yx65x1eD5H",
+        api_base="https://api.fireworks.ai/inference/v1",
+        model_id="accounts/fireworks/models/qwen2p5-72b-instruct",
+        temperature = 0.1,
+        max_tokens = 100
+        #top_p = 1
+        
     def __call__(self, question: str) -> str:
         print(f"Agent received question (first 50 chars): {question[:50]}...")
+
+
+        uav_agent = CodeAgent(
+        tools=[final_answer, DuckDuckGoSearchTool],#[detection_tool],
+        model=model,
+        max_steps=5,
+        verbosity_level=1,
+        planning_interval=2,
+        #additional_authorized_imports=["json", "from PIL import Image, ImageDraw, ImageFont", "import matplotlib.pyplot as plt", "from typing import List, Dict"],
+        #managed_agents=[detection_agent]
+        name="agent_instance",
+        description="Solves the task"
+        )
         
         
-        
-        url = "https://api.fireworks.ai/inference/v1/chat/completions"
-        payload = {
-          "model": "accounts/fireworks/models/qwen2p5-72b-instruct",
-          "max_tokens": 100,
-          "top_p": 1,
-          "top_k": 40,
-          "presence_penalty": 0,
-          "frequency_penalty": 0,
-          "temperature": 0.05,
-          "messages": [
-            {
-              "role": "user",
-              "content": f"""{question} Provide only what is requested in answer SHORTLY. If there is requested to provide ONE word, provide it. 
-                THis is for the Benchmark evaluation, so the answer should be REALLY SHORT and ACCURATE.
-                """
-            }
-          ]
-        }
-        headers = {
-          "Accept": "application/json",
-          "Content-Type": "application/json",
-          "Authorization": "Bearer fw_3ZSrkw3Qv5eeA1yx65x1eD5H"
-        }
-        response = requests.post(url, headers=headers, data=json.dumps(payload))
+        # url = "https://api.fireworks.ai/inference/v1/chat/completions"
+        # payload = {
+        #   "model": "accounts/fireworks/models/qwen2p5-72b-instruct",
+        #   "max_tokens": 100,
+        #   "top_p": 1,
+        #   "top_k": 40,
+        #   "presence_penalty": 0,
+        #   "frequency_penalty": 0,
+        #   "temperature": 0.05,
+        #   "messages": [
+        #     {
+        #       "role": "user",
+        #       "content": f"""{question} Provide only what is requested in answer SHORTLY. If there is requested to provide ONE word, provide it. 
+        #         THis is for the Benchmark evaluation, so the answer should be REALLY SHORT and ACCURATE.
+        #         """
+        #     }
+        #   ]
+        # }
+        # headers = {
+        #   "Accept": "application/json",
+        #   "Content-Type": "application/json",
+        #   "Authorization": "Bearer fw_3ZSrkw3Qv5eeA1yx65x1eD5H"
+        # }
+        # response = requests.post(url, headers=headers, data=json.dumps(payload))
                 
         
-        response_data = response.json()
-        assistant_reply = response_data["choices"][0]["message"]["content"]
+        # response_data = response.json()
+        # assistant_reply = response_data["choices"][0]["message"]["content"]
         
         
-        fixed_answer = assistant_reply
-        #fixed_answer = "This is a default answer."
+        # fixed_answer = assistant_reply
+        # #fixed_answer = "This is a default answer."
 
 
 
@@ -67,7 +89,7 @@ class BasicAgent:
 
 
         
-        print(f"Agent returning fixed answer: {fixed_answer}")
+        # print(f"Agent returning fixed answer: {fixed_answer}")
         return fixed_answer