Update app.py

app.py

@@ -1,6 +1,5 @@
 import gradio as gr
 from openai import OpenAI
-import os
 import httpx
 
 # List of popular styles
@@ -48,6 +47,12 @@ def generate_image(api_key, prompt, style, negative_prompt):
     
     return response.data[0].url, enhanced_prompt
 
+def process_and_generate(api_key, prompt, style, negative_prompt):
+    client = OpenAI(api_key=api_key, http_client=httpx.Client())
+    enhanced_prompt = enhance_prompt(client, prompt, style)
+    image_url, _ = generate_image(api_key, enhanced_prompt, style, negative_prompt)
+    return image_url, enhanced_prompt
+
 with gr.Blocks() as demo:
     gr.Markdown("# DALL-E 3 Image Generator")
     
@@ -64,7 +69,7 @@ with gr.Blocks() as demo:
             enhanced_prompt_output = gr.Textbox(label="Enhanced Prompt")
     
     generate_btn.click(
-        generate_image,
+        process_and_generate,
         inputs=[api_key, prompt, style, negative_prompt],
         outputs=[image_output, enhanced_prompt_output]
     )