Update app.py

app.py

@@ -9,26 +9,47 @@ from Gradio_UI import GradioUI
 
 
 @tool
-def mutant_unicorn(animal_name:str) -> str:
+#def mutant_unicorn(animal_name:str) -> str:
+#    """
+#    A tool that generates an image of a mutant unicorn given an animal name.
+#    Args: 
+#        animal_name: A string containing the name of an animal, real or mytical. (e.g., 'lion, dragon, worm, etc')
+#    Returns:
+#        The generated image or an error message.
+#    """
+#    description = f"""Close-up photograph of a/an {animal_name} unicorn with butterfly wings holding 
+#    a small brightfull orb with its tiny paws, looking at the center of the orb with a 
+#    concentrated gaze. The  unicorn is in a clearing in the forest on a moonlit night
+#    """
+#    
+#    try:
+#        # Import tool from Hub
+#        image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
+#        image = image_generation_tool(description)
+#        return final_answer(image)
+#    except Exception as e:
+#        return f"Error creating mutant animal '{description}': {str(e)}"
+#
+def mutant_unicorn(theme:str) -> str: #it's import to specify the return type
+    #Keep this format for the description / args / args description but feel free to modify the tool
+    """A tool that generates a custom Valentine's Day card based on a given theme.
+    Args:
+        theme: A string representing the theme for the Valentine's Day card.
     """
-    A tool that generates an image of a mutant unicorn given an animal name.
-    Args:    
-        animal_name: A string containing the name of an animal, real or mytical. (e.g., 'lion, dragon, worm, etc')
-    Returns: 
-        The generated image or an error message.
-    """
-    description = f"""Close-up photograph of a/an {animal_name} unicorn with butterfly wings holding 
-    a small brightfull orb with its tiny paws, looking at the center of the orb with a 
-    concentrated gaze. The  unicorn is in a clearing in the forest on a moonlit night
-    """
-    
     try:
         # Import tool from Hub
+        '''
+        image_generation_tool = Tool.from_space(
+            space_id="black-forest-labs/FLUX.1-schnell",
+            name="generate-image",
+            description="Generate image from prompt"
+        )
+        '''
         image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
-        image = image_generation_tool(description)
+        image = image_generation_tool(theme)
         return final_answer(image)
     except Exception as e:
-        return f"Error creating mutant animal '{description}': {str(e)}"
+        return f"Error creating card for theme '{theme}': {str(e)}"
 
 final_answer = FinalAnswerTool()
 
@@ -42,10 +63,6 @@ model_id='Qwen/Qwen2.5-Coder-32B-Instruct',# it is possible that this model may
 custom_role_conversions=None,
 )
 
-
-# Import tool from Hub
-# image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
-
 with open("prompts.yaml", 'r') as stream:
     prompt_templates = yaml.safe_load(stream)