Update app.py

app.py

@@ -4,7 +4,6 @@ import requests
 from PIL import Image
 import io
 import logging
-import base64
 
 # Set up logging
 logging.basicConfig(level=logging.DEBUG, format='%(asctime)s - %(levelname)s - %(message)s')
@@ -59,30 +58,24 @@ def enhance_prompt(google_api_key, prompt, style):
         raise
 
 def generate_image(stability_api_key, enhanced_prompt, style, negative_prompt):
-    url = "https://api.stability.ai/v1/generation/stable-diffusion-v1-5/text-to-image"
+    url = "https://api.stability.ai/v2beta/stable-image/generate/sd3"
     
     headers = {
-        "Content-Type": "application/json",
-        "Accept": "application/json",
+        "Accept": "image/jpeg",
         "Authorization": f"Bearer {stability_api_key}"
     }
     
     payload = {
-        "text_prompts": [
-            {
-                "text": f"{enhanced_prompt}, Style: {style}",
-                "weight": 1
-            },
-            {
-                "text": negative_prompt,
-                "weight": -1
-            }
-        ],
-        "cfg_scale": 7,
-        "height": 512,
-        "width": 512,
-        "samples": 1,
-        "steps": 30
+        "prompt": f"{enhanced_prompt}, Style: {style}",
+        "negative_prompt": negative_prompt,
+        "model": "sd3.5-large-turbo",
+        "width": 1024,
+        "height": 1024,
+        "num_images": 1,
+        "steps": 30,
+        "cfg_scale": 7.5,
+        "seed": 0,
+        "output_format": "jpeg"
     }
     
     try:
@@ -92,13 +85,8 @@ def generate_image(stability_api_key, enhanced_prompt, style, negative_prompt):
         logging.debug(f"Response headers: {response.headers}")
         logging.debug(f"Response content type: {response.headers.get('content-type')}")
         
-        if response.headers.get('content-type') == 'application/json':
-            result = response.json()
-            if 'artifacts' in result and len(result['artifacts']) > 0:
-                image_data = result['artifacts'][0]['base64']
-                return base64.b64decode(image_data)
-            else:
-                raise Exception("No image data found in the response")
+        if response.headers.get('content-type') == 'image/jpeg':
+            return response.content
         else:
             error_message = response.text
             logging.error(f"Unexpected content type: {response.headers.get('content-type')}. Response: {error_message}")