Update app.py

app.py

@@ -25,7 +25,6 @@ def call_api(negative_prompt, positive_prompt, api_key):
     # Check if the request was successful
     if response.status_code == 200:
         data = response.json()['result'][0]['data']
-        format = response.json()['result'][0]['format']
         
         # Decode the base64 string to bytes
         image_data = base64.b64decode(data)
@@ -33,7 +32,13 @@ def call_api(negative_prompt, positive_prompt, api_key):
         # Convert bytes data to an image
         image = Image.open(BytesIO(image_data))
         
-        return image
+        # Save the image as PNG to a buffer
+        buffer = BytesIO()
+        image.save(buffer, format="PNG")
+        
+        # Return the buffer content to display it directly in the interface
+        buffer.seek(0)
+        return buffer
     else:
         return "Failed to fetch the image. Please check the API key and inputs."