Update app.py

app.py

@@ -3,11 +3,17 @@ import datetime
 import requests
 import pytz
 import yaml
+import os
+import sys
 from tools.final_answer import FinalAnswerTool
+from tools.web_search import DuckDuckGoSearchTool
+from tools.visit_webpage import VisitWebpageTool
 
 from Gradio_UI import GradioUI
 
+
 # 1 - Timezone tool
+
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
     """A tool that fetches the current local time in a specified timezone.
@@ -28,7 +34,7 @@ def get_current_time_in_timezone(timezone: str) -> str:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
 
-# 2 - Image generation
+# 2- Image generation
 @tool
 def generate_image_from_text(prompt: str) -> str:
     """A tool that generates an image based on a text description.
@@ -46,9 +52,10 @@ def generate_image_from_text(prompt: str) -> str:
         # Handle any errors that occur during image generation
         return f"Error generating image: {str(e)}"
 
-# 3 - Web search
-# Initialize the DuckDuckGo search tool
-search_tool = DuckDuckGoSearchTool()
+
+# 3 - WEB SEARCH TOOL - Using the existing DuckDuckGoSearchTool class
+# Initialize the DuckDuckGo search tool from the tools directory
+web_search_tool = DuckDuckGoSearchTool()
 
 @tool
 def search_web(query: str) -> str:
@@ -58,7 +65,7 @@ def search_web(query: str) -> str:
     """
     try:
         # Execute the search query using DuckDuckGo
-        search_results = search_tool(query)
+        search_results = web_search_tool(query)
         
         # Format and return the search results
         return f"Search results for '{query}':\n\n{search_results}"
@@ -66,12 +73,38 @@ def search_web(query: str) -> str:
         # Handle any errors that occur during the search
         return f"Error searching the web: {str(e)}"
 
-
-# This tool is required for the agent to provide final answers
+# FINAL ANSWER TOOL (REQUIRED)
+# Initialize the final answer tool - required for the agent to provide final answers
 final_answer = FinalAnswerTool()
 
 
-# Model configuration
+# 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:
+    # Import the image generation tool from Hugging Face Hub
+    # This tool will be used by the generate_image_from_text function
+    image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
+    print("✓ Image generation tool loaded successfully")
+except Exception as e:
+    print(f"✗ Failed to load image generation tool: {str(e)}")
+    # Provide a fallback if the image generation tool fails to load
+    def image_generation_tool(prompt):
+        return f"[MOCK IMAGE] Generated from prompt: {prompt}"
+
+print("✓ Time zone tool initialized")
+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'
 model = HfApiModel(
@@ -80,26 +113,34 @@ model = HfApiModel(
     model_id='Qwen/Qwen2.5-Coder-32B-Instruct',  # Using Qwen 2.5 Coder model
     custom_role_conversions=None,
 )
+print("✓ Model configured successfully")
 
-# Load External Tools
-# Import the image generation tool from Hugging Face Hub
-# This tool will be used by the generate_image_from_text function
-image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
 
+# LOAD PROMPT TEMPLATES
+print("Loading prompt templates...")
+# Create an upload directory if it doesn't exist
+os.makedirs("uploads", exist_ok=True)
 
-# Load Prompt Templates
 # Load prompt templates from YAML file for consistent agent responses
-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)
+    print("✓ Prompt templates loaded successfully")
+except Exception as e:
+    print(f"✗ Failed to load prompt templates: {str(e)}")
+    # Provide default empty templates if loading fails
+    prompt_templates = {}
     
 
-# Agent Configuration
+# AGENT CONFIGURATION
+print("Configuring agent...")
 agent = CodeAgent(
     model=model,
     tools=[
         get_current_time_in_timezone,  # Tool 1: Time zone tool
         generate_image_from_text,      # Tool 2: Image generation tool
         search_web,                    # Tool 3: Web search tool
+        visit_webpage,                 # Tool 4: Visit webpage tool (added from tools directory)
         final_answer                   # Required final answer tool
     ],
     max_steps=6,                       # Maximum number of reasoning steps
@@ -110,7 +151,9 @@ agent = CodeAgent(
     description=None,                  # No custom agent description
     prompt_templates=prompt_templates  # Using loaded prompt templates
 )
+print("✓ Agent configured successfully")
 
-# Launch GRADIO UI
-# Start the Gradio interface with our configured agent
-GradioUI(agent).launch()
+# LAUNCH THE GRADIO UI
+print("Launching Gradio UI...")
+# Start the Gradio interface with our configured agent and file upload directory
+GradioUI(agent, file_upload_folder="uploads").launch(debug=True)