Update app.py

app.py

@@ -10,11 +10,13 @@ from Gradio_UI import GradioUI
 
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 @tool
-def mutante_unicorn(theme:str) -> str: #it's import to specify the return type
+def mutante_unicorn(animal_name: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 close-up photograph of a mutant unicorn based from the name of any animal.
+    """A tool that generates a close-up photograph of a mutant unicorn based on an animal name (e.g. tiger, duck, snake, etc).
+       This mutant unicorn has butterfly wings and is 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. 
     Args:
-        theme: A string representing a name of any animal.
+        animal_name: A string representing a name of any animal.
     """
     try:
         # Import tool from Hub
@@ -25,15 +27,15 @@ def mutante_unicorn(theme:str) -> str: #it's import to specify the return type
             description="Generate image from prompt"
         )
         '''
-        description = f'''
-        aa close-up photograph of a {theme} 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'''
+        #description = f'''
+        #aa close-up photograph of a {theme} 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'''
         image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
-        image = image_generation_tool(description)
+        image = image_generation_tool(animal_name)
         return final_answer(image)
     except Exception as e:
-        return f"Error creating card for theme '{theme}': {str(e)}"
+        return f"Error creating card for theme '{animal_name}': {str(e)}"
 
 #image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)