Spaces:
Running
Running
| import os | |
| from openai import OpenAI | |
| import gradio as gr | |
| from dotenv import load_dotenv | |
| # Load environment variables from the .env file | |
| load_dotenv() | |
| # Ensure the OPENAI_API_KEY environment variable is set; it's automagically loaded into the client | |
| client = OpenAI() | |
| if client.api_key is None: | |
| raise ValueError("Die Umgebungsvariable OPENAI_API_KEY ist nicht gesetzt.") | |
| # Load the APP_PASSWORD from the environment | |
| APP_PASSWORD = os.getenv("APP_PASSWORD") | |
| def generate_image(hair_color, eye_color, person, age, style, password): | |
| if password != APP_PASSWORD: | |
| return "Incorrect password.", None | |
| # Constructing the prompt based on user input | |
| prompt = f"Create a beautiful artistic profile picture of a smiling {age} year-old {person} with {hair_color} hair, {eye_color} eyes. Style: {style}." | |
| # Displaying the prompt to the user | |
| print("Prompt:", prompt) | |
| try: | |
| response = client.images.generate( | |
| model="dall-e-3", | |
| prompt=prompt, | |
| size="1024x1024", | |
| quality="standard", | |
| n=1 | |
| ) | |
| # 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 | |
| hair_color_options = ["Black", "Brown", "Blonde", "Red", "Gray", "White", "Green", "Rainbow", "Blue"] | |
| eye_color_options = ["Blue", "Brown", "Green", "Gray", "Hazel"] | |
| person_options = ["Boy", "Girl"] | |
| style_options = ["Fortnite Character", "3D Pixar Render", "Picasso Painting", "Van Gogh Painting", "Abstract Art", "Realistic Portrait"] | |
| # Define the Gradio interface | |
| iface = gr.Interface( | |
| fn=generate_image, | |
| inputs=[ | |
| gr.Dropdown(choices=hair_color_options, label="Hair Color"), | |
| gr.Dropdown(choices=eye_color_options, label="Eye Color"), | |
| gr.Dropdown(choices=person_options, label="Person"), | |
| gr.Number(label="Age", value=12, minimum=8, maximum=120, step=1, precision=0), | |
| gr.Dropdown(choices=style_options, label="Style"), | |
| gr.Textbox(label="Password", type="password"), | |
| ], | |
| outputs=[gr.Text(label="Prompt"), gr.Image(label="Generated Image")], | |
| title="Profile Picture Generator", | |
| description="Enter the attributes for your profile picture.", | |
| concurrency_limit=1 # Set concurrency limit here | |
| ) | |
| # Launch the Gradio app | |
| iface.launch() |