Update app.py

app.py

@@ -4,6 +4,7 @@ import spaces
 import torch
 import random
 import os
+import tempfile
 from PIL import Image
 
 # Import the pipeline from diffusers
@@ -76,8 +77,12 @@ def chat_fn(message, chat_history, seed, randomize_seed, guidance_scale, steps,
     pipe = pipe.to("cpu")
     torch.cuda.empty_cache()
     
-    # Return the PIL Image directly - ChatInterface will handle it properly
-    return image
+    # Save image to temporary file and return the path
+    with tempfile.NamedTemporaryFile(delete=False, suffix=".png") as tmp_file:
+        image.save(tmp_file.name, format="PNG")
+        temp_path = tmp_file.name
+    
+    return temp_path
 
 # --- UI Definition using gr.ChatInterface ---