add queuing and password

app.py

@@ -1,6 +1,10 @@
-from openai import OpenAI
 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()
@@ -8,7 +12,13 @@ client = OpenAI()
 if client.api_key is None:
     raise ValueError("Die Umgebungsvariable OPENAI_API_KEY ist nicht gesetzt.")
 
-def generate_image(hair_color, eye_color, person, age, style):
+# 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}."
     
@@ -45,11 +55,13 @@ iface = gr.Interface(
         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.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."
+    description="Enter the attributes for your profile picture.",
+    queue=True  # Enable queuing for the image generation process
 )
 
 # Run the Gradio app

requirements.txt

@@ -1,2 +1,3 @@
 openai
-gradio
+gradio
+python-dotenv