Update app.py

convert to PIL image in the comic generator

app.py

@@ -1,5 +1,7 @@
 from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool
 import datetime
+import os
+from PIL import Image
 import requests
 import pytz
 import yaml
@@ -33,20 +35,34 @@ def get_current_time_in_timezone(timezone: str) -> str:
     except Exception as e:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
+
+
 @tool
 def comic_strip_generator(scene_description: str) -> str:
-    """Generates a comic strip from a scene description using image generation.
+    """Generates a comic strip image from a scene description using image generation.
     Args:
         scene_description: A creative, descriptive text outlining a scene or story.
     """
     try:
-        # Call the existing image generation tool with the input
-        result = image_generation_tool(scene_description)
-        return f"Here's your comic scene: {result}"
+        image = image_generation_tool(scene_description)
+        # Generate a unique filename
+        filename = f"comic_{datetime.now().strftime('%Y%m%d_%H%M%S')}.jpg"
+        filepath = os.path.join("generated_comics", filename)
+        
+        # Ensure the output directory exists
+        os.makedirs("generated_comics", exist_ok=True)
+
+        # Save image if it's a PIL image
+        if isinstance(image, Image.Image):
+            image.save(filepath)
+            return f"Here's your comic scene saved at: {filepath}"
+        else:
+            return f"Unexpected image output: {type(image)}"
     except Exception as e:
         return f"Error generating comic: {str(e)}"
 
 
+
 final_answer = FinalAnswerTool()
 
 # If the agent does not answer, the model is overloaded, please use another model or the following Hugging Face Endpoint that also contains qwen2.5 coder: