Update app.py

app.py

@@ -56,7 +56,7 @@ 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/ultra"
+    url = "https://api.stability.ai/v2beta/stable-image/generate/sd3"
     
     headers = {
         "Accept": "image/*",
@@ -66,7 +66,12 @@ def generate_image(stability_api_key, enhanced_prompt, style, negative_prompt):
     data = {
         "prompt": f"{enhanced_prompt}, Style: {style}",
         "negative_prompt": negative_prompt,
-        "output_format": "webp"
+        "model": "sd3.5-large-turbo",
+        "output_format": "jpeg",
+        "width": 1024,
+        "height": 1024,
+        "num_images": 1,
+        "steps": 4,  # SD3.5 Large Turbo generates high-quality images in just 4 steps
     }
     
     try:
@@ -93,23 +98,23 @@ def process_and_generate(google_api_key, stability_api_key, prompt, style, negat
         image_bytes = generate_image(stability_api_key, enhanced_prompt, style, negative_prompt)
         
         # Save image to a file
-        with open("generated_image.webp", "wb") as f:
+        with open("generated_image.jpeg", "wb") as f:
             f.write(image_bytes)
         
         # Return the file path and enhanced prompt
-        return "generated_image.webp", enhanced_prompt
+        return "generated_image.jpeg", enhanced_prompt
     except Exception as e:
         logging.error(f"Error in process_and_generate: {str(e)}")
         return str(e), enhanced_prompt if 'enhanced_prompt' in locals() else "Error occurred before prompt enhancement"
 
 with gr.Blocks() as demo:
-    gr.Markdown("# Stability AI Ultra Image Generator with Google Gemini Prompt Enhancement")
+    gr.Markdown("# Stability AI SD3.5 Large Turbo Image Generator with Google Gemini Prompt Enhancement")
     
     with gr.Row():
         with gr.Column(scale=1):
             google_api_key = gr.Textbox(label="Google AI API Key", type="password")
             stability_api_key = gr.Textbox(label="Stability AI API Key", type="password")
-            prompt = gr.Textbox(label="Describe Your Image")
+            prompt = gr.Textbox(label="Prompt")
             style = gr.Dropdown(label="Style", choices=STYLES)
             negative_prompt = gr.Textbox(label="Negative Prompt", value=DEFAULT_NEGATIVE_PROMPT)
             submit_btn = gr.Button("Generate Image")