Update app.py

app.py

@@ -34,34 +34,23 @@ def get_current_time_in_timezone(timezone: str) -> str:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
 @tool
-def image_generation_tool():
-    """Construct a pure image generation Tool derivation.
-    
-    Returns a Tool instance configured for the FLUX.1-schnell image generation model,
-    maintaining referential transparency through immutable configuration.
-    The resulting Tool acts as a pure function from prompts to images.
-    
-    Tool Configuration:
-        space: "black-forest-labs/FLUX.1-schnell"
-        name: "image_generator" 
-        description: "Generate an image from a prompt"
-    
-    Returns:
-        Tool: An immutable Tool instance that maps text prompts to generated images.
-            The returned Tool maintains purity by producing deterministic outputs
-            for given inputs within the constraints of the underlying model.
-    
-    Note:
-        While the Tool construction is pure, the actual image generation may involve
-        non-deterministic elements from the underlying model. The function itself
-        remains referentially transparent as it consistently returns the same Tool
-        configuration.
-    """
-    return Tool.from_space(
-        "black-forest-labs/FLUX.1-schnell",
-        name="image_generator",
-        description="Generate an image from a prompt"
-    )
+def image_generation_tool(prompt: str) -> str:
+   """A tool that generates images using the FLUX.1-schnell model.
+   Args:
+       prompt: A string containing the image generation prompt.
+   Returns:
+       str: Status message indicating success or error of image generation.
+   """
+   try:
+       tool = Tool.from_space(
+           "black-forest-labs/FLUX.1-schnell",
+           name="image_generator",
+           description="Generate an image from a prompt"
+       )
+       result = tool.generate(prompt)
+       return f"Successfully generated image from prompt: {prompt}"
+   except Exception as e:
+       return f"Error generating image from prompt '{prompt}': {str(e)}"
 
 final_answer = FinalAnswerTool()