Update app.py

app.py

@@ -5,10 +5,13 @@ from datetime import datetime
 from pathlib import Path
 
 # ============================
-#  USER ACCESS LIST
+#  USER DATABASE
 # ============================
-PREMIUM_USERS = ["Akhmad123", "user_premium_1"]
-SUPER_USERS = ["Akhmad123", "user_super_1"]
+USER_DB = {
+    "akhmad": {"password": "12345", "tier": "super"},
+    "premium_user": {"password": "abcde", "tier": "premium"},
+    "free_user": {"password": "00000", "tier": "free"},
+}
 
 # ============================
 #  LOAD TEMPLATES
@@ -108,7 +111,7 @@ def generate_variants(kind, goal, tone, audience, length, constraints, language,
 
     return variants
 
-def generate_json(username, goal, kind, tone, audience, length, constraints, language, n, tier):
+def generate_json(username, tier, goal, kind, tone, audience, length, constraints, language, n):
     goal = normalize(goal, "Describe the task clearly.")
     tone = normalize(tone, "neutral")
     audience = normalize(audience, "general")
@@ -135,51 +138,89 @@ def generate_json(username, goal, kind, tone, audience, length, constraints, lan
     return json.dumps(data, ensure_ascii=False, indent=2)
 
 # ============================
-#  MAIN HANDLER (LOGIN)
+#  LOGIN HANDLER
 # ============================
-def main_handler(request: gr.Request, goal, kind, tone, audience, length, constraints, language, n):
-    username = request.username or "anonymous"
-
-    if username in SUPER_USERS:
-        tier = "super"
-    elif username in PREMIUM_USERS:
-        tier = "premium"
-    else:
-        tier = "free"
-
-    return generate_json(username, goal, kind, tone, audience, length, constraints, language, n, tier)
+def login(username, password):
+    if username in USER_DB and USER_DB[username]["password"] == password:
+        return True, USER_DB[username]["tier"], f"Login berhasil. Selamat datang, {username}!"
+    return False, None, "❌ Username atau password salah."
 
 # ============================
 #  UI
 # ============================
 with gr.Blocks() as demo:
 
-    gr.Markdown("# 🔐 Login Required")
-    gr.Markdown("Silakan login dengan akun HuggingFace Anda untuk melanjutkan.")
+    gr.Markdown("# 🔐 Login untuk Menggunakan AIPromptLab")
+
+    login_status = gr.State(False)
+    login_user = gr.State("")
+    login_tier = gr.State("free")
+
+    username = gr.Textbox(label="Username")
+    password = gr.Textbox(label="Password", type="password")
+    login_btn = gr.Button("Login")
+    login_msg = gr.Markdown("")
+
+    # ============================
+    #  PROMPT UI (TERSEMBUNYI)
+    # ============================
+    with gr.Group(visible=False) as prompt_ui:
+
+        gr.Markdown("# 🌟 AIPromptLab — JSON Output Edition")
+
+        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")
+
+        generate_btn = gr.Button("🚀 Generate JSON")
+        output_json = gr.Textbox(label="Output JSON", lines=25, elem_id="json-output")
+        copy_btn = gr.Button("Copy JSON")
+
+    # ============================
+    #  LOGIN LOGIC
+    # ============================
+    def handle_login(username, password):
+        ok, tier, msg = login(username, password)
+        return (
+            msg,
+            gr.update(visible=ok),
+            ok,
+            username if ok else "",
+            tier if ok else "free"
+        )
 
-    gr.Markdown("---")
+    login_btn.click(
+        handle_login,
+        [username, password],
+        [login_msg, prompt_ui, login_status, login_user, login_tier]
+    )
 
-    gr.Markdown("# 🌟 AIPromptLab — JSON Output Edition")
+    # ============================
+    #  GENERATE JSON
+    # ============================
+    def handle_generate(login_status, login_user, login_tier,
+                        goal, kind, tone, audience, length, constraints, language, n):
 
-    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")
+        if not login_status:
+            return "❌ Anda belum login."
 
-    btn = gr.Button("🚀 Generate JSON")
-    out = gr.Textbox(label="Output JSON", lines=25, elem_id="json-output")
-    copy_btn = gr.Button("Copy JSON")
+        return generate_json(login_user, login_tier, goal, kind, tone, audience, length, constraints, language, n)
 
-    btn.click(
-        main_handler,
-        [goal, kind, tone, audience, length, constraints, language, n],
-        out
+    generate_btn.click(
+        handle_generate,
+        [login_status, login_user, login_tier,
+         goal, kind, tone, audience, length, constraints, language, n],
+        output_json
     )
 
+    # ============================
+    #  COPY JSON
+    # ============================
     copy_btn.click(
         None,
         None,
@@ -193,6 +234,5 @@ with gr.Blocks() as demo:
 demo.launch(
     server_name="0.0.0.0",
     server_port=7860,
-    theme=gr.themes.Soft(),
-    auth_message="Silakan login untuk menggunakan AIPromptLab."
+    theme=gr.themes.Soft()
 )