Update app.py

app.py

@@ -9,71 +9,43 @@ from tools.final_answer import FinalAnswerTool
 from Gradio_UI import GradioUI
 
 
-
 @tool
-def generate_passover_coloring_pages(age: int, gender: str) -> str:
-    """Generates Passover-themed coloring pages based on a child's age and gender.
+def generate_passover_coloring_pages(age: int) -> str:
+    """Generates Passover-themed coloring pages based on a child's age.
+    
     Args:
         age (int): The age of the child (1-12).
-        gender (str): The gender of the child ('boy' or 'girl').
+    
     Returns:
         str: A list of suggested Passover-themed coloring pages or an error message.
     """
-
-    # Validate inputs
+    
     if not isinstance(age, int) or not (1 <= age <= 12):
         return "Error: Please provide a valid age between 1 and 12."
-
-    gender = gender.lower()
-    if gender not in ['boy', 'girl']:
-        return "Error: Please specify gender as 'boy' or 'girl'."
-
-    # Determine complexity level based on age
+    
     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)
-
-    # Define Passover-themed coloring pages
+    
     passover_pages = {
-        "simple": {
-            "boy": ["Simple matzah", "Basic Seder plate", "Moses in a basket", "Simple frog (plague)"],
-            "girl": ["Simple matzah", "Basic Seder plate", "Moses in a basket", "Miriam with tambourine"]
-        },
-        "basic": {
-            "boy": ["Parting of the Red Sea", "Ten Plagues scenes", "Four Questions illustrations", "Afikoman hunt"],
-            "girl": ["Parting of the Red Sea", "Ten Plagues scenes", "Four Questions illustrations", "Miriam dancing"]
-        },
-        "intermediate": {
-            "boy": ["Detailed Seder table", "Exodus story scenes", "Hebrew letter decorations", "Egyptian pyramids"],
-            "girl": ["Detailed Seder table", "Exodus story scenes", "Hebrew letter decorations", "Family Seder celebration"]
-        },
-        "advanced": {
-            "boy": ["Intricate Haggadah illustrations", "Detailed symbolism scene", "Combined Passover story panels", "Hebrew calligraphy art"],
-            "girl": ["Intricate Haggadah illustrations", "Detailed symbolism scene", "Combined Passover story panels", "Hebrew calligraphy art"]
-        }
+        "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][gender]
-
-    try:
-        # Load the image generation tool
-        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)}"
-
-    # Fallback text response
-    response = f"Passover Coloring Pages for a {age}-year-old {gender}:\n\n"
+    
+    selected_pages = passover_pages[complexity]
+    
+    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()
 
@@ -91,7 +63,7 @@ with open("prompts.yaml", 'r') as stream:
 # Initialize the agent with registered tools
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, generate_passover_coloring_pages],  # Ensure generate_passover_coloring_pages is a tool
+    tools=[final_answer, generate_passover_coloring_pages],
     max_steps=6,
     verbosity_level=1
 )