app.py

import gradio as gr
import time
from dotenv import load_dotenv
import os
from openai import OpenAI
import secrets

load_dotenv()
client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))

if client.api_key is None:
    raise ValueError("Die Umgebungsvariable OPENAI_API_KEY ist nicht gesetzt.")

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", 60))

def enhance_prompt(original_prompt):
    try:
        response = client.chat.completions.create(
            model="gpt-4o-mini",
            messages=[
                {"role": "system", "content": "Verbessere den folgenden Bildgenerierungsprompt, um ihn detaillierter und kreativer zu machen."},
                {"role": "user", "content": original_prompt}
            ],
            max_tokens=300
        )
        enhanced_prompt = response.choices[0].message.content.strip()
        return enhanced_prompt
    except Exception as e:
        print("Ein Fehler ist beim Verbessern des Prompts aufgetreten:", e)
        return original_prompt

def generate_image(prompt, use_magic_prompt, style, password, user_id, last_generation_time):
    print(f"\nGenerierungsversuch von User {user_id} mit Prompt:\n{prompt}")
    
    if password != APP_PASSWORD:
        print("Generierung abgebrochen: Falsches Passwort.")
        return "Falsches Passwort. Bitte versuche es erneut.", None,  last_generation_time

    current_time = time.time()
    if current_time - last_generation_time < GENERATION_TIMEOUT:
        remaining_time = max(0, int(GENERATION_TIMEOUT - (current_time - last_generation_time)))
        print("Generierung abgebrochen: Cooldown noch nicht abgelaufen")
        return f"Bitte warte noch {remaining_time} Sekunden bis zur nächsten Bildgenerierungen.\n\nNutze die Wartezeit, um deinen Prompt zu verfeinern. &#128521;", None,  last_generation_time

    if use_magic_prompt:
        prompt = enhance_prompt(prompt)
        print("\nFinaler Prompt:", prompt)

    try:
        response = client.images.generate(
            model="dall-e-3",
            prompt=prompt,
            size="1024x1024",
            quality="standard",
            style=style,
            n=1
        )
        
        image_url = response.data[0].url
        
        # Update the last generation time for this user
        last_generation_time = current_time
        
        return prompt, image_url, last_generation_time
    except Exception as e:
        error_message = str(e)
        if "content_policy_violation" in error_message:
            return "Leider können wir deinen Prompt nicht verarbeiten, da er evtl. gegen die Content Policy von OpenAI verstösst. Wenn du denkst, dass das ein Fehler ist, versuche den Prompt so umzuformulieren, dass er jugendfreundlich ist.", None
        print("Ein Fehler ist aufgetreten:", e)
        return f"Ein Fehler ist beim Generieren des Bildes aufgetreten: {error_message}", None, last_generation_time

examples = [
    "Astronaut im Dschungel, kalte Farbpalette, gedämpfte Farben, detailliert, 8k",
    "Ein Astronaut reitet auf einem grünen Pferd",
    "Ein köstliches Ceviche-Käsekuchenstück",
    "Ein wunderschönes Glumanda in einer blühenden Wiese",
    "Ein futuristisches Stadtbild bei Nacht mit Neonlichtern",
    "Ein Roboter, der in einem antiken Tempel tanzt",
    "Ein magisches Einhorn, das über einen Regenbogen springt",
    "Ein mittelalterlicher Ritter, der gegen einen Drachen kämpft"
]

def update_button_state(last_generation_time):
    current_time = time.time()
    remaining_time = max(0, int(GENERATION_TIMEOUT - (current_time - last_generation_time)))
    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)
    
def update_welcome_message(user_id):
    return f"Willkommen, {user_id}!"

def generate_user_id():
    return secrets.token_urlsafe(16)

def initialize_session():
    new_user_id = generate_user_id()
    return new_user_id, update_welcome_message(new_user_id), 0

with gr.Blocks() as demo:
    user_id = gr.State(generate_user_id)
    last_generation_time = gr.State(0)

    gr.Markdown("# codora DALL-E 3 Bildgenerator")
    welcome_message = gr.Markdown("Willkommen!")
    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():
        with gr.Column(scale=1):
            prompt = gr.Textbox(label="Bildprompt")
            use_magic_prompt = gr.Checkbox(label="Magischen Prompt verwenden")
            style = gr.Radio(
                label="Stil",
                choices=["vivid", "natural"],
                value="vivid"
            )
            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")  # Removed scale argument

    # Place examples below the generation button
    gr.Examples(
        examples=examples,
        inputs=[prompt]
    )

    generate_button.click(
        fn=generate_image,
        inputs=[prompt, use_magic_prompt, style, password, user_id, last_generation_time],
        outputs=[final_prompt, generated_image, last_generation_time]
    )

    demo.load(fn=initialize_session, outputs=[user_id, welcome_message, last_generation_time])

    demo.load(
        fn=update_button_state,
        inputs=[last_generation_time],
        outputs=[generate_button],
        every=1
    )

demo.launch()