Update app.py

app.py

@@ -11,9 +11,9 @@ from tools.visit_webpage import VisitWebpageTool
 
 from Gradio_UI import GradioUI
 
-
-# 1 - Timezone tool
-
+#############################################################
+# TOOL 1: TIME ZONE TOOL
+#############################################################
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
     """A tool that fetches the current local time in a specified timezone.
@@ -34,26 +34,43 @@ def get_current_time_in_timezone(timezone: str) -> str:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
 
-# 2- Image generation
+#############################################################
+# TOOL 2: IMAGE GENERATION TOOL
+#############################################################
 @tool
 def generate_image_from_text(prompt: str) -> str:
-    """A tool that generates an image based on a text description.
+    """A tool that generates an image based on a text description and saves it to a file.
     Args:
         prompt: A detailed text description of the image you want to generate.
     """
     try:
-        # Call the image generation tool loaded from Hugging Face Hub
-        # The tool is loaded further down in the script before it's used here
-        result = image_generation_tool(prompt)
+        import os
+        import uuid
+        from datetime import datetime
+        
+        # Create images directory if it doesn't exist
+        os.makedirs("uploads/images", exist_ok=True)
+        
+        # Generate a unique filename
+        timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
+        unique_id = str(uuid.uuid4())[:8]
+        filename = f"uploads/images/{timestamp}_{unique_id}.jpg"
+        
+        # Call the image generation tool to get the PIL image
+        pil_image = image_generation_tool(prompt)
+        
+        # Save the PIL image to file
+        pil_image.save(filename)
         
-        # Return success message with the result (which should contain image URL or path)
-        return f"Image generated successfully: {result}"
+        # Return the file path
+        return filename
     except Exception as e:
         # Handle any errors that occur during image generation
         return f"Error generating image: {str(e)}"
 
-
-# 3 - WEB SEARCH TOOL - Using the existing DuckDuckGoSearchTool class
+#############################################################
+# TOOL 3: WEB SEARCH TOOL - Using the existing DuckDuckGoSearchTool class
+#############################################################
 # Initialize the DuckDuckGo search tool from the tools directory
 web_search_tool = DuckDuckGoSearchTool()
 
@@ -73,17 +90,21 @@ def search_web(query: str) -> str:
         # Handle any errors that occur during the search
         return f"Error searching the web: {str(e)}"
 
+#############################################################
 # FINAL ANSWER TOOL (REQUIRED)
+#############################################################
 # Initialize the final answer tool - required for the agent to provide final answers
 final_answer = FinalAnswerTool()
 
-
+#############################################################
 # VISIT WEBPAGE TOOL - From tools directory
+#############################################################
 # Initialize the visit webpage tool
 visit_webpage = VisitWebpageTool()
 
-
+#############################################################
 # TOOL INITIALIZATION AND VERIFICATION
+#############################################################
 # Ensure all tools are properly loaded and print status
 print("Initializing tools...")
 try:
@@ -102,21 +123,22 @@ print("✓ Web search tool initialized")
 print("✓ Visit webpage tool initialized")
 print("✓ Final answer tool initialized")
 
-
+#############################################################
 # MODEL CONFIGURATION
+#############################################################
 print("Configuring model...")
-# If the agent does not answer, the model is overloaded
-# Alternative endpoint: 'https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud'
+# Use the alternative endpoint since the primary one seems to have authentication issues
 model = HfApiModel(
     max_tokens=2096,           # Maximum number of tokens in the response
     temperature=0.5,           # Controls randomness: lower = more deterministic
-    model_id='Qwen/Qwen2.5-Coder-32B-Instruct',  # Using Qwen 2.5 Coder model
+    model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud',  # Alternative endpoint
     custom_role_conversions=None,
 )
 print("✓ Model configured successfully")
 
-
+#############################################################
 # LOAD PROMPT TEMPLATES
+#############################################################
 print("Loading prompt templates...")
 # Create an upload directory if it doesn't exist
 os.makedirs("uploads", exist_ok=True)
@@ -131,8 +153,9 @@ except Exception as e:
     # Provide default empty templates if loading fails
     prompt_templates = {}
     
-
+#############################################################
 # AGENT CONFIGURATION
+#############################################################
 print("Configuring agent...")
 agent = CodeAgent(
     model=model,
@@ -153,7 +176,9 @@ agent = CodeAgent(
 )
 print("✓ Agent configured successfully")
 
+#############################################################
 # LAUNCH THE GRADIO UI
+#############################################################
 print("Launching Gradio UI...")
 # Start the Gradio interface with our configured agent and file upload directory
 # Remove the debug parameter as it's already set in the GradioUI.launch method