Update app.py

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 from openai import OpenAI
 import os
+import httpx
 
 # List of popular styles
 STYLES = [
@@ -10,6 +11,13 @@ STYLES = [
     "Art Nouveau", "Pop Art", "Minimalist"
 ]
 
+# Default negative prompt
+DEFAULT_NEGATIVE_PROMPT = """
+ugly, tiling, poorly drawn hands, poorly drawn feet, poorly drawn face, out of frame, 
+extra limbs, disfigured, deformed, body out of frame, bad anatomy, watermark, signature, 
+cut off, low contrast, underexposed, overexposed, bad art, beginner, amateur, distorted face
+"""
+
 def enhance_prompt(client, prompt, style):
     enhanced_prompt_request = f"Enhance the following prompt for DALL-E 3 image generation in the style of {style}. Make it more detailed and vivid, while keeping the original intent: '{prompt}'"
     
@@ -23,14 +31,16 @@ def enhance_prompt(client, prompt, style):
     
     return response.choices[0].message.content.strip()
 
-def generate_image(api_key, prompt, style):
-    client = OpenAI(api_key=api_key)
+def generate_image(api_key, prompt, style, negative_prompt):
+    client = OpenAI(api_key=api_key, http_client=httpx.Client())
     
     enhanced_prompt = enhance_prompt(client, prompt, style)
     
+    full_prompt = f"{enhanced_prompt}\nNegative prompt: {negative_prompt}"
+    
     response = client.images.generate(
         model="dall-e-3",
-        prompt=enhanced_prompt,
+        prompt=full_prompt,
         size="1024x1024",
         quality="standard",
         n=1
@@ -46,6 +56,7 @@ with gr.Blocks() as demo:
             api_key = gr.Textbox(label="OpenAI API Key", type="password")
             prompt = gr.Textbox(label="Prompt")
             style = gr.Dropdown(label="Style", choices=STYLES)
+            negative_prompt = gr.Textbox(label="Negative Prompt", value=DEFAULT_NEGATIVE_PROMPT)
             generate_btn = gr.Button("Generate Image")
         
         with gr.Column(scale=1):
@@ -54,7 +65,7 @@ with gr.Blocks() as demo:
     
     generate_btn.click(
         generate_image,
-        inputs=[api_key, prompt, style],
+        inputs=[api_key, prompt, style, negative_prompt],
         outputs=[image_output, enhanced_prompt_output]
     )