Update app.py

app.py

@@ -6,13 +6,20 @@ import yaml
 import re
 import tempfile
 import os
+import time
+import shutil
+from PIL import Image
 from tools.final_answer import FinalAnswerTool
 from Gradio_UI import GradioUI
 
+# Создаем постоянную директорию для хранения изображений
+IMAGES_DIR = os.path.join(os.path.dirname(os.path.abspath(__file__)), "coloring_images")
+os.makedirs(IMAGES_DIR, exist_ok=True)
+
 @tool
 def generate_passover_coloring_pages(age: int) -> str:
     """
-    Generates Passover-themed coloring pages based on a child's age.
+    Generates Jewish Passover/Pesach-themed coloring pages based on a child's age.
     Args:
         age (int): The age of the child (1-12).
     Returns:
@@ -30,68 +37,104 @@ 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 for better coloring pages
+    # Enhanced prompts specifically for Jewish Passover/Pesach themes
     passover_pages = {
         "simple": [
-            "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"
+            "Line drawing of matzah for Jewish Passover coloring page, thick black outlines, white background, child-friendly",
+            "Simple black outline of a Seder plate with spaces for the six items (karpas, charoset, maror, zeroa, beitzah, chazeret), coloring book style, thick lines",
+            "Simple outline drawing of the Afikoman (broken piece of matzah) hidden in fabric, black and white for coloring",
+            "Simple outline of a Kiddush cup for Passover seder, black and white for coloring"
         ],
         "basic": [
-            "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"
+            "Black and white outline of Moses parting the Red Sea, traditional Jewish Pesach story, no Christian symbols, coloring book style",
+            "Outline drawing of the Ten Plagues (blood, frogs, lice, wild animals, cattle disease, boils, hail, locusts, darkness, death of firstborn) in separated panels, Jewish Passover theme",
+            "Haggadah Four Questions illustrations in Hebrew and English for coloring, Jewish Passover tradition",
+            "Afikoman hunt maze with matzah at the center, black and white outline for Jewish Passover coloring"
         ],
         "intermediate": [
-            "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"
+            "Detailed black and white outline of a traditional Jewish Seder table with all symbolic foods and Haggadah, no crosses or Christian symbols",
+            "Hebrew letter decorations spelling 'Pesach' (פסח) for coloring, traditional Jewish style",
+            "Outline drawing of ancient Israelites leaving Egypt, Jewish Passover story, historically accurate depiction",
+            "Black and white outline of the Four Sons from the Haggadah, traditional Jewish Passover story characters"
         ],
         "advanced": [
-            "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"
+            "Intricate Haggadah page illustrations in black and white, showing the order of the Seder, traditional Jewish style",
+            "Detailed scene of Miriam and women celebrating after crossing the Red Sea with tambourines, Jewish Passover story",
+            "Intricate Hebrew calligraphy spelling 'Next year in Jerusalem' (לשנה הבאה בירושלים) for coloring",
+            "Detailed symbolic Jewish Passover illustrations showing the journey from slavery to freedom, with proper Jewish symbols only"
         ]
     }
     
     selected_pages = passover_pages[complexity]
-    descriptions = [f"- {page.split(',')[0]}" for page in selected_pages]
+    descriptions = [page.split(',')[0] if ',' in page else page for page in selected_pages]
+    
+    # Очищаем директорию от старых изображений
+    for file in os.listdir(IMAGES_DIR):
+        if file.endswith(".png"):
+            file_path = os.path.join(IMAGES_DIR, file)
+            try:
+                os.remove(file_path)
+            except:
+                pass
+    
+    # Создаем заглушки для изображений на случай сбоя в генерации
+    def create_placeholder_image(description, index):
+        img = Image.new('RGB', (800, 600), color='white')
+        return img
     
-    # Try to generate images with improved prompts
+    # Попытка сгенерировать изображения
     try:
         image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
         
-        # 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")
+            enhanced_prompts.append(f"{page}, Jewish traditional Passover symbols only, no Christian imagery, suitable for printing as a coloring page")
         
-        images = [image_generation_tool(prompt) for prompt in enhanced_prompts]
-        
-        # Save images to temporary files
+        # Сохраняем пути к изображениям
         image_paths = []
-        for i, img in enumerate(images):
-            # Create a temporary file with a .png extension
-            with tempfile.NamedTemporaryFile(suffix='.png', delete=False) as temp_file:
-                temp_path = temp_file.name
-                img.save(temp_path)
-                image_paths.append(temp_path)
         
-        # Create a formatted response with descriptions
-        response = f"## Passover Coloring Pages for a {age}-year-old:\n\n"
-        response += "\n".join(descriptions)
+        # Генерируем и сохраняем изображения
+        for i, prompt in enumerate(enhanced_prompts):
+            try:
+                img = image_generation_tool(prompt)
+                # Сохраняем в постоянную директорию с уникальным именем
+                timestamp = int(time.time())
+                img_path = os.path.join(IMAGES_DIR, f"passover_{age}_{i}_{timestamp}.png")
+                img.save(img_path)
+                image_paths.append(img_path)
+            except Exception as e:
+                # При ошибке создаем заглушку
+                placeholder = create_placeholder_image(descriptions[i], i)
+                img_path = os.path.join(IMAGES_DIR, f"passover_{age}_{i}_placeholder.png")
+                placeholder.save(img_path)
+                image_paths.append(img_path)
+                print(f"Error generating image {i}: {str(e)}")
+        
+        # Создаем форматированный ответ с описаниями
+        response = f"## Jewish Passover (Pesach) Coloring Pages for a {age}-year-old:\n\n"
+        response += "\n".join([f"- {desc}" for desc in descriptions])
         
-        # Return the response with image paths
-        return f"{response}\n\nImages are saved and ready for display."
+        # Сохраняем пути к изображениям в ответе в специальном формате
+        image_paths_str = "||".join(image_paths)
+        return f"{response}\n\nIMAGE_PATHS:{image_paths_str}"
+    
     except Exception as e:
-        # 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)}"
+        # Если генерация изображений полностью не удалась, вернуть только текстовое описание
+        response = f"## Jewish Passover (Pesach) Coloring Pages for a {age}-year-old:\n\n"
+        response += "\n".join([f"- {desc}" for desc in descriptions])
+        
+        # Создаем заглушки для каждого описания
+        image_paths = []
+        for i, desc in enumerate(descriptions):
+            placeholder = create_placeholder_image(desc, i)
+            img_path = os.path.join(IMAGES_DIR, f"passover_{age}_{i}_placeholder.png")
+            placeholder.save(img_path)
+            image_paths.append(img_path)
+        
+        # Сохраняем пути к заглушкам
+        image_paths_str = "||".join(image_paths)
+        return f"{response}\n\nIMAGE_PATHS:{image_paths_str}"
 
 # Custom UI wrapper to handle image display
 class CustomGradioUI(GradioUI):
@@ -105,7 +148,8 @@ class CustomGradioUI(GradioUI):
         with gr.Blocks() as demo:
             with gr.Row():
                 with gr.Column(scale=1):
-                    gr.Markdown("# Passover Coloring Pages Generator")
+                    gr.Markdown("# Jewish Passover (Pesach) Coloring Pages Generator")
+                    gr.Markdown("Generate traditional Jewish Passover coloring pages for children")
                 
             with gr.Row():
                 with gr.Column(scale=1):
@@ -117,7 +161,7 @@ class CustomGradioUI(GradioUI):
                     output_text = gr.Markdown(label="Generated Pages")
                     
             with gr.Row():
-                gallery = gr.Gallery(label="Coloring Pages", show_label=True)
+                gallery = gr.Gallery(label="Jewish Passover Coloring Pages", show_label=True)
                 
             def process_input(input_text):
                 try:
@@ -129,17 +173,25 @@ class CustomGradioUI(GradioUI):
                     # Generate the coloring pages
                     result = generate_passover_coloring_pages(age)
                     
-                    # Get the image paths from the temporary files
+                    # Extract image paths from the result
                     image_paths = []
-                    for file in os.listdir(tempfile.gettempdir()):
-                        if file.endswith('.png'):
-                            image_paths.append(os.path.join(tempfile.gettempdir(), file))
+                    if "IMAGE_PATHS:" in result:
+                        paths_part = result.split("IMAGE_PATHS:")[1]
+                        image_paths = paths_part.split("||")
+                        # Удаляем маркер путей из результата
+                        result = result.split("\n\nIMAGE_PATHS:")[0]
                     
-                    # Display the most recent images (last 4 if there are more)
-                    image_paths = sorted(image_paths, key=os.path.getmtime, reverse=True)[:4]
+                    # Если нет путей к изображениям, ищем в директории
+                    if not image_paths:
+                        for file in os.listdir(IMAGES_DIR):
+                            if file.endswith('.png') and f"passover_{age}_" in file:
+                                image_paths.append(os.path.join(IMAGES_DIR, file))
+                    
+                    # Проверяем, что пути ведут к реальным файлам
+                    valid_image_paths = [path for path in image_paths if os.path.exists(path)]
                     
                     # Return the result and image paths
-                    return result, image_paths
+                    return result, valid_image_paths
                 except ValueError:
                     return "Please enter a valid number for the child's age.", []
                 except Exception as e:
@@ -152,13 +204,16 @@ class CustomGradioUI(GradioUI):
 
 # Modified prompt templates to handle simple age input
 custom_prompt_templates = {
-    "react-system": """You are an helpful agent that helps children learn about Passover through coloring pages.
+    "react-system": """You are a helpful agent that helps children learn about Jewish Passover/Pesach 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 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.
+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.
 
 To use a tool, output an action in this format:
 ```py