Update app.py

ajuste de metodo custon

app.py

@@ -11,26 +11,23 @@ import os
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 @tool
 def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return type
-    """A tool that generates text or an image based on the provided arguments.
+    """A tool that generates text based on the provided arguments.
     Args:
         arg1: A string input to guide the generation.
-        arg2: An integer input to specify the type of generation (1 for text, 2 for image).
+        arg2: An integer input to specify the type of generation (1 for detailed description, 2 for summary).
     """
     if arg2 == 1:
-        # Generate text using LLM
+        # Generate detailed description using LLM
         prompt = f"Generate a detailed description based on the following input: {arg1}"
         response = model.generate(prompt)
         return response['choices'][0]['text']
     elif arg2 == 2:
-        # Generate image using image generation tool
-        prompt = f"Generate an image based on the following input: {arg1}"
-        try:
-            image_response = image_generation_tool(prompt)
-            return f"Image generated: {image_response['image_url']}"
-        except Exception as e:
-            return f"Failed to generate image. Error: {str(e)}"
+        # Generate summary using LLM
+        prompt = f"Generate a summary based on the following input: {arg1}"
+        response = model.generate(prompt)
+        return response['choices'][0]['text']
     else:
-        return "Invalid argument for arg2. Use 1 for text generation and 2 for image generation."
+        return "Invalid argument for arg2. Use 1 for detailed description and 2 for summary."
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str: