app.py

import gradio as gr
import openai
import os

# List of popular styles
STYLES = [
    "Photorealistic", "Oil Painting", "Watercolor", "Anime", 
    "Studio Ghibli", "Black and White", "Polaroid", "Sketch", 
    "3D Render", "Pixel Art", "Cyberpunk", "Steampunk", 
    "Art Nouveau", "Pop Art", "Minimalist"
]

def enhance_prompt(prompt, style):
    openai.api_key = os.environ.get("OPENAI_API_KEY")
    
    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}'"
    
    response = openai.ChatCompletion.create(
        model="gpt-4",
        messages=[
            {"role": "system", "content": "You are an expert at creating detailed, vivid prompts for image generation."},
            {"role": "user", "content": enhanced_prompt_request}
        ]
    )
    
    return response.choices[0].message.content.strip()

def generate_image(api_key, prompt, style):
    openai.api_key = api_key
    
    enhanced_prompt = enhance_prompt(prompt, style)
    
    response = openai.Image.create(
        model="dall-e-3",
        prompt=enhanced_prompt,
        size="1024x1024",
        quality="standard",
        n=1
    )
    
    return response.data[0].url, enhanced_prompt

with gr.Blocks() as demo:
    gr.Markdown("# DALL-E 3 Image Generator")
    
    with gr.Row():
        with gr.Column(scale=1):
            api_key = gr.Textbox(label="OpenAI API Key", type="password")
            prompt = gr.Textbox(label="Prompt")
            style = gr.Dropdown(label="Style", choices=STYLES)
            generate_btn = gr.Button("Generate Image")
        
        with gr.Column(scale=1):
            image_output = gr.Image(label="Generated Image")
            enhanced_prompt_output = gr.Textbox(label="Enhanced Prompt")
    
    generate_btn.click(
        generate_image,
        inputs=[api_key, prompt, style],
        outputs=[image_output, enhanced_prompt_output]
    )

demo.launch()