Update app.py

app.py

@@ -14,12 +14,11 @@ from groq import Groq
 REPLICATE_API_TOKEN = "r8_OHfjZvEbVoU9fuioi4qfQ50OHdpAmqV4OAa5b"
 os.environ["REPLICATE_API_TOKEN"] = REPLICATE_API_TOKEN
 
-# === MASUKKAN API KEY GROQ DI SINI ===
 GROQ_API_KEY = "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"
 client = Groq(api_key=GROQ_API_KEY)
 
 MAX_CHAPTERS = 10
-PANELS_PER_CHAPTER = 4  # komik: 4 panel per bab
+PANELS_PER_CHAPTER = 8  # sesuai permintaan
 
 # ============================
 #  USER DB
@@ -48,7 +47,7 @@ def ensure_outputs_dir():
     os.makedirs("outputs", exist_ok=True)
 
 # ============================
-#  AI TEXT GENERATOR (GROQ – KOMIK PANEL)
+#  AI TEXT GENERATOR (GROQ)
 # ============================
 def ai_generate(prompt):
     try:
@@ -61,6 +60,9 @@ def ai_generate(prompt):
     except Exception as e:
         return f"AI gagal menghasilkan teks. Error: {str(e)}"
 
+# ============================
+#  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.
@@ -72,15 +74,15 @@ def generate_comic_panels(goal, style, tone, chapter_title, panels=PANELS_PER_CH
     Tone: {tone}
     Target: anak-anak SD, Islami, lembut, positif.
 
-    Format output HARUS seperti ini (tanpa penjelasan tambahan):
+    Format output HARUS seperti ini:
 
-    Panel 1: [deskripsi singkat adegan + dialog/narasi]
-    Panel 2: [deskripsi singkat adegan + dialog/narasi]
+    Panel 1: ...
+    Panel 2: ...
     Panel 3: ...
+    ...
     Panel {panels}: ...
 
-    Jangan pakai bullet, jangan pakai tanda bintang, jangan pakai heading.
-    Hanya baris-baris Panel X: ...
+    Jangan pakai bullet, jangan pakai heading.
     """
     raw = ai_generate(prompt)
     lines = [l.strip() for l in raw.split("\n") if l.strip()]
@@ -93,7 +95,24 @@ def generate_comic_panels(goal, style, tone, chapter_title, panels=PANELS_PER_CH
     return panels_text[:panels]
 
 # ============================
-#  REPLICATE IMAGE MODELS
+#  VISUAL PROMPT GENERATOR
+# ============================
+def convert_panel_to_visual_prompt(panel_text, character_mode, character_desc):
+    """
+    Mengubah teks panel menjadi prompt visual pastel cartoon.
+    """
+    base_style = (
+        "Cute pastel cartoon illustration, soft colors, round shapes, kid-friendly, "
+        "Islamic-friendly, simple background, environmental setting, high quality."
+    )
+
+    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}."
+
+# ============================
+#  IMAGE GENERATOR (FLUX)
 # ============================
 def generate_image_flux(prompt):
     try:
@@ -107,18 +126,6 @@ def generate_image_flux(prompt):
         return ""
     return ""
 
-def generate_image_sdxl(prompt):
-    try:
-        out = replicate.run(
-            "stability-ai/sdxl-lite",
-            input={"prompt": prompt}
-        )
-        if isinstance(out, list) and out:
-            return out[0]
-    except:
-        return ""
-    return ""
-
 def download_image(url, filename):
     if not url:
         return ""
@@ -134,9 +141,9 @@ def download_image(url, filename):
         return ""
 
 # ============================
-#  PDF BUILDER (KOMIK)
+#  PDF BUILDER (COMIC)
 # ============================
-class EbookPDF(FPDF):
+class ComicPDF(FPDF):
     def header(self):
         if self.page_no() == 1:
             return
@@ -149,17 +156,17 @@ class EbookPDF(FPDF):
         self.set_font("Helvetica", "I", 9)
         self.cell(0, 10, f"Page {self.page_no()}", 0, 0, "C")
 
-def build_ebook_pdf_with_images(username, goal, genre, tone, style, length,
-                                cover_brief, chapters_count):
+def build_comic_pdf(username, goal, genre, tone, style, cover_brief,
+                    chapters_count, character_mode, character_desc, comic_name):
 
     ensure_outputs_dir()
-    pdf = EbookPDF(format="A4")
+    pdf = ComicPDF(format="A4")
     pdf.set_margins(15, 15, 15)
 
-    # COVER IMAGE (lebih realistik tapi tetap ramah anak)
+    # COVER
     cover_prompt = (
-        f"{cover_brief} | {goal} | {genre} | {style} | "
-        "realistic comic cover, soft lighting, kid-friendly, Islamic, high detail"
+        f"Cute pastel cartoon cover, soft colors, kid-friendly, Islamic-friendly, "
+        f"{cover_brief}, theme: {goal}"
     )
     cover_url = generate_image_flux(cover_prompt)
     cover_path = download_image(cover_url, f"cover_{uuid.uuid4().hex}.png")
@@ -171,15 +178,14 @@ def build_ebook_pdf_with_images(username, goal, genre, tone, style, length,
         pdf.image(cover_path, x=0, y=0, w=210)
         pdf.set_y(210)
     else:
-        pdf.set_fill_color(230, 240, 255)
+        pdf.set_fill_color(240, 230, 255)
         pdf.rect(0, 0, 210, 297, "F")
         pdf.set_y(50)
 
     pdf.set_auto_page_break(True, margin=15)
     pdf.set_y(max(pdf.get_y(), 50))
 
-    # COVER TEXT
-    pdf.set_font("Helvetica", "B", 20)
+    pdf.set_font("Helvetica", "B", 22)
     pdf.multi_cell(180, 10, safe_text(goal), 0)
 
     pdf.set_font("Helvetica", "", 12)
@@ -192,92 +198,64 @@ def build_ebook_pdf_with_images(username, goal, genre, tone, style, length,
     pdf.cell(0, 10, "Daftar Isi (Komik)", 0, 1)
     pdf.ln(5)
 
-    pdf.set_font("Helvetica", "", 12)
     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 → KOMIK PANEL
+    # CHAPTERS
     for idx, ch in enumerate(chapters, start=1):
         pdf.add_page()
         pdf.set_font("Helvetica", "B", 16)
         pdf.cell(0, 10, safe_text(ch), 0, 1)
         pdf.ln(4)
 
-        # Generate panel texts (Groq)
+        # Generate panels
         panels_text = generate_comic_panels(goal, style, tone, ch, PANELS_PER_CHAPTER)
 
-        # Untuk tiap panel: gambar + teks
         for p_idx, panel in enumerate(panels_text, start=1):
-            # ILLUSTRATION (lebih realistik, gaya komik)
-            illus_prompt = (
-                f"Realistic comic panel illustration of an Islamic children's story about {goal}. "
-                f"Scene: {panel}. Style: semi-realistic, soft colors, expressive faces, "
-                f"cinematic lighting, kid-friendly, high detail."
+
+            # Visual prompt
+            visual_prompt = convert_panel_to_visual_prompt(
+                panel,
+                character_mode,
+                character_desc
+            )
+
+            # Generate image
+            illus_url = generate_image_flux(visual_prompt)
+            illus_path = download_image(
+                illus_url,
+                f"chapter_{idx}_panel_{p_idx}_{uuid.uuid4().hex}.png"
             )
-            illus_url = generate_image_flux(illus_prompt)
-            illus_path = download_image(illus_url, f"chapter_{idx}_panel_{p_idx}_{uuid.uuid4().hex}.png")
 
+            # Panel title
             pdf.set_font("Helvetica", "B", 12)
             pdf.cell(0, 8, safe_text(f"Panel {p_idx}"), 0, 1)
             pdf.ln(1)
 
+            # Image
             if illus_path:
                 y_start = pdf.get_y()
-                # Lebar panel komik
                 pdf.image(illus_path, x=20, y=y_start, w=170)
                 pdf.set_y(y_start + 80)
 
+            # Text
             pdf.set_font("Helvetica", "", 11)
             pdf.multi_cell(180, 6, safe_text(panel), 0)
             pdf.ln(3)
 
-    filename = os.path.join("outputs", f"comic_{uuid.uuid4().hex}.pdf")
+    # OUTPUT FILE
+    if comic_name.strip():
+        filename = os.path.join("outputs", f"{comic_name}.pdf")
+    else:
+        filename = os.path.join("outputs", "comic_output.pdf")
+
     pdf.output(filename)
     return filename
 
-# ============================
-#  E-BOOK JSON GENERATOR (MASIH BISA DIPAKAI)
-# ============================
-def generate_ebook_json(username, tier,
-                        goal, genre, tone, style,
-                        platform, length, chapters):
-
-    goal = normalize(goal, "Islamic children's story about good manners.")
-    genre = normalize(genre, "Islamic Children Story (Comic)")
-    tone = normalize(tone, "Friendly")
-    style = normalize(style, "Comic style")
-    platform = normalize(platform, "KDP")
-    length = normalize(length, "Medium (30–50 pages)")
-
-    try:
-        chapters = int(chapters)
-    except:
-        chapters = 5
-
-    chapters = max(1, min(MAX_CHAPTERS, chapters))
-
-    chapter_titles = [f"Bab {i}: Judul Bab {i}" for i in range(1, chapters + 1)]
-
-    data = {
-        "type": "comic",
-        "generated_at": now_iso(),
-        "user": username,
-        "tier": tier,
-        "platform": platform,
-        "genre": genre,
-        "tone": tone,
-        "writing_style": style,
-        "length": length,
-        "goal": goal,
-        "chapters": chapter_titles,
-        "panels_per_chapter": PANELS_PER_CHAPTER,
-    }
-
-    return json.dumps(data, ensure_ascii=False, indent=2)
-
 # ============================
 #  LOGIN
 # ============================
@@ -295,6 +273,7 @@ with gr.Blocks() as demo:
     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")
@@ -302,72 +281,54 @@ with gr.Blocks() as demo:
         login_btn = gr.Button("Login")
         login_msg = gr.Markdown("")
 
+    # MAIN UI
     with gr.Group(visible=False) as main_ui:
 
-        gr.Markdown("# 🌟 AIPromptLab — Comic JSON & PDF Generator (Groq + Replicate)")
+        gr.Markdown("# 🌟 AIPromptLab — Comic Generator 2.0 (Cartoon Pastel Edition)")
 
         with gr.Tabs():
 
-            with gr.Tab("Comic Prompt & PDF"):
+            with gr.Tab("Comic Generator"):
+
+                comic_name = gr.Textbox(label="Nama Komik (tanpa .pdf)")
+
+                comic_goal = gr.Textbox(label="Goal / Ide Cerita Komik")
 
-                ebook_goal = gr.Textbox(label="Goal / Ide Komik")
-                ebook_genre = gr.Dropdown(
+                comic_genre = gr.Dropdown(
                     label="Genre",
                     choices=[
                         "Islamic Children Comic",
-                        "Islamic Adventure",
-                        "Islamic Moral Story",
-                        "Fantasy",
-                        "Education",
                         "Adventure",
-                        "Mystery"
+                        "Fantasy",
+                        "Moral Story",
+                        "Education"
                     ],
                     value="Islamic Children Comic"
                 )
-                ebook_tone = gr.Dropdown(
+
+                comic_tone = gr.Dropdown(
                     label="Tone",
                     choices=[
                         "Friendly",
                         "Inspirational",
                         "Educational",
-                        "Storytelling",
-                        "Humorous",
-                        "Calm"
+                        "Calm",
+                        "Humorous"
                     ],
                     value="Friendly"
                 )
-                ebook_style = gr.Dropdown(
+
+                comic_style = gr.Dropdown(
                     label="Comic Style",
                     choices=[
-                        "Semi-realistic comic",
-                        "Cartoon comic",
-                        "Pastel Islamic comic",
-                        "Manga-like",
-                        "Webtoon style"
+                        "Cartoon Pastel",
+                        "Cute Cartoon",
+                        "Soft Pastel"
                     ],
-                    value="Semi-realistic comic"
-                )
-                ebook_platform = gr.Dropdown(
-                    label="Target Platform",
-                    choices=[
-                        "KDP",
-                        "Webtoon",
-                        "PDF / E-Learning",
-                        "Print"
-                    ],
-                    value="PDF / E-Learning"
-                )
-                ebook_length = gr.Dropdown(
-                    label="Length",
-                    choices=[
-                        "Short (1–2 chapters)",
-                        "Medium (3–5 chapters)",
-                        "Long (6–10 chapters)"
-                    ],
-                    value="Medium (3–5 chapters)"
+                    value="Cartoon Pastel"
                 )
 
-                ebook_chapters = gr.Slider(
+                chapters_slider = gr.Slider(
                     label="Jumlah Bab Komik",
                     minimum=1,
                     maximum=MAX_CHAPTERS,
@@ -375,18 +336,36 @@ with gr.Blocks() as demo:
                     value=1
                 )
 
-                ebook_cover_brief = gr.Textbox(
+                cover_brief = gr.Textbox(
                     label="Cover Illustration Brief",
-                    value="Realistic comic cover of Islamic children in adventure, soft colors, warm light, friendly faces."
+                    value="Cute pastel cartoon cover of Islamic children adventure."
+                )
+
+                # CHARACTER MODE
+                character_mode = gr.Dropdown(
+                    label="Mode Karakter",
+                    choices=["Otomatis", "Custom"],
+                    value="Otomatis"
                 )
 
-                ebook_generate_btn = gr.Button("🚀 Generate Comic JSON")
-                ebook_output = gr.Textbox(label="Comic JSON Output", lines=18)
+                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]
+                )
 
-                gr.Markdown("### 📄 Generate PDF Komik (A4 + Cover + Panel Komik)")
-                ebook_pdf_btn = gr.Button("Generate Comic PDF")
-                ebook_pdf_file = gr.File(label="Download Comic PDF")
+                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 (
@@ -404,53 +383,37 @@ with gr.Blocks() as demo:
         [login_msg, login_ui, main_ui, login_status, login_user, login_tier]
     )
 
-    def handle_ebook_generate(login_status, login_user, login_tier,
-                              goal, genre, tone, style, platform, length, chapters):
-        if not login_status:
-            return "❌ Anda belum login."
-        return generate_ebook_json(
-            login_user, login_tier,
-            goal, genre, tone, style,
-            platform, length, chapters
-        )
+    # PDF GENERATOR HANDLER
+    def handle_generate_pdf(login_status, login_user,
+                            comic_name, goal, genre, tone, style,
+                            chapters, cover_brief, character_mode, character_desc):
 
-    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_chapters
-        ],
-        ebook_output
-    )
-
-    def handle_ebook_pdf(login_status, login_user,
-                         goal, genre, tone, style, length, cover_brief, chapters):
         if not login_status:
             return None
 
-        goal_ = normalize(goal, "Islamic children's comic about good manners.")
-        genre_ = normalize(genre, "Islamic Children Comic")
-        tone_ = normalize(tone, "Friendly")
-        style_ = normalize(style, "Semi-realistic comic")
-        length_ = normalize(length, "Medium (3–5 chapters)")
-        cover_ = normalize(cover_brief, "Realistic Islamic children comic cover.")
-
-        pdf_path = build_ebook_pdf_with_images(
-            login_user, goal_, genre_, tone_, style_, length_, cover_, chapters
+        pdf_path = build_comic_pdf(
+            login_user,
+            goal,
+            genre,
+            tone,
+            style,
+            cover_brief,
+            chapters,
+            character_mode,
+            character_desc,
+            comic_name
         )
         return pdf_path
 
-    ebook_pdf_btn.click(
-        handle_ebook_pdf,
+    generate_pdf_btn.click(
+        handle_generate_pdf,
         [
             login_status, login_user,
-            ebook_goal, ebook_genre, ebook_tone,
-            ebook_style, ebook_length, ebook_cover_brief,
-            ebook_chapters
+            comic_name, comic_goal, comic_genre, comic_tone,
+            comic_style, chapters_slider, cover_brief,
+            character_mode, character_desc
         ],
-        ebook_pdf_file
+        pdf_output
     )
 
 demo.launch(