Update app.py

app.py

@@ -8,11 +8,11 @@ import tempfile
 import os
 import time
 import shutil
-from PIL import Image
+from PIL import Image, ImageDraw, ImageFont
 from tools.final_answer import FinalAnswerTool
 from Gradio_UI import GradioUI
 
-# Создаем постоянную директорию для хранения изображений
+# Создаем постоянную директорию для хранения изображений и HTML
 IMAGES_DIR = os.path.join(os.path.dirname(os.path.abspath(__file__)), "coloring_images")
 os.makedirs(IMAGES_DIR, exist_ok=True)
 
@@ -68,73 +68,133 @@ def generate_passover_coloring_pages(age: int) -> str:
     selected_pages = passover_pages[complexity]
     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
-    
-    # Создаем заглушки для изображений на случай сбоя в генерации
+    # Create placeholder images function
     def create_placeholder_image(description, index):
         img = Image.new('RGB', (800, 600), color='white')
+        draw = ImageDraw.Draw(img)
+        draw.rectangle([(10, 10), (790, 590)], outline='black', width=2)
+        
+        # Add text
+        draw.text((400, 300), description.split(',')[0] if ',' in description else description, 
+                 fill='black', anchor='mm')
+        draw.text((400, 350), "Jewish Passover Coloring Page", fill='black', anchor='mm')
+        draw.text((400, 400), f"For age: {age}", fill='black', anchor='mm')
+        
         return img
     
-    # Попытка сгенерировать изображения
+    # Generate and save images
+    image_paths = []
+    actual_files = []
+    timestamp = int(time.time())
+    
     try:
         image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
         
-        # Используем улучшенные подсказки для раскрасок
-        enhanced_prompts = []
-        for page in selected_pages:
-            enhanced_prompts.append(f"{page}, Jewish traditional Passover symbols only, no Christian imagery, suitable for printing as a coloring page")
-        
-        # Сохраняем пути к изображениям
-        image_paths = []
-        
-        # Генерируем и сохраняем изображения
-        for i, prompt in enumerate(enhanced_prompts):
+        # Generate images
+        for i, (prompt, desc) in enumerate(zip(selected_pages, descriptions)):
             try:
-                img = image_generation_tool(prompt)
-                # Сохраняем в постоянную директорию с уникальным именем
-                timestamp = int(time.time())
-                img_path = os.path.join(IMAGES_DIR, f"passover_{age}_{i}_{timestamp}.png")
+                img = image_generation_tool(f"{prompt}, Jewish traditional Passover symbols only, no Christian imagery, suitable for printing as a coloring page")
+                img_filename = f"passover_{age}_{i}_{timestamp}.png"
+                img_path = os.path.join(IMAGES_DIR, img_filename)
                 img.save(img_path)
                 image_paths.append(img_path)
+                actual_files.append((img_filename, desc))
+                
             except Exception as e:
-                # При ошибке создаем заглушку
-                placeholder = create_placeholder_image(descriptions[i], i)
-                img_path = os.path.join(IMAGES_DIR, f"passover_{age}_{i}_placeholder.png")
+                # Create placeholder if image generation fails
+                placeholder = create_placeholder_image(desc, i)
+                img_filename = f"passover_{age}_{i}_placeholder_{timestamp}.png"
+                img_path = os.path.join(IMAGES_DIR, img_filename)
                 placeholder.save(img_path)
                 image_paths.append(img_path)
+                actual_files.append((img_filename, desc))
                 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])
-        
-        # Сохраняем пути к изображениям в ответе в специальном формате
-        image_paths_str = "||".join(image_paths)
-        return f"{response}\n\nIMAGE_PATHS:{image_paths_str}"
     
     except Exception as 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 = []
+        # If all image generation fails, create placeholders
         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")
+            img_filename = f"passover_{age}_{i}_placeholder_{timestamp}.png"
+            img_path = os.path.join(IMAGES_DIR, img_filename)
             placeholder.save(img_path)
             image_paths.append(img_path)
