Update app.py

app.py

@@ -1,451 +1,224 @@
 import gradio as gr
-from datetime import datetime
-from fpdf import FPDF
-import os
-import uuid
-import requests
 import replicate
+import uuid
+import os
 from groq import Groq
+from fpdf import FPDF
 
 # ============================
-#  CONFIG
+# CONFIG
 # ============================
-REPLICATE_API_TOKEN = "r8_OHfjZvEbVoU9fuioi4qfQ50OHdpAmqV4OAa5b"
-os.environ["REPLICATE_API_TOKEN"] = REPLICATE_API_TOKEN
-
-GROQ_API_KEY = "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"
-client = Groq(api_key=GROQ_API_KEY)
-
-MAX_CHAPTERS = 10
-PANELS_PER_CHAPTER = 8  # sesuai permintaan
+os.environ["REPLICATE_API_TOKEN"] = "r8_Ap7YprjUOrZ7chAcXy2m1HAEDpFRbwV1dIM2Y"   # GANTI DENGAN TOKENMU
+client = Groq(api_key="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")
 
 # ============================
-#  USER DB
+# NANO BANANA ENGINE
 # ============================
-USER_DB = {
-    "akhmad": {"password": "12345", "tier": "super"},
-    "premium_user": {"password": "abcde", "tier": "premium"},
-    "free_user": {"password": "00000", "tier": "free"},
-}
+def nano_banana(prompt, aspect_ratio="1:1", resolution="1K"):
+    output = replicate.run(
+        "google/nano-banana-2",
+        input={
+            "prompt": prompt,
+            "resolution": resolution,
+            "image_input": [],
+            "aspect_ratio": aspect_ratio,
+            "image_search": False,
+            "google_search": False,
+            "output_format": "jpg"
+        }
+    )
 
-# ============================
-#  HELPERS
-# ============================
-def safe_text(text):
-    if not text:
-        return ""
-    return ''.join(c if 32 <= ord(c) <= 126 else ' ' for c in str(text))
+    if not hasattr(output, "url"):
+        return None, None
+
+    url = output.url
 
-def ensure_outputs_dir():
     os.makedirs("outputs", exist_ok=True)
+    filename = f"nano_{uuid.uuid4().hex}.jpg"
+    filepath = os.path.join("outputs", filename)
 
-# ============================
-#  AI TEXT GENERATOR (GROQ)
-# ============================
-def ai_generate(prompt):
-    try:
-        response = client.chat.completions.create(
-            model="llama-3.3-70b-versatile",
-            messages=[{"role": "user", "content": prompt}],
-            max_tokens=800
-        )
-        return response.choices[0].message.content
-    except Exception as e:
-        return f"AI gagal menghasilkan teks. Error: {str(e)}"
+    with open(filepath, "wb") as f:
+        f.write(output.read())
 
-# ============================
-#  PANEL GENERATOR
-# ============================
-def generate_comic_panels(goal, style, tone, chapter_title, panels=PANELS_PER_CHAPTER):
-    prompt = f"""
-    Kamu adalah penulis komik anak Islami berbahasa Indonesia.
-
-    Buatkan {panels} panel komik untuk satu bab cerita.
-    Judul bab: {chapter_title}
-    Tema utama: {goal}
-    Gaya: {style}
-    Tone: {tone}
-    Target: anak-anak SD, Islami, lembut, positif.
-
-    Format output HARUS seperti ini:
-
-    Panel 1: ...
-    Panel 2: ...
-    Panel 3: ...
-    ...
-    Panel {panels}: ...
-
-    Jangan pakai bullet, jangan pakai heading.
-    """
-    raw = ai_generate(prompt)
-    lines = [l.strip() for l in raw.split("\n") if l.strip()]
-    panels_text = []
-    for l in lines:
-        if l.lower().startswith("panel"):
-            panels_text.append(l)
-    if not panels_text:
-        panels_text = [f"Panel {i+1}: {raw}" for i in range(panels)]
-    return panels_text[:panels]
+    return url, filepath
 
 # ============================
-#  VISUAL PROMPT GENERATOR
+# GROQ TEXT GENERATOR
 # ============================
