Update app.py

app.py

@@ -11,62 +11,68 @@ from Gradio_UI import GradioUI
 
 @tool
 def generate_passover_coloring_pages(age: int) -> str:
-    """Generates Passover-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:
-        str: A list of suggested Passover-themed coloring pages or an error message.
+        str: A list of suggested Passover-themed coloring pages.
     """
-    
+
     if not isinstance(age, int) or not (1 <= age <= 12):
         return "Error: Please provide a valid age between 1 and 12."
-    
+
     complexity_levels = {
         (1, 3): "simple",
         (4, 6): "basic",
         (7, 9): "intermediate",
         (10, 12): "advanced"
     }
-    
     complexity = next(value for (low, high), value in complexity_levels.items() if low <= age <= high)
-    
+
     passover_pages = {
         "simple": ["Simple matzah", "Basic Seder plate", "Moses in a basket", "Simple frog (plague)"],
         "basic": ["Parting of the Red Sea", "Ten Plagues scenes", "Four Questions illustrations", "Afikoman hunt"],
         "intermediate": ["Detailed Seder table", "Exodus story scenes", "Hebrew letter decorations", "Egyptian pyramids"],
         "advanced": ["Intricate Haggadah illustrations", "Detailed symbolism scene", "Combined Passover story panels", "Hebrew calligraphy art"]
     }
-    
+
     selected_pages = passover_pages[complexity]
-    
+
+    # Попытка сгенерировать изображения
+    try:
+        image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
+        images = [image_generation_tool(page) for page in selected_pages]
+        return final_answer(images)
+    except Exception as e:
+        return f"Error generating images: {str(e)}"
+
     response = f"Passover Coloring Pages for a {age}-year-old:\n\n"
     response += "\n".join(f"- {page}" for page in selected_pages)
     
     return response
 
-# Initialize FinalAnswerTool
+# Настройка агента
 final_answer = FinalAnswerTool()
 
-# Hugging Face Model Configuration
 model = HfApiModel(
     max_tokens=2096,
     temperature=0.5,
     model_id='Qwen/Qwen2.5-Coder-32B-Instruct'
 )
 
-# Load prompt templates
 with open("prompts.yaml", 'r') as stream:
     prompt_templates = yaml.safe_load(stream)
-
-# Initialize the agent with registered tools
+    
 agent = CodeAgent(
     model=model,
     tools=[final_answer, generate_passover_coloring_pages],
     max_steps=6,
-    verbosity_level=1
+    verbosity_level=1,
+    prompt_templates=prompt_templates
 )
 
-# Launch the UI
 GradioUI(agent).launch()
+