Unique UUIDs for user sessions

app.py

@@ -5,6 +5,7 @@ from dotenv import load_dotenv
 import os
 from openai import OpenAI
 import random
+import uuid
 
 load_dotenv()
 client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
@@ -16,10 +17,7 @@ APP_PASSWORD = os.getenv("APP_PASSWORD")
 if APP_PASSWORD is None:
     raise ValueError("Die Umgebungsvariable APP_PASSWORD ist nicht gesetzt.")
 
-
-GENERATION_TIMEOUT = float(os.getenv("GENERATION_TIMEOUT"))
-if GENERATION_TIMEOUT is None:
-    GENERATION_TIMEOUT = 60
+GENERATION_TIMEOUT = float(os.getenv("GENERATION_TIMEOUT", 60))
 
 last_generation_time = defaultdict(float)
 
@@ -39,7 +37,7 @@ def enhance_prompt(original_prompt):
         print("Ein Fehler ist beim Verbessern des Prompts aufgetreten:", e)
         return original_prompt
 
-def generate_image(prompt, use_magic_prompt, style, user_id, password):
+def generate_image(prompt, use_magic_prompt, style, password, user_id):
     global last_generation_time
     
     if password != APP_PASSWORD:
@@ -47,7 +45,8 @@ def generate_image(prompt, use_magic_prompt, style, user_id, password):
 
     current_time = time.time()
     if current_time - last_generation_time[user_id] < GENERATION_TIMEOUT:
-        return f"Bitte warte {GENERATION_TIMEOUT} Sekunden zwischen den Bildgenerierungen.", None
+        remaining_time = int(GENERATION_TIMEOUT - (current_time - last_generation_time[user_id]))
+        return f"Bitte warte {remaining_time} Sekunden zwischen den Bildgenerierungen.", None
 
     if use_magic_prompt:
         prompt = enhance_prompt(prompt)
@@ -77,6 +76,9 @@ def generate_image(prompt, use_magic_prompt, style, user_id, password):
         print("Ein Fehler ist aufgetreten:", e)
         return f"Ein Fehler ist beim Generieren des Bildes aufgetreten: {error_message}", None
 
+def get_user_id():
+    return str(uuid.uuid4())
+
 examples = [
     "Astronaut im Dschungel, kalte Farbpalette, gedämpfte Farben, detailliert, 8k",
     "Ein Astronaut reitet auf einem grünen Pferd",
@@ -88,8 +90,19 @@ examples = [
     "Ein mittelalterlicher Ritter, der gegen einen Drachen kämpft"
 ]
 
+def update_button_state(user_id):
+    current_time = time.time()
+    remaining_time = max(0, int(GENERATION_TIMEOUT - (current_time - last_generation_time[user_id])))
+    if remaining_time > 0:
+        return gr.update(value=f"Bitte warte {remaining_time} Sekunden", interactive=False)
+    else:
+        return gr.update(value="Bild generieren", interactive=True)
+
 with gr.Blocks() as demo:
+    user_id = get_user_id()
+    
     gr.Markdown("# codora DALL-E 3 Bildgenerator")
+    gr.Markdown(f"Willkommen, {user_id}!")
     gr.Markdown(f"Gib einen Bildprompt ein und verwende optional die magische Prompt-Funktion, um ihn zu verbessern. Du kannst nur ein Bild alle {GENERATION_TIMEOUT} Sekunden generieren.")
 
     with gr.Row():
@@ -101,13 +114,12 @@ with gr.Blocks() as demo:
                 choices=["vivid", "natural"],
                 value="vivid"
             )
-            user_id = gr.Textbox(label="Benutzer-ID", visible=False)
             password = gr.Textbox(label="App Passwort", type="password")
             generate_button = gr.Button("Bild generieren")
 
         with gr.Column(scale=1):
             final_prompt = gr.Textbox(label="Finaler Prompt")
-            generated_image = gr.Image(label="Generiertes Bild", scale=0.5)  # Set scale to 0.5 for 50% width
+            generated_image = gr.Image(label="Generiertes Bild")  # Removed scale argument
 
     # Place examples below the generation button
     gr.Examples(
@@ -117,8 +129,15 @@ with gr.Blocks() as demo:
 
     generate_button.click(
         fn=generate_image,
-        inputs=[prompt, use_magic_prompt, style, user_id, password],
+        inputs=[prompt, use_magic_prompt, style, password, gr.State(user_id)],
         outputs=[final_prompt, generated_image]
     )
 
-demo.launch()
+    demo.load(
+        fn=update_button_state,
+        inputs=[gr.State(user_id)],
+        outputs=[generate_button],
+        every=1
+    )
+
+demo.launch()