Update app.py

app.py

@@ -8,6 +8,7 @@ import sys
 import uuid
 import base64
 from io import BytesIO
+from typing import Any, Union, List
 from tools.final_answer import FinalAnswerTool
 from tools.web_search import DuckDuckGoSearchTool
 from tools.visit_webpage import VisitWebpageTool
@@ -89,13 +90,13 @@ def search_web(query: str) -> str:
         return f"Error searching the web: {str(e)}"
 
 #############################################################
-# FINAL ANSWER TOOL (REQUIRED & CUSTOMIZED)
+# CUSTOM FINAL ANSWER TOOL
 #############################################################
-# Initialize the final answer tool - required for the agent to provide final answers
+# Initialize the base final answer tool
 final_answer_base = FinalAnswerTool()
 
 @tool
-def final_answer(answer) -> str:
+def final_answer(answer: Any) -> Any:
     """Provides a final answer to the given problem.
     Args:
         answer: The final answer to the problem
@@ -149,7 +150,15 @@ 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}"
+        from PIL import Image, ImageDraw, ImageFont
+        import numpy as np
+        
+        # Create a blank image with text saying this is a fallback
+        img = Image.new('RGB', (512, 512), color=(245, 245, 245))
+        d = ImageDraw.Draw(img)
+        d.text((10, 10), f"Fallback image for: {prompt}", fill=(0, 0, 0))
+        d.text((10, 30), "Image generation tool failed to load", fill=(255, 0, 0))
+        return img
 
 print("✓ Time zone tool initialized")
 print("✓ Web search tool initialized")
@@ -175,6 +184,7 @@ print("✓ Model configured successfully")
 print("Loading prompt templates...")
 # Create an upload directory if it doesn't exist
 os.makedirs("uploads", exist_ok=True)
+os.makedirs("uploads/images", exist_ok=True)
 
 # Load prompt templates from YAML file for consistent agent responses
 try:
@@ -197,7 +207,7 @@ agent = CodeAgent(
         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
+        final_answer                   # Custom final answer tool
     ],
     max_steps=6,                       # Maximum number of reasoning steps
     verbosity_level=1,                 # Level of detail in agent's output