more options

app.py

@@ -8,9 +8,9 @@ client = OpenAI()
 if client.api_key is None:
     raise ValueError("Die Umgebungsvariable OPENAI_API_KEY ist nicht gesetzt.")
 
-def generate_image(hair_color, mood, eye_color):
+def generate_image(hair_color, eye_color, person, age, style):
     # 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}."
+    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)
@@ -32,13 +32,24 @@ def generate_image(hair_color, mood, eye_color):
         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.Textbox(label="Hair Color"), gr.Textbox(label="Mood"), gr.Textbox(label="Eye Color")],
+    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", default=12, step=1, precision=0),
+        gr.Dropdown(choices=style_options, label="style"),
+    ],
     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."
+    description="Enter the attributes for your profile picture."
 )
 
 # Run the Gradio app