Update app.py

app.py

@@ -6,54 +6,13 @@ import yaml
 import re
 import tempfile
 import os
-import random
-import io
-import base64
-from PIL import Image, ImageDraw, ImageFont
 from tools.final_answer import FinalAnswerTool
 from Gradio_UI import GradioUI
 
-# Alternative image generation function that doesn't rely on the paid API
-def generate_simple_image(prompt, width=800, height=800):
-    """
-    Generate a simple placeholder image based on the prompt.
-    This is a fallback when the primary image generation service fails.
-    """
-    try:
-        # Create a blank white image
-        image = Image.new('RGB', (width, height), color='white')
-        draw = ImageDraw.Draw(image)
-        
-        # Try to load a font, fall back to default if not available
-        try:
-            font = ImageFont.truetype("arial.ttf", 20)
-        except IOError:
-            font = ImageFont.load_default()
-        
-        # Draw a border
-        draw.rectangle([(10, 10), (width-10, height-10)], outline='black', width=2)
-        
-        # Add text from the prompt (first 50 characters)
-        short_prompt = prompt[:50] + "..." if len(prompt) > 50 else prompt
-        text_position = (width//2, height//2)
-        
-        # Draw centered text
-        draw.text(text_position, short_prompt, fill='black', font=font, anchor='mm')
-        
-        # Add a message that this is a placeholder
-        draw.text((width//2, height-40), "Placeholder - Image generation unavailable", 
-                 fill='red', font=font, anchor='mm')
-        
-        return image
-    except Exception as e:
-        # If even this fails, create an absolute minimum fallback
-        image = Image.new('RGB', (400, 400), color='white')
-        return image
-
 @tool
 def generate_passover_coloring_pages(age: int) -> str:
     """
-    Generates Jewish Passover/Pesach-themed coloring pages based on a child's age.
+    Generates Passover-themed coloring pages based on a child's age.
     Args:
         age (int): The age of the child (1-12).
     Returns:
@@ -71,80 +30,47 @@ def generate_passover_coloring_pages(age: int) -> str:
     
     complexity = next(value for (low, high), value in complexity_levels.items() if low <= age <= high)
     
-    # Enhanced prompts specifically for Jewish Passover/Pesach themes with anti-hallucination guidance
+    # Enhanced prompts for better coloring pages
     passover_pages = {
         "simple": [
-            "Matzah bread outline only for coloring page, simple flat black lines on white background, no humans, no hands",
-            "Empty Seder plate outline with labeled sections for coloring, simple black lines on white, no people",
-            "Afikoman illustration, folded matzah in napkin, simple black outline on white background",
-            "Simple Kiddush cup outline drawing for coloring, plain black lines on white"
+            "Line drawing of a matzah for a simple coloring page, thick black outlines, white background, perfect for a child's coloring book",
+            "Simple black and white outline of a Seder plate with clear divisions, coloring book style, thick lines",
+            "Black and white line art of baby Moses in a basket for a child's coloring book, simple outlines",
+            "Simple black and white outline of a frog (plague) for children's coloring book, thick lines, no shading"
         ],
         "basic": [
-            "Red Sea parting scene outline with simple waves and pathway, no detailed humans, black lines on white background for coloring",
-            "Ten Plagues symbols in separate boxes (frog, locusts, etc.), simple black outlines for coloring, no human figures",
-            "Four Questions illustrated as symbols with Hebrew letters, simple black outlines for coloring",
-            "Simple Afikoman maze with straight lines, black outlines on white background"
+            "Black and white outline of the Parting of the Red Sea scene for a coloring book, clear distinct lines",
+            "Coloring book page with Ten Plagues illustrations, black outline on white background, no shading",
+            "Black and white outline illustrations of the Four Questions for a Passover coloring book",
+            "Afikoman hunt maze in black and white outline for a coloring book, thick lines"
         ],
         "intermediate": [
-            "Seder table with symbolic foods outlined in black and white, overhead view only, no people",
-            "Hebrew letters spelling 'Pesach' (פסח) in block style for coloring, black outlines on white",
-            "Stylized exodus scene with distant simplified figures crossing sea, black outline only, no detailed humans",
-            "Four Sons from Haggadah represented by simple symbols (book, question mark, etc.), no human figures"
+            "Detailed black and white outline of a Seder table for a coloring book, clear distinct elements",
+            "Black and white outline of Exodus story scenes for coloring book, clear lines",
+            "Hebrew letter decorations for 'Pesach' in black and white outline style for coloring book",
+            "Black and white outline of Egyptian pyramids with Israelites for a coloring book"
         ],
         "advanced": [
-            "Haggadah page layout with decorative borders and Hebrew text outlines, black lines on white, no figures",
-            "Tambourines and musical notes pattern for coloring, black outlines representing celebration, no human figures",
-            "Hebrew calligraphy 'Next year in Jerusalem' (לשנה הבאה בירושלים) with decorative elements, black outline",
-            "Symbolic Passover journey map with simple icons (pyramid, sea, mountain), black outline for coloring, no detailed humans"
+            "Intricate black and white line art of Haggadah illustrations for a coloring book, detailed outlines",
+            "Detailed symbolism scene in black and white outline for an advanced coloring book",
+            "Combined Passover story panels in black and white line art for coloring book",
+            "Hebrew calligraphy art in black and white outline for a coloring book, intricate details"
         ]
     }
     
-    # Added anti-hallucination qualifiers
-    anti_hallucination_qualifiers = [
-        "simple black and white outlines only", 
-        "no anatomical errors", 
-        "avoid human figures with details", 
-        "correct proportions", 
-        "coloring book style with clear lines",
-        "no shading or gradients",
-        "anatomically correct if any animals appear",
-        "simplified representations",
-        "no realistic humans",
-        "coloring page suitable for printing"
-    ]
-    
     selected_pages = passover_pages[complexity]
-    descriptions = [page.split(',')[0] if ',' in page else page for page in selected_pages]
+    descriptions = [f"- {page.split(',')[0]}" for page in selected_pages]
     
-    # Try multiple image generation approaches with fallbacks
+    # Try to generate images with improved prompts
     try:
-        # First attempt: Try to use the original tool if available
-        try:
-            # Try to load the original image generation tool
-            image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
-            
-            # Use more specific prompts for coloring pages with anti-hallucination emphasis
-            enhanced_prompts = []
-            for page in selected_pages:
-                # Add random anti-hallucination qualifiers to each prompt
-                qualifiers = ", ".join(anti_hallucination_qualifiers)
-                enhanced_prompt = f"{page}, Jewish traditional Passover symbols only, {qualifiers}, black and white coloring book page"
-                enhanced_prompts.append(enhanced_prompt)
-            
-            # Generate images with the tool
-            images = [image_generation_tool(prompt) for prompt in enhanced_prompts]
+        image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
         
-        except Exception as e:
-            # If the primary approach fails, try an alternative API
-            try:
-                # This is a placeholder for an alternative API call
-                # In a real implementation, you'd replace this with another API
-                # For now, we'll use our fallback function
-                raise Exception("Using fallback image generation")
-                
-            except Exception as alt_error:
-                # If both approaches fail, use our simple fallback image generator
-                images = [generate_simple_image(prompt) for prompt in selected_pages]
+        # Use more specific prompts for coloring pages
+        enhanced_prompts = []
+        for page in selected_pages:
+            enhanced_prompts.append(f"{page}, suitable for printing as a coloring page")
+        
+        images = [image_generation_tool(prompt) for prompt in enhanced_prompts]
         
         # Save images to temporary files
         image_paths = []
@@ -156,16 +82,16 @@ def generate_passover_coloring_pages(age: int) -> str:
                 image_paths.append(temp_path)
         
         # Create a formatted response with descriptions
-        response = f"## Jewish Passover (Pesach) Coloring Pages for a {age}-year-old:\n\n"
-        response += "\n".join([f"- {desc}" for desc in descriptions])
+        response = f"## Passover Coloring Pages for a {age}-year-old:\n\n"
+        response += "\n".join(descriptions)
         
         # Return the response with image paths
         return f"{response}\n\nImages are saved and ready for display."
     except Exception as e:
-        # If all image generation methods fail, return text description
-        response = f"## Jewish Passover (Pesach) Coloring Pages for a {age}-year-old:\n\n"
-        response += "\n".join([f"- {desc}" for desc in descriptions])
-        return f"{response}\n\nWe couldn't generate images at this time, but here are descriptions of coloring pages you can create:\n\n" + "\n".join([f"- {desc}" for desc in descriptions])
+        # If image generation fails, return text description
+        response = f"## Passover Coloring Pages for a {age}-year-old:\n\n"
+        response += "\n".join(descriptions)
+        return f"{response}\n\nError generating images: {str(e)}"
 
 # Custom UI wrapper to handle image display
 class CustomGradioUI(GradioUI):
@@ -179,8 +105,7 @@ class CustomGradioUI(GradioUI):
         with gr.Blocks() as demo:
             with gr.Row():
                 with gr.Column(scale=1):
-                    gr.Markdown("# Jewish Passover (Pesach) Coloring Pages Generator")
-                    gr.Markdown("Generate traditional Jewish Passover coloring pages for children")
+                    gr.Markdown("# Passover Coloring Pages Generator")
                 
             with gr.Row():
                 with gr.Column(scale=1):
@@ -192,7 +117,7 @@ class CustomGradioUI(GradioUI):
                     output_text = gr.Markdown(label="Generated Pages")
                     
             with gr.Row():
-                gallery = gr.Gallery(label="Jewish Passover Coloring Pages", show_label=True)
+                gallery = gr.Gallery(label="Coloring Pages", show_label=True)
                 
             def process_input(input_text):
                 try:
@@ -213,17 +138,6 @@ class CustomGradioUI(GradioUI):
                     # Display the most recent images (last 4 if there are more)
                     image_paths = sorted(image_paths, key=os.path.getmtime, reverse=True)[:4]
                     
-                    # If no images were found, generate placeholder images
-                    if not image_paths:
-                        placeholder_paths = []
-                        for i, desc in enumerate(result.split('\n')):
-                            if desc.startswith('- '):
-                                img = generate_simple_image(desc[2:])
-                                path = os.path.join(tempfile.gettempdir(), f"placeholder_{i}.png")
-                                img.save(path)
-                                placeholder_paths.append(path)
-                        image_paths = placeholder_paths[:4]
-                    
                     # Return the result and image paths
                     return result, image_paths
                 except ValueError:
@@ -238,16 +152,13 @@ class CustomGradioUI(GradioUI):
 
 # Modified prompt templates to handle simple age input
 custom_prompt_templates = {
-    "react-system": """You are a helpful agent that helps children learn about Jewish Passover/Pesach through coloring pages.
+    "react-system": """You are an helpful agent that helps children learn about Passover through coloring pages.
 You have access to the following tools:
 
 {tools_string}
 
-When a user enters a number between 1 and 12, interpret it as the age of a child and generate Jewish Passover coloring pages suitable for that age using the generate_passover_coloring_pages tool.
-If you're asked to generate coloring pages or if the input is just a number, assume the user wants Jewish Passover coloring pages for a child of that age.
-
-Important: Only generate authentic Jewish Passover (Pesach) symbols and imagery - including matzah, seder plate, haggadah, afikoman, four cups of wine, the exodus story, etc.
-Do NOT include any Christian symbols or imagery like crosses, Jesus, or other non-Jewish religious elements.
+When a user enters a number between 1 and 12, interpret it as the age of a child and generate Passover coloring pages suitable for that age using the generate_passover_coloring_pages tool.
+If you're asked to generate coloring pages or if the input is just a number, assume the user wants Passover coloring pages for a child of that age.
 
 To use a tool, output an action in this format:
 ```py