Update app.py

app.py

@@ -1,36 +1,47 @@
-from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool
+from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, load_tool, tool
 import datetime
 import requests
 import pytz
 import yaml
 from tools.final_answer import FinalAnswerTool
-
 from Gradio_UI import GradioUI
 
+# IMPORTANT: Initialize model BEFORE using it in tools
+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,
+)
 
 @tool
-def translate_to_spanish (english : str)-> str: 
+def translate_to_spanish(english: str) -> str: 
     """A tool that translates text from english to spanish using DuckDuckGoSearchTool
     Args:
         english (str): the text in english to be translated to spanish
     Returns: 
         str: the text translated into spanish
-
     """
-    query = "Please translate the following text into Spanish {english}"
-
-# Use a temporary agent for the search (reuse model for efficiency)
+    # FIXED: Proper f-string formatting and return statement
+    query = f"Please translate the following text into Spanish: {english}"
+    
+    # Use a temporary agent for the search (reuse model for efficiency)
     temp_agent = CodeAgent(
         tools=[DuckDuckGoSearchTool()],
         model=model
     )
     results = temp_agent.run(query)
     
+    # FIXED: Actually return the results
+    return str(results)
+
 @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').
+    Returns:
+        str: The current time in the specified timezone
     """
     try:
         # Create timezone object
@@ -41,29 +52,35 @@ def get_current_time_in_timezone(timezone: str) -> str:
     except Exception as e:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
-
+# Initialize tools
 final_answer = FinalAnswerTool()
 
-# 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' 
-
-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,
-)
+# FIXED: Added error handling for optional dependencies
+try:
+    # Import tool from Hub
+    image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
+    tools_list = [final_answer, translate_to_spanish, get_current_time_in_timezone, 
+                  DuckDuckGoSearchTool(), image_generation_tool]
+except Exception as e:
+    print(f"Warning: Could not load image generation tool: {e}")
+    tools_list = [final_answer, translate_to_spanish, get_current_time_in_timezone, 
+                  DuckDuckGoSearchTool()]
 
+# FIXED: Added error handling for prompts.yaml
+try:
+    with open("prompts.yaml", 'r') as stream:
+        prompt_templates = yaml.safe_load(stream)
+except FileNotFoundError:
+    print("Warning: prompts.yaml not found, using default templates")
+    prompt_templates = None
+except Exception as e:
+    print(f"Warning: Error loading prompts.yaml: {e}")
+    prompt_templates = None
 
-# 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)
-    
+# Create agent
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, translate_to_spanish, get_current_time_in_timezone, DuckDuckGoSearchTool(), image_generation_tool,], ## add your tools here (don't remove final answer)
+    tools=tools_list,  # Use the dynamically created tools list
     max_steps=6,
     verbosity_level=1,
     grammar=None,
@@ -73,5 +90,9 @@ agent = CodeAgent(
     prompt_templates=prompt_templates
 )
 
-
-GradioUI(agent).launch()
+# FIXED: Added error handling for Gradio launch
+try:
+    GradioUI(agent).launch()
+except Exception as e:
+    print(f"Error launching Gradio UI: {e}")
+    print("You may need to install gradio or check your Gradio_UI module")