Update app.py

app.py

@@ -1,155 +1,106 @@
-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
 
-
-
-from Gradio_UI import GradioUI
-
-
-
-# Below is an example of a tool that does nothing. Amaze us with your creativity !
-
-@tool
-
-def search(arg:str)-> 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 searches on the web and return the results 
-
-    Args:
-
-        arg: the topic to search about
-
-    """
-
-    return DuckDuckGoSearchTool(arg)
-
-
+# --- Define Your Tools ---
 
 @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').
-
-    """
-
-    try:
-
-        # Create timezone object
-
-        tz = pytz.timezone(timezone)
-
-        # Get current time in that timezone
-
-        local_time = datetime.datetime.now(tz).strftime("%Y-%m-%d %H:%M:%S")
-
-        return f"The current local time in {timezone} is: {local_time}"
-
-    except Exception as e:
-
-        return f"Error fetching time for timezone '{timezone}': {str(e)}"
-
-
-
+    """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').
+    """
+    try:
+        # Create timezone object
+        tz = pytz.timezone(timezone)
+        # Get current time in that timezone
+        local_time = datetime.datetime.now(tz).strftime("%Y-%m-%d %H:%M:%S")
+        return f"The current local time in {timezone} is: {local_time}"
+    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='Qwen/Qwen2.5-Coder-32B-Instruct',# it is possible that this model may be overloaded
-
-custom_role_conversions=None,
-
+    max_tokens=2096,
+    temperature=0.5,
+    model_id='mistralai/Mistral-7B-Instruct-v0.2', # Using Mistral 7B for stability
+    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)
 
+# --- Load Prompts ---
 
-
-# Import tool from Hub
-
-image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
-
-
-
-with open("prompts.yaml", 'r') as stream:
-
-    prompt_templates = yaml.safe_load(stream)
-
-    
-
+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
 agent = CodeAgent(
-
-    model=model,
-
-    tools=[final_answer,
-
-           get_current_time_in_timezone,
-
-           search_tool], ## add your tools here (don't remove final answer)
-
-    max_steps=6,
-
-    verbosity_level=1,
-
-    grammar=None,
-
-    planning_interval=None,
-
-    name=None,
-
-    description=None,
-
-    prompt_templates=prompt_templates
-
+    model=model,
+    tools=[
+        final_answer,
+        get_current_time_in_timezone,
+        search_tool # Add the search tool instance here
+    ],
+    max_steps=6,
+    verbosity_level=2, # Set to 2 for more detailed logs during debugging
+    grammar=None,
+    planning_interval=None,
+    name=None,
+    description=None,
+    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