Revert app.py

app.py

@@ -6,21 +6,23 @@ 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
 
-    prompt = (
-        f"Create a beautiful artistic profile picture of a smiling "
-        f"{age} year-old {person} with {hair_color} hair, {eye_color} eyes. "
-        f"Style: {style}."
-    )
+    # 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:
@@ -28,38 +30,39 @@ def generate_image(hair_color, eye_color, person, age, style, password):
             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"]
+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),
-        gr.Dropdown(choices=style_options,      label="Style"),
+        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.Textbox(label="Prompt"),
-        gr.Image(label="Generated Image")
-    ],
+    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
+    concurrency_limit=1  # Set concurrency limit here
 )
 
-if __name__ == "__main__":
-    iface.launch()
+# Launch the Gradio app
+iface.launch()