Upload 4 files

app.py

@@ -279,12 +279,31 @@ async def generate(request: GenerateRequest):
             seed=seed,
         )
         
-        # Convert to PIL Image
-        if hasattr(output, 'data'):
-            # Output is raw image data
-            image = Image.frombytes('RGB', (width, height), output.data)
+        print(f"Output type: {type(output)}")
+        
+        # Handle different output formats from stable-diffusion-cpp
+        if isinstance(output, list):
+            # Returns list of images, take first one
+            img_data = output[0]
+        else:
+            img_data = output
+        
+        # Convert to PIL Image based on data type
+        if isinstance(img_data, bytes):
+            image = Image.open(io.BytesIO(img_data))
+        elif hasattr(img_data, 'data'):
+            # Raw pixel data
+            image = Image.frombytes('RGB', (width, height), img_data.data)
+        elif hasattr(img_data, 'tobytes'):
+            # NumPy array or similar
+            import numpy as np
+            arr = np.array(img_data)
+            image = Image.fromarray(arr.astype('uint8'))
+        elif isinstance(img_data, Image.Image):
+            image = img_data
         else:
-            image = output
+            # Try direct conversion
+            image = Image.fromarray(img_data)
         
         # Convert to base64
         buffer = io.BytesIO()