Update app.py

app.py

@@ -1,18 +1,28 @@
+from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool
 import datetime
 import requests
 import pytz
 import yaml
-from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, load_tool, tool
 from tools.final_answer import FinalAnswerTool
-# from Gradio_UI import GradioUI # Keep this commented out while debugging via logs
 
-# --- Define Your Tools ---
+from Gradio_UI import GradioUI
+
+# Below is an example of a tool that does nothing. Amaze us with your creativity !
+@tool
+def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return type
+    #Keep this format for the description / args / args description but feel free to modify the tool
+    """A tool that does nothing yet 
+    Args:
+        arg1: the first argument
+        arg2: the second argument
+    """
+    return "What magic will you build ?"
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
     """A tool that fetches the current local time in a specified timezone.
     Args:
-        timezone: A string representing a valid timezone (e.g., 'America/New_York', 'Europe/London').
+        timezone: A string representing a valid timezone (e.g., 'America/New_York').
     """
     try:
         # Create timezone object
@@ -23,54 +33,31 @@ def get_current_time_in_timezone(timezone: str) -> str:
     except Exception as e:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
-# --- Instantiate Your Tools ---
-
-# Instantiate the DuckDuckGo search tool
-search_tool = DuckDuckGoSearchTool()
 
-# Instantiate the tool for giving the final answer
 final_answer = FinalAnswerTool()
 
-# --- Configure the Language Model ---
+# If the agent does not answer, the model is overloaded, please use another model or the following Hugging Face Endpoint that also contains qwen2.5 coder:
+# model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud' 
 
-# Use a reliable model for testing first
-# You can switch back to larger models or endpoints later once this works.
 model = HfApiModel(
-    max_tokens=2096,
-    temperature=0.5,
-    model_id='mistralai/Mistral-7B-Instruct-v0.2', # Using Mistral 7B for stability
-    custom_role_conversions=None,
+max_tokens=2096,
+temperature=0.5,
+model_id='Qwen/Qwen2.5-Coder-32B-Instruct',# it is possible that this model may be overloaded
+custom_role_conversions=None,
 )
 
-# --- Load Extra Tools (Optional) ---
 
-# Import tool from Hub (Keep it here, but we won't add it to the agent yet)
-# image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
+# Import tool from Hub
+image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
 
-# --- Load Prompts ---
-
-try:
-    with open("prompts.yaml", 'r') as stream:
-        prompt_templates = yaml.safe_load(stream)
-except FileNotFoundError:
-    print("ERROR: prompts.yaml not found. Please ensure the file exists.")
-    exit() # Exit if prompts are missing, as the agent can't work.
-except yaml.YAMLError as e:
-    print(f"ERROR: Could not parse prompts.yaml: {e}")
-    exit()
-
-# --- Create the Agent ---
-
-# Create the agent with the tools it should use
+with open("prompts.yaml", 'r') as stream:
+    prompt_templates = yaml.safe_load(stream)
+    
 agent = CodeAgent(
     model=model,
-    tools=[
-        final_answer,
-        get_current_time_in_timezone,
-        search_tool # Add the search tool instance here
-    ],
+    tools=[final_answer], ## add your tools here (don't remove final answer)
     max_steps=6,
-    verbosity_level=2, # Set to 2 for more detailed logs during debugging
+    verbosity_level=1,
     grammar=None,
     planning_interval=None,
     name=None,
@@ -78,29 +65,5 @@ agent = CodeAgent(
     prompt_templates=prompt_templates
 )
 
-# --- Run the Agent (Directly for Debugging) ---
-
-print("--- Starting Agent Run ---")
-try:
-    # Use a query that requires search, like your weather question
-    # Or test the time tool: "What time is it in Europe/London?"
-    result = agent.run("what is the weather in aleppo city in syria?")
-    print("--- Agent Run Finished ---")
-    print("Agent Result:", result)
-except Exception as e:
-    print("--- An error occurred during agent run: ---")
-    import traceback
-    traceback.print_exc() # Print the full error traceback
-
-# --- Launch Gradio UI (Commented Out) ---
-# Once the agent works reliably via logs, you can uncomment this
-# print("--- Launching Gradio UI ---")
-# GradioUI(agent).launch()
 
-try:
-    result = agent.run("what is the weather in aleppo city in syria?") # Or another test message
-    print("Agent Result:", result)
-except Exception as e:
-    print("An error occurred during agent run:", e)
-    import traceback
-    traceback.print_exc() # <--- This will print the full error
+GradioUI(agent).launch()