+            actual_files.append((img_filename, desc))
+    
+    # Create a printable HTML file
+    html_filename = f"passover_coloring_pages_{age}_{timestamp}.html"
+    html_path = os.path.join(IMAGES_DIR, html_filename)
+    
+    with open(html_path, 'w') as f:
+        f.write(f"""<!DOCTYPE html>
+<html>
+<head>
+    <title>Jewish Passover Coloring Pages - Age {age}</title>
+    <style>
+        body {{ font-family: Arial, sans-serif; max-width: 900px; margin: 0 auto; padding: 20px; }}
+        h1, h2 {{ color: #2c3e50; text-align: center; }}
+        .controls {{ background-color: #f8f9fa; padding: 15px; margin-bottom: 20px; text-align: center; border-radius: 5px; }}
+        .page {{ page-break-after: always; margin-bottom: 30px; }}
+        img {{ max-width: 100%; display: block; margin: 0 auto; border: 1px solid #ddd; }}
+        .page-title {{ text-align: center; margin-top: 10px; font-weight: bold; }}
+        .print-button {{ background-color: #3498db; color: white; border: none; padding: 10px 20px; 
+                        border-radius: 4px; cursor: pointer; font-size: 16px; }}
+        .print-button:hover {{ background-color: #2980b9; }}
+        @media print {{
+            .controls {{ display: none; }}
+            .page {{ margin-bottom: 0; }}
+            body {{ padding: 0; margin: 0; }}
+        }}
+    </style>
+</head>
+<body>
+    <div class="controls">
+        <h1>Jewish Passover (Pesach) Coloring Pages</h1>
+        <h2>For Ages {age} Years</h2>
+        <p>Click the button below to print all coloring pages.</p>
+        <button class="print-button" onclick="window.print()">Print All Pages</button>
+    </div>
+""")
+        
+        # Add cover page
+        f.write(f"""
+    <div class="page">
+        <h1 style="margin-top: 100px;">Jewish Passover (Pesach)</h1>
+        <h1>Coloring Pages</h1>
+        <h2>For Ages {age} Years</h2>
+        <div style="text-align: center; margin-top: 50px;">
+            <p>Created for Jewish children and families</p>
+            <p>to celebrate the story of Passover</p>
+        </div>
+    </div>
+""")
         
-        # Сохраняем пути к заглушкам
-        image_paths_str = "||".join(image_paths)
-        return f"{response}\n\nIMAGE_PATHS:{image_paths_str}"
+        # Add each coloring page
+        for filename, desc in actual_files:
+            f.write(f"""
+    <div class="page">
+        <img src="{filename}" alt="{desc}">
+        <div class="page-title">{desc}</div>
+    </div>
+""")
+        
+        f.write("""
+</body>
+</html>
+""")
+    
+    # Create formatted response
+    response = f"## Jewish Passover (Pesach) Coloring Pages for a {age}-year-old:\n\n"
+    response += "\n".join([f"- {desc}" for desc in descriptions])
+    
+    # Add information about the printable HTML file
+    response += f"\n\n**Coloring pages have been generated!**\n\n"
+    response += f"To print the coloring pages, please open this file in your web browser:\n\n"
+    response += f"{html_path}\n\n"
+    response += f"The individual image files are also available in the 'coloring_images' folder."
+    
+    # Add image paths for the gallery display attempt
+    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):
@@ -178,16 +238,16 @@ class CustomGradioUI(GradioUI):
                     if "IMAGE_PATHS:" in result:
                         paths_part = result.split("IMAGE_PATHS:")[1]
                         image_paths = paths_part.split("||")
-                        # Удаляем маркер путей из результата
+                        # Remove the paths marker from the result
                         result = result.split("\n\nIMAGE_PATHS:")[0]
                     
-                    # Если нет путей к изображениям, ищем в директории
+                    # If no paths found, look in the directory
                     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))
                     
-                    # Проверяем, что пути ведут к реальным файлам
+                    # Check that paths lead to real files
                     valid_image_paths = [path for path in image_paths if os.path.exists(path)]
                     
                     # Return the result and image paths