Update app.py

app.py

@@ -6,10 +6,10 @@ import yaml
 from tools.final_answer import FinalAnswerTool 
 from Gradio_UI import GradioUI
 
-# Enhanced custom tool for anime image generation
+# Enhanced custom tool for anime image generation using the actual image generation tool
 @tool
-def anime_image_generator(description: str, style: str = "anime") -> str:
-    """A tool that generates anime-style images based on user descriptions.
+def generate_anime_image(description: str, style: str = "anime") -> str:
+    """A tool that generates actual anime-style images based on user descriptions.
     
     Args:
         description: Detailed description of the anime image to generate
@@ -18,18 +18,19 @@ def anime_image_generator(description: str, style: str = "anime") -> str:
     try:
         # Enhance the prompt with anime-specific keywords
         anime_keywords = {
-            "anime": "anime style, high quality, detailed, vibrant colors, japanese animation style",
-            "manga": "manga style, black and white, detailed lineart, dramatic shading",
-            "kawaii": "kawaii style, cute, adorable, pastel colors, chibi proportions",
-            "realistic_anime": "realistic anime style, detailed face, expressive eyes, high resolution"
+            "anime": "anime style, high quality, detailed, vibrant colors, japanese animation style, 2d animation",
+            "manga": "manga style, black and white, detailed lineart, dramatic shading, monochrome",
+            "kawaii": "kawaii style, cute, adorable, pastel colors, chibi proportions, moe",
+            "realistic_anime": "realistic anime style, detailed face, expressive eyes, high resolution, semi-realistic"
         }
         
         style_prompt = anime_keywords.get(style, anime_keywords["anime"])
-        enhanced_description = f"{description}, {style_prompt}, masterpiece, best quality"
+        enhanced_description = f"{description}, {style_prompt}, masterpiece, best quality, detailed artwork"
         
-        return f"Generated anime image with description: '{enhanced_description}' in {style} style"
+        # This will be used with the actual image generation tool in the agent workflow
+        return enhanced_description
     except Exception as e:
-        return f"Error generating anime image: {str(e)}"
+        return f"Error preparing anime image prompt: {str(e)}"
 
 @tool
 def analyze_image_description(user_input: str) -> str:
@@ -129,15 +130,18 @@ with open("prompts.yaml", 'r') as stream:
 # Create the agent with anime-focused tools
 tools_list = [
     final_answer,
-    anime_image_generator,
+    generate_anime_image,
     analyze_image_description,
     enhance_anime_prompt,
     get_current_time_in_timezone
 ]
 
-# Add the image generation tool if it loaded successfully
+# Add the image generation tool if it loaded successfully - this is the key tool for actual image generation
 if image_generation_tool:
     tools_list.append(image_generation_tool)
+    print("Image generation tool loaded successfully!")
+else:
+    print("Warning: Image generation tool not available - only text descriptions will be provided")
 
 agent = CodeAgent(
     model=model,