Update app.py

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 import google.generativeai as genai
 import requests
-import base64
+import io
 
 # List of popular styles
 STYLES = [
@@ -51,29 +51,27 @@ def generate_image(stability_api_key, enhanced_prompt, style, negative_prompt):
     url = "https://api.stability.ai/v2beta/stable-image/generate/sd3"
     
     headers = {
-        "Accept": "application/json",
-        "Content-Type": "application/json",
+        "Accept": "image/*",
         "Authorization": f"Bearer {stability_api_key}"
     }
     
-    payload = {
-        "model": "sd3.5-large-turbo",
+    data = {
         "prompt": f"{enhanced_prompt}, Style: {style}",
         "negative_prompt": negative_prompt,
-        "width": 1024,
-        "height": 1024,
-        "num_images": 1,
-        "steps": 30,
-        "cfg_scale": 7.5,
-        "seed": 0,
+        "model": "sd3.5-large-turbo",
+        "width": "1024",
+        "height": "1024",
+        "num_images": "1",
+        "steps": "30",
+        "cfg_scale": "7.5",
+        "seed": "0",
         "output_format": "png"
     }
     
-    response = requests.post(url, json=payload, headers=headers)
+    response = requests.post(url, headers=headers, data=data)
     
     if response.status_code == 200:
-        image_data = response.json()['images'][0]['image_base64']
-        return base64.b64decode(image_data)
+        return response.content
     else:
         raise Exception(f"Image generation failed: {response.text}")