Update app.py

app.py

@@ -2,11 +2,9 @@
 import gradio as gr
 import json
 from datetime import datetime
-from pathlib import Path
-from PIL import Image
 
 # ============================
-#  USER DATABASE
+#  USER DATABASE (SIMPLE)
 # ============================
 USER_DB = {
     "akhmad": {"password": "12345", "tier": "super"},
@@ -14,43 +12,6 @@ USER_DB = {
     "free_user": {"password": "00000", "tier": "free"},
 }
 
-# ============================
-#  LOAD TEMPLATES
-# ============================
-TEMPLATES_PATH = Path("templates.json")
-if TEMPLATES_PATH.exists():
-    with open(TEMPLATES_PATH, "r", encoding="utf-8") as f:
-        TEMPLATES = json.load(f)
-else:
-    TEMPLATES = {
-        "text": [
-            "{goal}\nAudience: {audience}\nTone: {tone}\nLength: {length}\nConstraints: {constraints}\n\nWrite a clear, structured output with headings and examples."
-        ],
-        "image": [
-            "Scene: {goal}. Style: {tone}. Camera: 35mm; Lighting: golden hour; Colors: warm. Details: {constraints}. Negative: avoid text, watermarks."
-        ],
-        "code": [
-            "Write code to {goal}. Language: {language}. Requirements: {constraints}. Include comments and tests."
-        ]
-    }
-
-# ============================
-#  IMAGE CAPTIONING MODEL
-# ============================
-from transformers import BlipProcessor, BlipForConditionalGeneration
-
-processor = BlipProcessor.from_pretrained("Salesforce/blip-image-captioning-base")
-model = BlipForConditionalGeneration.from_pretrained("Salesforce/blip-image-captioning-base")
-
-def describe_image(img):
-    if img is None:
-        return ""
-    image = Image.open(img)
-    inputs = processor(image, return_tensors="pt")
-    out = model.generate(**inputs)
-    caption = processor.decode(out[0], skip_special_tokens=True)
-    return caption
-
 # ============================
 #  HELPERS
 # ============================
@@ -59,91 +20,135 @@ def normalize(v, default=""):
         return default
     return str(v).strip()
 
-def augment(prompt, idx, tier="free"):
-    base = [
-        "\nInclude examples and a short checklist.",
-        "\nUse bullet points and a short summary.",
-        "\nProvide 3 alternative phrasings.",
-        "\nAdd a short FAQ section.",
-        "\nInclude a call-to-action."
-    ]
-
-    premium_extra = [
-        "\nOptimize for clarity and conversion.",
-        "\nAdd variations for A/B testing.",
-    ]
-
-    super_extra = [
-        "\nAdd SEO optimization with keywords.",
-        "\nUse power words to increase impact.",
-        "\nAdd a storytelling angle.",
-        "\nProvide 5 headline variations.",
-    ]
-
-    if tier == "premium":
-        base += premium_extra
-    if tier == "super":
-        base += premium_extra + super_extra
+def now_iso():
+    return datetime.utcnow().isoformat() + "Z"
 
-    if idx < len(base):
-        return prompt + base[idx]
-    return prompt
+# ============================
+#  E-BOOK JSON GENERATOR
+# ============================
+def generate_ebook_json(username, tier,
+                        goal, genre, tone, style,
+                        platform, length):
 
-def recommend(kind, tier="free"):
-    base = {
-        "text": "Use temperature 0.2–0.5 for factual text.",
-        "image": "Use negative prompts to remove artifacts.",
-        "code": "Validate output before production."
-    }.get(kind, "General guidance.")
+    goal = normalize(goal, "Islamic children's story about good manners.")
+    genre = normalize(genre, "Islamic Children Story")
+    tone = normalize(tone, "Friendly")
+    style = normalize(style, "Storybook style")
+    platform = normalize(platform, "KDP")
+    length = normalize(length, "Medium (30–50 pages)")
 
-    if tier == "premium":
-        return base + " Premium: cocok untuk workflow profesional."
-    if tier == "super":
-        return base + " Super Premium: optimized for expert-level output."
+    data = {
+        "type": "ebook",
+        "generated_at": now_iso(),
+        "user": username,
+        "tier": tier,
+        "platform": platform,
+        "genre": genre,
+        "tone": tone,
+        "writing_style": style,
+        "length": length,
+        "goal": goal,
+        "prompts": {
+            "title_prompt": (
+                f"Generate 10 title ideas for an {genre} ebook about '{goal}'. "
+                f"Tone: {tone}. Style: {style}. Platform: {platform}. "
+                "Titles should be short, catchy, and suitable for children."
+            ),
+            "outline_prompt": (
+                f"Create a detailed chapter outline for an {genre} ebook about '{goal}'. "
+                f"Tone: {tone}. Style: {style}. Length: {length}. "
+                "Include chapter titles and 2–3 bullet points per chapter."
+            ),
+            "chapter_prompt": (
+                f"Write Chapter 1 of an {genre} ebook about '{goal}'. "
+                f"Tone: {tone}. Style: {style}. Length: {length}. "
+                "Use simple language suitable for children and include dialogues."
+            )
+        }
+    }
 
-    return base
+    return json.dumps(data, ensure_ascii=False, indent=2)
 
 # ============================
-#  GENERATOR
+#  IMAGE DESIGN JSON GENERATOR
 # ============================
-def generate_variants(kind, goal, tone, audience, length, constraints, language, n, tier):
-    templates = TEMPLATES.get(kind, TEMPLATES["text"])
-    variants = []
-
-    for i in range(n):
-        tpl = templates[i % len(templates)]
-        prompt = tpl.format(
-            goal=goal,
-            tone=tone,
-            audience=audience,
-            length=length,
-            constraints=constraints,
-            language=language
-        )
-        prompt = augment(prompt, i, tier)
-        variants.append({
-            "id": f"v{i+1}",
-            "prompt": prompt,
-            "notes": recommend(kind, tier)
-        })
+def generate_image_json(username, tier,
+                        goal, design_style, color_palette,
+                        platform, aspect_ratio):
+
+    goal = normalize(goal, "Islamic children's story cover about kindness.")
+    design_style = normalize(design_style, "Pastel Islamic")
+    color_palette = normalize(color_palette, "Pastel")
+    platform = normalize(platform, "Midjourney")
+    aspect_ratio = normalize(aspect_ratio, "1:1")
+
+    prompt = (
+        f"{goal}, {design_style} style, {color_palette} colors, "
+        f"high quality, aspect ratio {aspect_ratio}, soft lighting, kid-friendly, Islamic theme."
+    )
 
-    return variants
+    negative_prompt = (
+        "no text, no watermark, no logo, no distortion, no blur, no low quality, no violence."
+    )
 
-def generate_json(username, tier, goal, kind, tone, audience, length, constraints, language, n):
-    limits = {"free": 5, "premium": 10, "super": 20}
-    n = max(1, min(limits[tier], int(n)))
+    data = {
+        "type": "image_design",
+        "generated_at": now_iso(),
+        "user": username,
+        "tier": tier,
+        "platform": platform,
+        "goal": goal,
+        "style": design_style,
+        "color_palette": color_palette,
+        "aspect_ratio": aspect_ratio,
+        "prompt": prompt,
+        "negative_prompt": negative_prompt
+    }
 
-    variants = generate_variants(
-        kind, goal, tone, audience, length, constraints, language, n, tier
-    )
+    return json.dumps(data, ensure_ascii=False, indent=2)
+
+# ============================
+#  VIDEO AI JSON GENERATOR
+# ============================
+def generate_video_json(username, tier,
+                        goal, video_style, camera_movement,
+                        platform, duration):
+
+    goal = normalize(goal, "Islamic children's story animation about honesty.")
+    video_style = normalize(video_style, "Pastel Kids Animation")
+    camera_movement = normalize(camera_movement, "Slow Zoom")
+    platform = normalize(platform, "Pika Labs")
+    duration = normalize(duration, "10 seconds")
+
+    scenes = [
+        {
+            "scene": 1,
+            "description": (
+                f"A child reading an Islamic storybook about '{goal}' in a cozy room, "
+                f"{video_style.lower()}, soft pastel colors, warm lighting."
+            ),
+            "camera": camera_movement
+        },
+        {
+            "scene": 2,
+            "description": (
+                "Close-up of the book pages glowing softly, Arabic patterns in the background, "
+                "gentle motion, peaceful atmosphere."
+            ),
+            "camera": "Static Shot"
+        }
+    ]
 
     data = {
-        "generated_at": datetime.utcnow().isoformat() + "Z",
+        "type": "video_ai",
+        "generated_at": now_iso(),
         "user": username,
         "tier": tier,
-        "kind": kind,
+        "platform": platform,
+        "style": video_style,
+        "duration": duration,
         "goal": goal,
-        "variants": variants
+        "scenes": scenes
     }
 
     return json.dumps(data, ensure_ascii=False, indent=2)
@@ -177,36 +182,185 @@ with gr.Blocks() as demo:
     # MAIN UI (hidden first)
     with gr.Group(visible=False) as main_ui:
 
-        gr.Markdown("# 🌟 AIPromptLab — JSON Output Edition")
-
-        input_mode = gr.Radio(["text", "image"], value="text", label="Input Mode")
-
-        text_input = gr.Textbox(label="Text Input", visible=True)
-        image_input = gr.Image(label="Upload Image", type="filepath", visible=False)
-
-        def switch_input(mode):
-            return (
-                gr.update(visible=(mode=="text")),
-                gr.update(visible=(mode=="image"))
-            )
-
-        input_mode.change(
-            switch_input,
-            [input_mode],
-            [text_input, image_input]
-        )
-
-        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")
+        gr.Markdown("# 🌟 AIPromptLab — Multi Output JSON Prompt Generator")
+
+        with gr.Tabs():
+
+            # ============================
+            #  TAB 1: E-BOOK
+            # ============================
+            with gr.Tab("E-Book Prompt"):
+
+                ebook_goal = gr.Textbox(label="Goal / Ide E-book")
+                ebook_genre = gr.Dropdown(
+                    label="Genre",
+                    choices=[
+                        "Islamic Children Story",
+                        "Parenting",
+                        "Motivation",
+                        "Fantasy",
+                        "Education",
+                        "Adventure",
+                        "Mystery"
+                    ],
+                    value="Islamic Children Story"
+                )
+                ebook_tone = gr.Dropdown(
+                    label="Tone",
+                    choices=[
+                        "Friendly",
+                        "Inspirational",
+                        "Educational",
+                        "Storytelling",
+                        "Formal",
+                        "Casual"
+                    ],
+                    value="Friendly"
+                )
+                ebook_style = gr.Dropdown(
+                    label="Writing Style",
+                    choices=[
+                        "Storybook style",
+                        "Narrative",
+                        "Descriptive",
+                        "Persuasive",
+                        "Expository",
+                        "Dialog-based"
+                    ],
+                    value="Storybook style"
+                )
+                ebook_platform = gr.Dropdown(
+                    label="Target Platform",
+                    choices=[
+                        "KDP",
+                        "Wattpad",
+                        "Google Play Books",
+                        "PDF / E-Learning"
+                    ],
+                    value="KDP"
+                )
+                ebook_length = gr.Dropdown(
+                    label="Length",
+                    choices=[
+                        "Short (10–20 pages)",
+                        "Medium (30–50 pages)",
+                        "Long (80–120 pages)"
+                    ],
+                    value="Medium (30–50 pages)"
+                )
+
+                ebook_generate_btn = gr.Button("🚀 Generate E-book JSON")
+                ebook_output = gr.Textbox(label="E-book JSON Output", lines=20, elem_id="ebook-json-output")
+                ebook_copy_btn = gr.Button("Copy E-book JSON")
+
+            # ============================
+            #  TAB 2: IMAGE DESIGN
+            # ============================
+            with gr.Tab("Image Design Prompt"):
+
+                img_goal = gr.Textbox(label="Goal / Ide Desain Gambar")
+                img_style = gr.Dropdown(
+                    label="Design Style",
+                    choices=[
+                        "Pastel Islamic",
+                        "Flat Illustration",
+                        "3D Render",
+                        "Watercolor",
+                        "Minimalist",
+                        "Cartoon Kids",
+                        "Realistic Photography"
+                    ],
+                    value="Pastel Islamic"
+                )
+                img_palette = gr.Dropdown(
+                    label="Color Palette",
+                    choices=[
+                        "Pastel",
+                        "Vibrant",
+                        "Monochrome",
+                        "Earth Tone",
+                        "Neon"
+                    ],
+                    value="Pastel"
+                )
+                img_platform = gr.Dropdown(
+                    label="Target Platform",
+                    choices=[
+                        "Midjourney",
+                        "DALL·E",
+                        "Stable Diffusion",
+                        "Canva Design Brief"
+                    ],
+                    value="Midjourney"
+                )
+                img_ratio = gr.Dropdown(
+                    label="Aspect Ratio",
+                    choices=[
+                        "1:1",
+                        "9:16",
+                        "16:9",
+                        "4:5"
+                    ],
+                    value="1:1"
+                )
+
+                img_generate_btn = gr.Button("🚀 Generate Image Design JSON")
+                img_output = gr.Textbox(label="Image Design JSON Output", lines=20, elem_id="image-json-output")
+                img_copy_btn = gr.Button("Copy Image JSON")
+
+            # ============================
+            #  TAB 3: VIDEO AI
+            # ============================
+            with gr.Tab("Video AI Prompt"):
+
+                vid_goal = gr.Textbox(label="Goal / Ide Video AI")
+                vid_style = gr.Dropdown(
+                    label="Video Style",
+                    choices=[
+                        "Cinematic",
+                        "Pastel Kids Animation",
+                        "Anime",
+                        "Realistic",
+                        "Claymation",
+                        "Watercolor Motion"
+                    ],
+                    value="Pastel Kids Animation"
+                )
+                vid_camera = gr.Dropdown(
+                    label="Camera Movement",
+                    choices=[
+                        "Slow Zoom",
+                        "Pan Left",
+                        "Pan Right",
+                        "Dolly In",
+                        "Static Shot"
+                    ],
+                    value="Slow Zoom"
+                )
+                vid_platform = gr.Dropdown(
+                    label="Target Platform",
+                    choices=[
+                        "Pika Labs",
+                        "Runway Gen-2",
+                        "Kaiber",
+                        "Luma Dream Machine"
+                    ],
+                    value="Pika Labs"
+                )
+                vid_duration = gr.Dropdown(
+                    label="Duration",
+                    choices=[
+                        "5 seconds",
+                        "10 seconds",
+                        "15 seconds",
+                        "30 seconds"
+                    ],
+                    value="10 seconds"
+                )
+
+                vid_generate_btn = gr.Button("🚀 Generate Video AI JSON")
+                vid_output = gr.Textbox(label="Video AI JSON Output", lines=20, elem_id="video-json-output")
+                vid_copy_btn = gr.Button("Copy Video JSON")
 
     # LOGIN LOGIC
     def handle_login(username, password):
@@ -226,35 +380,68 @@ with gr.Blocks() as demo:
         [login_msg, login_ui, main_ui, login_status, login_user, login_tier]
     )
 
