Update app.py

app.py

@@ -1,6 +1,8 @@
 import gradio as gr
 import google.generativeai as genai
 import requests
+from PIL import Image
+import io
 
 # List of popular styles
 STYLES = [
@@ -76,8 +78,12 @@ def generate_image(stability_api_key, enhanced_prompt, style, negative_prompt):
 
 def process_and_generate(google_api_key, stability_api_key, prompt, style, negative_prompt):
     enhanced_prompt = enhance_prompt(google_api_key, prompt, style)
-    image_data = generate_image(stability_api_key, enhanced_prompt, style, negative_prompt)
-    return image_data, enhanced_prompt
+    image_bytes = generate_image(stability_api_key, enhanced_prompt, style, negative_prompt)
+    
+    # Convert bytes to PIL Image
+    image = Image.open(io.BytesIO(image_bytes))
+    
+    return image, enhanced_prompt
 
 with gr.Blocks() as demo:
     gr.Markdown("# Stability AI Image Generator with Google Gemini Prompt Enhancement")