Update app.py

app.py

@@ -113,29 +113,22 @@ def generate_image(stability_api_key, enhanced_prompt, style, negative_prompt):
     }
     
     data = {
-        "prompt": f"{enhanced_prompt}, Style: {style}, highly detailed, photorealistic, high quality, natural lighting, realistic textures and colors",
+        "prompt": f"{enhanced_prompt}, Style: {style}, highly detailed, photorealistic, high quality",
         "negative_prompt": negative_prompt,
         "model": "sd3.5-large-turbo",
         "output_format": "jpeg",
         "width": 1024,
         "height": 1024,
         "num_images": 1,
-        "steps": 30,  # Increased from 20 to 30
-        "cfg_scale": 7,
+        "steps": 4,  # SD3.5 Large Turbo generates high-quality images in just 4 steps
     }
     
     try:
-        response = requests.post(url, headers=headers, files={"none": ''}, data=data, timeout=60)  # Added 60-second timeout
+        response = requests.post(url, headers=headers, files={"none": ''}, data=data)
         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').startswith('image/'):
-            image_data = response.content
-            if len(image_data) < 1000:  # Check if the image data is too small (likely incomplete)
-                raise Exception("Received incomplete image data")
-            return image_data
+            return response.content
         else:
             error_message = response.text
             logging.error(f"Unexpected content type: {response.headers.get('content-type')}. Response: {error_message}")