Update app.py

app.py

@@ -1,14 +1,12 @@
 # app.py
 import gradio as gr
 import json
-import os
 from datetime import datetime
 from pathlib import Path
 
 # ============================
 #  USER ACCESS LIST
 # ============================
-# Daftar username HuggingFace yang punya akses premium/super premium
 PREMIUM_USERS = ["Akhmad123", "user_premium_1"]
 SUPER_USERS = ["Akhmad123", "user_super_1"]
 
@@ -110,7 +108,7 @@ def generate_variants(kind, goal, tone, audience, length, constraints, language,
 
     return variants
 
-def generate(goal, kind, tone, audience, length, constraints, language, n, tier="free"):
+def generate_json(username, goal, kind, tone, audience, length, constraints, language, n, tier):
     goal = normalize(goal, "Describe the task clearly.")
     tone = normalize(tone, "neutral")
     audience = normalize(audience, "general")
@@ -125,26 +123,23 @@ def generate(goal, kind, tone, audience, length, constraints, language, n, tier=
         kind, goal, tone, audience, length, constraints, language, n, tier
     )
 
-    ready = []
-    for v in variants:
-        if kind == "text":
-            ready.append(f"### ChatGPT Prompt\n{v['prompt']}")
-        elif kind == "image":
-            ready.append(f"### Image Prompt\n{v['prompt']}")
-        elif kind == "code":
-            ready.append(f"### Code Prompt\n{v['prompt']}")
+    data = {
+        "generated_at": datetime.utcnow().isoformat() + "Z",
+        "user": username,
+        "tier": tier,
+        "kind": kind,
+        "goal": goal,
+        "variants": variants
+    }
 
-    return "\n\n---\n\n".join(ready)
+    return json.dumps(data, ensure_ascii=False, indent=2)
 
 # ============================
 #  MAIN HANDLER (LOGIN)
 # ============================
-def main_handler(request: gr.Request, goal=None, kind=None, tone=None, audience=None,
-                 length=None, constraints=None, language=None, n=None):
-
+def main_handler(request: gr.Request, goal, kind, tone, audience, length, constraints, language, n):
     username = request.username or "anonymous"
 
-    # Tentukan tier user
     if username in SUPER_USERS:
         tier = "super"
     elif username in PREMIUM_USERS:
@@ -152,23 +147,19 @@ def main_handler(request: gr.Request, goal=None, kind=None, tone=None, audience=
     else:
         tier = "free"
 
-    return generate(goal, kind, tone, audience, length, constraints, language, n, tier)
+    return generate_json(username, goal, kind, tone, audience, length, constraints, language, n, tier)
 
 # ============================
 #  UI
 # ============================
 with gr.Blocks() as demo:
 
-    gr.Markdown("# 🌟 AIPromptLab — Unified Login Edition")
+    gr.Markdown("# 🔐 Login Required")
+    gr.Markdown("Silakan login dengan akun HuggingFace Anda untuk melanjutkan.")
 
-    gr.Markdown(
-        """
-        Sistem otomatis mendeteksi level akses berdasarkan akun HuggingFace Anda:
-        - **Free** → User biasa  
-        - **Premium** → User terdaftar  
-        - **Super Premium** → User VIP  
-        """
-    )
+    gr.Markdown("---")
+
+    gr.Markdown("# 🌟 AIPromptLab — JSON Output Edition")
 
     goal = gr.Textbox(label="Goal")
     kind = gr.Radio(["text", "image", "code"], value="text")
@@ -179,9 +170,9 @@ with gr.Blocks() as demo:
     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 = gr.Button("🚀 Generate JSON")
+    out = gr.Textbox(label="Output JSON", lines=25, elem_id="json-output")
+    copy_btn = gr.Button("Copy JSON")
 
     btn.click(
         main_handler,
@@ -193,7 +184,7 @@ with gr.Blocks() as demo:
         None,
         None,
         None,
-        js="navigator.clipboard.writeText(document.getElementById('copy-output').value)"
+        js="navigator.clipboard.writeText(document.getElementById('json-output').value)"
     )
 
 # ============================
@@ -203,5 +194,5 @@ demo.launch(
     server_name="0.0.0.0",
     server_port=7860,
     theme=gr.themes.Soft(),
-    auth_message="Silakan login untuk mengakses fitur sesuai level akun Anda."
+    auth_message="Silakan login untuk menggunakan AIPromptLab."
 )