Update app.py

app.py

@@ -6,11 +6,6 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
 from diffusers import DiffusionPipeline, LCMScheduler
 from PIL import Image, ImageFilter
 
-# ===============================
-# PASSWORD
-# ===============================
-PASSWORD = "CREEPERIMG"
-
 # ===============================
 # TEXT MODEL (PROMPT ENHANCER)
 # ===============================
@@ -20,10 +15,6 @@ tokenizer = AutoTokenizer.from_pretrained(TEXT_MODEL_ID)
 text_model = AutoModelForCausalLM.from_pretrained(TEXT_MODEL_ID)
 
 def enhance_prompt(user_prompt: str) -> str:
-    """
-    Enhances the user prompt for image generation.
-    Returns ONLY the enhanced prompt text.
-    """
     instruction = (
         "Please enhance this prompt so it is suitable for an image generator "
         "that requires clear instructions. Analyse the prompt, and output as "
@@ -42,7 +33,6 @@ def enhance_prompt(user_prompt: str) -> str:
 
     decoded = tokenizer.decode(outputs[0], skip_special_tokens=True)
 
-    # ⛔️ CRITICAL: extract only what comes AFTER "Enhanced prompt:"
     if "Enhanced prompt:" in decoded:
         decoded = decoded.split("Enhanced prompt:")[-1]
 
@@ -81,31 +71,20 @@ def estimate_time(steps, res):
 # ===============================
 # GENERATION FUNCTION
 # ===============================
-def generate(password, prompt, negative, resolution, steps):
+def generate(prompt, negative, resolution, steps):
     size = int(resolution)
 
-    # Password gate
-    if password != PASSWORD:
-        return (
-            [Image.new("RGB", (size, size), "white")],
-            "❌ Wrong password",
-            ""
-        )
-
     # Placeholder while thinking
     yield (
         [Image.new("RGB", (size, size), "white")],
-        "🧠 Enhancing prompt...",
-        ""
+        "🧠 Enhancing prompt..."
     )
 
     enhanced = enhance_prompt(prompt)
 
-    # Show enhanced prompt immediately
     yield (
         [Image.new("RGB", (size, size), "white")],
-        "🎨 Generating image...",
-        enhanced
+        "🎨 Generating image..."
     )
 
     seed = random.randint(0, 1_000_000_000)
@@ -125,20 +104,18 @@ def generate(password, prompt, negative, resolution, steps):
     ).images[0]
     elapsed = int(time.time() - start)
 
-    # Blur reveal (UI-only)
+    # Blur reveal
     for i in range(10):
         blur = image.filter(ImageFilter.GaussianBlur(radius=(10 - i)))
         yield (
             [blur],
-            f"👀 Revealing... ({i+1}/10)",
-            enhanced
+            f"👀 Revealing... ({i+1}/10)"
         )
         time.sleep(1)
 
     yield (
         [image],
-        f"✅ Done in {elapsed}s | Seed {seed}",
-        enhanced
+        f"✅ Done in {elapsed}s | Seed {seed}"
     )
 
 # ===============================
@@ -148,12 +125,11 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
     gr.Markdown("# 👾 Creeper AI — Image Generator")
     gr.Markdown(
         "1️⃣ The higher the resolution & steps, the longer the image takes.\n\n"
-        "2️⃣ More detailed prompts = better results."
+        "2️⃣ The more detailed the prompt and negative prompt, the better the result."
     )
 
     with gr.Row():
         with gr.Column():
-            password = gr.Textbox(label="Password", type="password")
             prompt = gr.Textbox(label="Prompt")
             negative = gr.Textbox(label="Negative Prompt")
 
@@ -164,10 +140,6 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
             generate_btn = gr.Button("Generate")
 
             status = gr.Markdown("🟢 Ready")
-            enhanced_box = gr.Textbox(
-                label="Enhanced Prompt (exact output from text AI)",
-                interactive=False
-            )
 
         with gr.Column():
             gallery = gr.Gallery(columns=1)
@@ -177,8 +149,8 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
 
     generate_btn.click(
         generate,
-        inputs=[password, prompt, negative, resolution, steps],
-        outputs=[gallery, status, enhanced_box]
+        inputs=[prompt, negative, resolution, steps],
+        outputs=[gallery, status]
     )
 
 demo.launch()