Update app.py

app.py

@@ -18,13 +18,16 @@ def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return
     """
     return "What magic will you build ?" 
 
-@tool 
-def generate_image(prompt : str) -> str : 
-    """ genere une image a partir d'une description textuelle .
-    Args : 
-    prompt : une description de l'image a générer 
-    returns : URL ou chemin de l'image générer """
+@tool
+def generate_image(prompt: str) -> str:
+    """Génère une image à partir d'une description textuelle.
+
+    Args:
+        prompt (str): Une description de l'image à générer.
 
+    Returns:
+        str: URL ou chemin de l'image générée.
+    """
     return image_generation_tool(prompt)