-    # GENERATE JSON
-    def handle_generate(login_status, login_user, login_tier,
-                        input_mode, text_input, image_input,
-                        kind, tone, audience, length, constraints, language, n):
+    # E-BOOK GENERATE
+    def handle_ebook_generate(login_status, login_user, login_tier,
+                              goal, genre, tone, style, platform, length):
+        if not login_status:
+            return "❌ Anda belum login."
+        return generate_ebook_json(login_user, login_tier, goal, genre, tone, style, platform, length)
+
+    ebook_generate_btn.click(
+        handle_ebook_generate,
+        [login_status, login_user, login_tier,
+         ebook_goal, ebook_genre, ebook_tone, ebook_style, ebook_platform, ebook_length],
+        ebook_output
+    )
 
+    # IMAGE GENERATE
+    def handle_image_generate(login_status, login_user, login_tier,
+                              goal, design_style, color_palette, platform, aspect_ratio):
         if not login_status:
             return "❌ Anda belum login."
+        return generate_image_json(login_user, login_tier, goal, design_style, color_palette, platform, aspect_ratio)
 
-        if input_mode == "text":
-            goal = text_input
-        else:
-            goal = describe_image(image_input)
+    img_generate_btn.click(
+        handle_image_generate,
+        [login_status, login_user, login_tier,
+         img_goal, img_style, img_palette, img_platform, img_ratio],
+        img_output
+    )
 
