Update app.py

app.py

@@ -6,10 +6,11 @@ from datetime import datetime
 from pathlib import Path
 
 # ============================
-#  TOKENS (via Secrets)
+#  USER ACCESS LIST
 # ============================
-PREMIUM_TOKEN = os.getenv("PREMIUM_KEY")
-SUPER_TOKEN = os.getenv("SUPER_PREMIUM_KEY")
+# Daftar username HuggingFace yang punya akses premium/super premium
+PREMIUM_USERS = ["Akhmad123", "user_premium_1"]
+SUPER_USERS = ["Akhmad123", "user_super_1"]
 
 # ============================
 #  LOAD TEMPLATES
@@ -136,84 +137,64 @@ def generate(goal, kind, tone, audience, length, constraints, language, n, tier=
     return "\n\n---\n\n".join(ready)
 
 # ============================
-#  TOKEN HANDLERS
+#  MAIN HANDLER (LOGIN)
 # ============================
-def premium_handler(token, *args):
-    if token != PREMIUM_TOKEN:
-        return "❌ Token Premium salah."
-    return generate(*args, tier="premium")
+def main_handler(request: gr.Request, goal=None, kind=None, tone=None, audience=None,
+                 length=None, constraints=None, language=None, n=None):
 
-def super_handler(token, *args):
-    if token != SUPER_TOKEN:
-        return "❌ Token Super Premium salah."
-    return generate(*args, tier="super")
+    username = request.username or "anonymous"
+
+    # Tentukan tier user
+    if username in SUPER_USERS:
+        tier = "super"
+    elif username in PREMIUM_USERS:
+        tier = "premium"
+    else:
+        tier = "free"
+
+    return generate(goal, kind, tone, audience, length, constraints, language, n, tier)
 
 # ============================
 #  UI
 # ============================
 with gr.Blocks() as demo:
 
-    gr.Markdown("# 🌟 AIPromptLab — Super Premium Edition")
-
-    # FREE TAB
-    with gr.Tab("Free"):
-        goal = gr.Textbox(label="Goal")
-        kind = gr.Radio(["text", "image", "code"], value="text")
-        tone = gr.Textbox(label="Tone")
-        audience = gr.Textbox(label="Audience")
-        length = gr.Textbox(label="Length")
-        constraints = gr.Textbox(label="Constraints")
-        language = gr.Textbox(label="Language", value="Indonesian")
-        n = gr.Slider(1, 5, value=3, step=1, label="Jumlah Varian")
-
-        btn = gr.Button("Generate (Free)")
-        out = gr.Textbox(label="Prompt Siap Pakai", lines=20, elem_id="copy-free")
-        copy_btn = gr.Button("Copy")
-
-        btn.click(generate, [goal, kind, tone, audience, length, constraints, language, n], out)
-        copy_btn.click(None, None, None, js="navigator.clipboard.writeText(document.getElementById('copy-free').value)")
-
-    # PREMIUM TAB
-    with gr.Tab("Premium"):
-        token_p = gr.Textbox(label="Token Premium", type="password")
-        goal_p = gr.Textbox(label="Goal")
-        kind_p = gr.Radio(["text", "image", "code"], value="text")
-        tone_p = gr.Textbox(label="Tone")
-        audience_p = gr.Textbox(label="Audience")
-        length_p = gr.Textbox(label="Length")
-        constraints_p = gr.Textbox(label="Constraints")
-        language_p = gr.Textbox(label="Language", value="Indonesian")
-        n_p = gr.Slider(1, 10, value=5, step=1, label="Jumlah Varian")
-
-        btn_p = gr.Button("Generate (Premium)")
-        out_p = gr.Textbox(label="Prompt Siap Pakai", lines=20, elem_id="copy-premium")
-        copy_p = gr.Button("Copy")
-
-        btn_p.click(premium_handler,
-                    [token_p, goal_p, kind_p, tone_p, audience_p, length_p, constraints_p, language_p, n_p],
-                    out_p)
-        copy_p.click(None, None, None, js="navigator.clipboard.writeText(document.getElementById('copy-premium').value)")
-
-    # SUPER PREMIUM TAB
-    with gr.Tab("Super Premium"):
-        token_s = gr.Textbox(label="Token Super Premium", type="password")
-        goal_s = gr.Textbox(label="Goal")
-        kind_s = gr.Radio(["text", "image", "code"], value="text")
-        tone_s = gr.Textbox(label="Tone")
-        audience_s = gr.Textbox(label="Audience")
-        length_s = gr.Textbox(label="Length")
-        constraints_s = gr.Textbox(label="Constraints")
-        language_s = gr.Textbox(label="Language", value="Indonesian")
-        n_s = gr.Slider(1, 20, value=10, step=1, label="Jumlah Varian")
-
-        btn_s = gr.Button("Generate (Super Premium)")
-        out_s = gr.Textbox(label="Prompt Siap Pakai", lines=20, elem_id="copy-super")
-        copy_s = gr.Button("Copy")
-
-        btn_s.click(super_handler,
-                    [token_s, goal_s, kind_s, tone_s, audience_s, length_s, constraints_s, language_s, n_s],
-                    out_s)
-        copy_s.click(None, None, None, js="navigator.clipboard.writeText(document.getElementById('copy-super').value)")
+    gr.Markdown("# 🌟 AIPromptLab — Unified Login Edition")
+
+    gr.Markdown(
+        """
+        Sistem otomatis mendeteksi level akses berdasarkan akun HuggingFace Anda:
+        - **Free** → User biasa  
+        - **Premium** → User terdaftar  
+        - **Super Premium** → User VIP  
+        """
+    )
+
+    goal = gr.Textbox(label="Goal")
+    kind = gr.Radio(["text", "image", "code"], value="text")
+    tone = gr.Textbox(label="Tone")
+    audience = gr.Textbox(label="Audience")
+    length = gr.Textbox(label="Length")
+    constraints = gr.Textbox(label="Constraints")
+    language = gr.Textbox(label="Language", value="Indonesian")
+    n = gr.Slider(1, 20, value=3, step=1, label="Jumlah Varian")
+
+    btn = gr.Button("🚀 Generate Prompt")
+    out = gr.Textbox(label="Prompt Siap Pakai", lines=20, elem_id="copy-output")
+    copy_btn = gr.Button("Copy")
+
+    btn.click(
+        main_handler,
+        [goal, kind, tone, audience, length, constraints, language, n],
+        out
+    )
+
+    copy_btn.click(
+        None,
+        None,
+        None,
+        js="navigator.clipboard.writeText(document.getElementById('copy-output').value)"
+    )
 
 # ============================
 #  LAUNCH
@@ -221,5 +202,6 @@ with gr.Blocks() as demo:
 demo.launch(
     server_name="0.0.0.0",
     server_port=7860,
-    theme=gr.themes.Soft()
+    theme=gr.themes.Soft(),
+    auth_message="Silakan login untuk mengakses fitur sesuai level akun Anda."
 )