app.py

import openai
import os
import gradio as gr

# Ensure the OPENAI_API_KEY environment variable is set
openai.api_key = os.getenv("OPENAI_API_KEY")

if openai.api_key is None:
    raise ValueError("Die Umgebungsvariable OPENAI_API_KEY ist nicht gesetzt.")

def generate_image(hair_color, mood, eye_color):
    # Constructing the prompt based on user input
    prompt = f"Create a beautiful artistic profile picture of a person with {hair_color} hair, {eye_color} eyes, and looks {mood}."
    
    # Displaying the prompt to the user
    print("Prompt:", prompt)

    try:
        response = openai.ImageGeneration.create(
            model="dall-e-3",
            prompt=prompt,
            size="512x512",  # Choose from "1024x1024", "1024x1792", or "1792x1024"
            quality="standard",  # Choose "standard" or "hd" for higher quality
            n=1  # Number of images to generate
        )
        
        # Get the image URL from the response
        image_url = response['data'][0]['url']
        
        return prompt, image_url
    except Exception as e:
        print("An error occurred:", e)
        return "An error occurred while generating the image.", None

# Define the Gradio interface
iface = gr.Interface(
    fn=generate_image,
    inputs=[gr.Textbox(label="Hair Color"), gr.Textbox(label="Mood"), gr.Textbox(label="Eye Color")],
    outputs=[gr.Text(label="Prompt"), gr.Image(label="Generated Image")],
    title="Profile Picture Generator",
    description="Enter the hair color, mood, and eye color to generate a profile picture."
)

# Run the Gradio app
iface.launch()