-def convert_panel_to_visual_prompt(panel_text, character_mode, character_desc):
-    base_style = (
-        "Cute pastel cartoon illustration, soft colors, round shapes, kid-friendly, "
-        "Islamic-friendly, simple background, environmental setting, high quality."
+def ai(prompt):
+    res = client.chat.completions.create(
+        model="llama-3.3-70b-versatile",
+        messages=[{"role": "user", "content": prompt}],
+        max_tokens=800
     )
-    if character_mode == "Custom" and character_desc.strip():
-        return f"{base_style} Featuring: {character_desc}. Scene: {panel_text}."
-    return f"{base_style} Scene: {panel_text}."
+    return res.choices[0].message.content
 
 # ============================
-#  IMAGE MODEL ROUTER (16 MODEL)
+# COMIC PANEL GENERATOR
 # ============================
+def generate_panels(story, style, chapters):
+    panels_per_chapter = 8
+    all_chapters = []
+
+    for c in range(1, chapters+1):
+        prompt = f"""
+        Buatkan {panels_per_chapter} panel komik.
+        Cerita: {story}
+        Style visual: {style}
+        Format:
+        Panel 1: ...
+        Panel 2: ...
+        ...
+        Panel {panels_per_chapter}: ...
+        """
+        raw = ai(prompt)
+        lines = [l for l in raw.split("\n") if l.strip().startswith("Panel")]
+        all_chapters.append(lines[:panels_per_chapter])
 
-MODEL_MAP = {
-    # Nama pendek -> ID model Replicate
-    "Pastel-Mix": "fofr/pastel-mix",
-    "SDXL": "stability-ai/sdxl",
-    "Flux Schnell": "black-forest-labs/flux-schnell",
-    "Flux Schnell Free": "black-forest-labs/flux-schnell-free",
-    "Flux Pro": "black-forest-labs/flux-1.1-pro",
-    "Flux Max": "black-forest-labs/flux-1.1-max",
-    "Flux Dev": "black-forest-labs/flux-1.1-dev",
-    "Flux Flex": "black-forest-labs/flux-1.1-flex",
-    "Seedream 4.5": "bytedance/seedream-4.5",
-    "Seedream 5 Lite": "bytedance/seedream-5-lite",
-    "Ideogram Turbo": "ideogram-ai/ideogram-v3-turbo",
-    "Ideogram Balanced": "ideogram-ai/ideogram-v3-balanced",
-    "Ideogram Quality": "ideogram-ai/ideogram-v3-quality",
-    "Recraft V4": "recraft-ai/recraft-v4",
-    "Recraft V4 SVG": "recraft-ai/recraft-v4-svg",
-    "Imagen Fast": "google/imagen-4-fast",
-}
-
-def generate_image_generic(model_id, prompt):
-    try:
-        out = replicate.run(model_id, input={"prompt": prompt})
-        # Beberapa model mengembalikan list, beberapa dict
-        if isinstance(out, list) and out:
-            return out[0]
-        if isinstance(out, dict):
-            # coba beberapa kemungkinan key umum
-            for key in ["output", "image", "images"]:
-                if key in out and out[key]:
-                    if isinstance(out[key], list):
-                        return out[key][0]
-                    return out[key]
-    except Exception:
-        return ""
-    return ""
-
-def generate_image_by_model(prompt, model_name):
-    model_id = MODEL_MAP.get(model_name)
-    if not model_id:
-        # fallback ke Pastel-Mix
-        model_id = MODEL_MAP["Pastel-Mix"]
-    url = generate_image_generic(model_id, prompt)
-    if not url:
-        # fallback kedua: Pastel-Mix
-        if model_name != "Pastel-Mix":
-            url = generate_image_generic(MODEL_MAP["Pastel-Mix"], prompt)
-    return url
-
-def download_image(url, filename):
-    if not url:
-        return ""
-    ensure_outputs_dir()
-    path = os.path.join("outputs", filename)
-    try:
-        r = requests.get(url, timeout=60)
-        r.raise_for_status()
-        with open(path, "wb") as f:
-            f.write(r.content)
-        return path
-    except Exception:
-        return ""
+    return all_chapters
 
 # ============================
-#  PDF BUILDER (COMIC)
+# VISUAL PROMPT BUILDER
 # ============================
-class ComicPDF(FPDF):
-    def header(self):
-        if self.page_no() == 1:
-            return
-        self.set_font("Helvetica", "I", 9)
-        self.cell(0, 8, "AIPromptLab Comic", 0, 1, "R")
-        self.ln(2)
-
-    def footer(self):
-        self.set_y(-15)
-        self.set_font("Helvetica", "I", 9)
-        self.cell(0, 10, f"Page {self.page_no()}", 0, 0, "C")
-
-def build_comic_pdf(username, goal, genre, tone, style, cover_brief,
-                    chapters_count, character_mode, character_desc,
-                    comic_name, image_model):
-
-    ensure_outputs_dir()
-    pdf = ComicPDF(format="A4")
-    pdf.set_margins(15, 15, 15)
-
-    # COVER
-    cover_prompt = (
-        f"Cute pastel cartoon cover, soft colors, kid-friendly, Islamic-friendly, "
-        f"{cover_brief}, theme: {goal}"
-    )
-    cover_url = generate_image_by_model(cover_prompt, image_model)
-    cover_path = download_image(cover_url, f"cover_{uuid.uuid4().hex}.png")
-
-    pdf.add_page()
-    pdf.set_auto_page_break(False)
+def build_visual_prompt(panel_text, style):
+    return f"""
+    {style}, isometric 3D miniature diorama, soft pastel lighting,
+    cinematic depth, Nano Banana 2 optimized.
+    Scene: {panel_text}
+    """
 
-    if cover_path:
-        pdf.image(cover_path, x=0, y=0, w=210)
-        pdf.set_y(210)
-    else:
-        pdf.set_fill_color(240, 230, 255)
-        pdf.rect(0, 0, 210, 297, "F")
-        pdf.set_y(50)
+# ============================
+# PDF BUILDER
+# ============================
+class ComicPDF(FPDF):
+    pass
 
+def build_pdf(story, style, chapters):
+    pdf = ComicPDF()
     pdf.set_auto_page_break(True, margin=15)
-    pdf.set_y(max(pdf.get_y(), 50))
-
-    pdf.set_font("Helvetica", "B", 22)
-    pdf.multi_cell(180, 10, safe_text(goal), 0)
-
-    pdf.set_font("Helvetica", "", 12)
-    pdf.multi_cell(180, 7, safe_text(f"Genre: {genre} (Comic)"), 0)
-    pdf.multi_cell(180, 7, safe_text(f"Tone: {tone} | Style: {style}"), 0)
 
-    # TABLE OF CONTENTS
-    pdf.add_page()
-    pdf.set_font("Helvetica", "B", 18)
-    pdf.cell(0, 10, "Daftar Isi (Komik)", 0, 1)
-    pdf.ln(5)
+    panels = generate_panels(story, style, chapters)
 
-    chapters_count = max(1, min(MAX_CHAPTERS, int(chapters_count)))
-    chapters = [f"Bab {i}: Peristiwa Penting {i}" for i in range(1, chapters_count + 1)]
-
-    pdf.set_font("Helvetica", "", 12)
-    for i, ch in enumerate(chapters, start=1):
-        pdf.cell(0, 8, safe_text(f"{i}. {ch}"), 0, 1)
-
-    # CHAPTERS
-    for idx, ch in enumerate(chapters, start=1):
+    for c_idx, chapter in enumerate(panels, start=1):
         pdf.add_page()
         pdf.set_font("Helvetica", "B", 16)
-        pdf.cell(0, 10, safe_text(ch), 0, 1)
-        pdf.ln(4)
-
-        panels_text = generate_comic_panels(goal, style, tone, ch, PANELS_PER_CHAPTER)
-
-        for p_idx, panel in enumerate(panels_text, start=1):
-            visual_prompt = convert_panel_to_visual_prompt(
-                panel,
-                character_mode,
-                character_desc
-            )
-
-            illus_url = generate_image_by_model(visual_prompt, image_model)
-            illus_path = download_image(
-                illus_url,
-                f"chapter_{idx}_panel_{p_idx}_{uuid.uuid4().hex}.png"
-            )
+        pdf.cell(0, 10, f"Bab {c_idx}", 0, 1)
 
+        for p_idx, panel in enumerate(chapter, start=1):
             pdf.set_font("Helvetica", "B", 12)
-            pdf.cell(0, 8, safe_text(f"Panel {p_idx}"), 0, 1)
-            pdf.ln(1)
+            pdf.cell(0, 8, f"Panel {p_idx}", 0, 1)
 
-            if illus_path:
-                y_start = pdf.get_y()
-                pdf.image(illus_path, x=20, y=y_start, w=170)
-                pdf.set_y(y_start + 80)
+            visual_prompt = build_visual_prompt(panel, style)
+            url, path = nano_banana(visual_prompt)
 
-            pdf.set_font("Helvetica", "", 11)
-            pdf.multi_cell(180, 6, safe_text(panel), 0)
-            pdf.ln(3)
+            if path:
+                pdf.image(path, x=20, w=170)
 
-    if comic_name.strip():
-        filename = os.path.join("outputs", f"{comic_name}.pdf")
-    else:
-        filename = os.path.join("outputs", "comic_output.pdf")
+            pdf.set_font("Helvetica", "", 11)
+            pdf.multi_cell(0, 6, panel)
+            pdf.ln(4)
 
+    filename = f"comic_{uuid.uuid4().hex}.pdf"
     pdf.output(filename)
     return filename
 
 # ============================
-#  LOGIN
+# GRADIO UI
 # ============================
-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."
+with gr.Blocks() as app:
 
-# ============================
-#  UI
-# ============================
-with gr.Blocks() as demo:
-
-    login_status = gr.State(False)
-    login_user = gr.State("")
-    login_tier = gr.State("free")
-
-    # LOGIN UI
-    with gr.Group(visible=True) as login_ui:
-        gr.Markdown("# 🔐 Login untuk Menggunakan AIPromptLab")
-        username = gr.Textbox(label="Username")
-        password = gr.Textbox(label="Password", type="password")
-        login_btn = gr.Button("Login")
-        login_msg = gr.Markdown("")
-
-    # MAIN UI
-    with gr.Group(visible=False) as main_ui:
-
-        gr.Markdown("# 🌟 AIPromptLab — Comic Generator 2.3 (Multi Model Image)")
-
-        with gr.Tabs():
-
-            with gr.Tab("Comic Generator"):
-
-                comic_name = gr.Textbox(label="Nama Komik (tanpa .pdf)")
-
-                comic_goal = gr.Textbox(label="Goal / Ide Cerita Komik")
-
-                comic_genre = gr.Dropdown(
-                    label="Genre",
-                    choices=[
-                        "Islamic Children Comic",
-                        "Adventure",
-                        "Fantasy",
-                        "Moral Story",
-                        "Education"
-                    ],
-                    value="Islamic Children Comic"
-                )
-
-                comic_tone = gr.Dropdown(
-                    label="Tone",
-                    choices=[
-                        "Friendly",
-                        "Inspirational",
-                        "Educational",
-                        "Calm",
-                        "Humorous"
-                    ],
-                    value="Friendly"
-                )
-
-                comic_style = gr.Dropdown(
-                    label="Comic Style",
-                    choices=[
-                        "Cartoon Pastel",
-                        "Cute Cartoon",
-                        "Soft Pastel"
-                    ],
-                    value="Cartoon Pastel"
-                )
-
-                chapters_slider = gr.Slider(
-                    label="Jumlah Bab Komik",
-                    minimum=1,
-                    maximum=MAX_CHAPTERS,
-                    step=1,
-                    value=1
-                )
-
-                cover_brief = gr.Textbox(
-                    label="Cover Illustration Brief",
-                    value="Cute pastel cartoon cover of Islamic children adventure."
-                )
-
-                character_mode = gr.Dropdown(
-                    label="Mode Karakter",
-                    choices=["Otomatis", "Custom"],
-                    value="Otomatis"
-                )
-
-                character_desc = gr.Textbox(
-                    label="Deskripsi Karakter (jika Custom)",
-                    visible=False
-                )
-
-                def toggle_character_desc(mode):
-                    return gr.update(visible=(mode == "Custom"))
-
-                character_mode.change(
-                    toggle_character_desc,
-                    [character_mode],
-                    [character_desc]
-                )
-
-                # IMAGE MODEL DROPDOWN (16 MODEL, A–Z)
-                image_model = gr.Dropdown(
-                    label="Model Gambar",
-                    choices=sorted(list(MODEL_MAP.keys())),
-                    value="Pastel-Mix"
-                )
-
-                generate_pdf_btn = gr.Button("🚀 Generate Comic PDF")
-                pdf_output = gr.File(label="Download Comic PDF")
-
-    # LOGIN HANDLER
-    def handle_login(username, password):
-        ok, tier, msg = login(username, password)
-        return (
-            msg,
-            gr.update(visible=not ok),
-            gr.update(visible=ok),
-            ok,
-            username if ok else "",
-            tier if ok else "free"
-        )
-
-    login_btn.click(
-        handle_login,
-        [username, password],
-        [login_msg, login_ui, main_ui, login_status, login_user, login_tier]
-    )
+    gr.Markdown("# 🌟 AIPromptLab 3.0 — Nano Banana Edition")
 
-    # PDF GENERATOR HANDLER
-    def handle_generate_pdf(login_status, login_user,
-                            comic_name, goal, genre, tone, style,
-                            chapters, cover_brief, character_mode,
-                            character_desc, image_model):
-
-        if not login_status:
-            return None
-
-        pdf_path = build_comic_pdf(
-            login_user,
-            goal,
-            genre,
-            tone,
-            style,
-            cover_brief,
-            chapters,
-            character_mode,
-            character_desc,
-            comic_name,
-            image_model
-        )
-        return pdf_path
-
-    generate_pdf_btn.click(
-        handle_generate_pdf,
-        [
-            login_status, login_user,
-            comic_name, comic_goal, comic_genre, comic_tone,
-            comic_style, chapters_slider, cover_brief,
-            character_mode, character_desc, image_model
-        ],
-        pdf_output
-    )
+    with gr.Tabs():
+
+        # ============================
+        # TAB 1 — COMIC GENERATOR
+        # ============================
+        with gr.Tab("Comic Generator"):
+            story = gr.Textbox(label="Ide Cerita Komik")
+            style = gr.Dropdown(
+                label="Style Visual",
+                choices=[
+                    "Pastel 3D Isometric",
+                    "Cute 3D Cartoon",
+                    "Realistic Cinematic",
+                    "Lowpoly Diorama",
+                    "Claymation 3D",
+                    "Toy Photography",
+                    "Anime 3D Soft Light"
+                ],
+                value="Pastel 3D Isometric"
+            )
+            chapters = gr.Slider(1, 10, value=1, step=1, label="Jumlah Bab")
+            btn = gr.Button("Generate Comic PDF")
+            out = gr.File()
+
+            btn.click(build_pdf, [story, style, chapters], out)
+
+        # ============================
+        # TAB 2 — IMAGE GENERATOR
+        # ============================
+        with gr.Tab("Image Generator"):
+            img_prompt = gr.Textbox(label="Prompt Gambar")
+            img_style = gr.Dropdown(
+                label="Style Visual",
+                choices=[
+                    "Pastel 3D Isometric",
+                    "Cute 3D Cartoon",
+                    "Realistic Cinematic",
+                    "Lowpoly Diorama",
+                    "Claymation 3D",
+                    "Toy Photography",
+                    "Anime 3D Soft Light"
+                ],
+                value="Pastel 3D Isometric"
+            )
+            aspect = gr.Dropdown(["1:1", "16:9", "9:16"], label="Aspect Ratio", value="1:1")
+            res = gr.Dropdown(["1K", "2K"], label="Resolution", value="1K")
+            btn2 = gr.Button("Generate Image")
+            img_out = gr.Image()
+
+            def generate_image(p, s, a, r):
+                prompt = f"{s}, Nano Banana optimized. Scene: {p}"
+                url, path = nano_banana(prompt, a, r)
+                return path
+
+            btn2.click(generate_image, [img_prompt, img_style, aspect, res], img_out)
+
+        # ============================
+        # TAB 3 — VIDEO GENERATOR (STORYBOARD)
+        # ============================
+        with gr.Tab("Video Storyboard"):
+            vid_desc = gr.Textbox(label="Deskripsi Video")
+            vid_style = gr.Dropdown(
+                label="Style Visual",
+                choices=[
+                    "Pastel 3D Isometric",
+                    "Cute 3D Cartoon",
+                    "Realistic Cinematic",
+                    "Lowpoly Diorama",
+                    "Claymation 3D",
+                    "Toy Photography",
+                    "Anime 3D Soft Light"
+                ],
+                value="Pastel 3D Isometric"
+            )
+            btn3 = gr.Button("Generate Storyboard Frames")
+            frame_gallery = gr.Gallery(label="Frames")
+
+            def generate_storyboard(desc, style):
+                prompt = f"Buatkan 8 scene untuk storyboard video. Deskripsi: {desc}"
+                scenes = ai(prompt).split("\n")
+                frames = []
+
+                for s in scenes[:8]:
+                    visual = build_visual_prompt(s, style)
+                    url, path = nano_banana(visual)
+                    if path:
+                        frames.append(path)
+
+                return frames
+
+            btn3.click(generate_storyboard, [vid_desc, vid_style], frame_gallery)
 
-demo.launch(
-    server_name="0.0.0.0",
-    server_port=7860,
-    theme=gr.themes.Soft(),
-    ssr_mode=True
-)
+app.launch()