Update app.py

app.py

@@ -4,6 +4,7 @@ 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')
@@ -58,35 +59,46 @@ 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/v2beta/stable-image/generate/sd3"
+    url = "https://api.stability.ai/v1/generation/stable-diffusion-v1-5/text-to-image"
     
     headers = {
-        "Accept": "image/jpeg",
+        "Content-Type": "application/json",
+        "Accept": "application/json",
         "Authorization": f"Bearer {stability_api_key}"
     }
     
-    files = {
-        "prompt": (None, f"{enhanced_prompt}, Style: {style}"),
-        "negative_prompt": (None, negative_prompt),
-        "model": (None, "sd3.5-large-turbo"),
-        "width": (None, "1024"),
-        "height": (None, "1024"),
-        "num_images": (None, "1"),
-        "steps": (None, "30"),
-        "cfg_scale": (None, "7.5"),
-        "seed": (None, "0"),
-        "output_format": (None, "jpeg")
+    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
     }
     
     try:
-        response = requests.post(url, headers=headers, files=files)
+        response = requests.post(url, headers=headers, json=payload)
         response.raise_for_status()
         
         logging.debug(f"Response headers: {response.headers}")
         logging.debug(f"Response content type: {response.headers.get('content-type')}")
         
-        if response.headers.get('content-type') == 'image/jpeg':
-            return response.content
+        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")
         else:
             error_message = response.text
             logging.error(f"Unexpected content type: {response.headers.get('content-type')}. Response: {error_message}")