-        return generate_json(login_user, login_tier, goal, kind, tone, audience, length, constraints, language, n)
+    # VIDEO GENERATE
+    def handle_video_generate(login_status, login_user, login_tier,
+                              goal, video_style, camera_movement, platform, duration):
+        if not login_status:
+            return "❌ Anda belum login."
+        return generate_video_json(login_user, login_tier, goal, video_style, camera_movement, platform, duration)
 
-    generate_btn.click(
-        handle_generate,
+    vid_generate_btn.click(
+        handle_video_generate,
         [login_status, login_user, login_tier,
-         input_mode, text_input, image_input,
-         kind, tone, audience, length, constraints, language, n],
-        output_json
+         vid_goal, vid_style, vid_camera, vid_platform, vid_duration],
+        vid_output
+    )
+
+    # COPY BUTTONS (JS)
+    ebook_copy_btn.click(
+        None,
+        None,
+        None,
+        js="navigator.clipboard.writeText(document.getElementById('ebook-json-output').value)"
+    )
+
+    img_copy_btn.click(
+        None,
+        None,
+        None,
+        js="navigator.clipboard.writeText(document.getElementById('image-json-output').value)"
     )
 
-    # COPY JSON
-    copy_btn.click(
+    vid_copy_btn.click(
         None,
         None,
         None,
-        js="navigator.clipboard.writeText(document.getElementById('json-output').value)"
+        js="navigator.clipboard.writeText(document.getElementById('video-json-output').value)"
     )
 
 # LAUNCH