Update app.py

app.py

@@ -19,6 +19,7 @@ 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
 
 # ============================
 #  USER DB
@@ -47,30 +48,49 @@ def ensure_outputs_dir():
     os.makedirs("outputs", exist_ok=True)
 
 # ============================
-#  AI TEXT GENERATOR (GROQ LLaMA‑3 70B)
+#  AI TEXT GENERATOR (GROQ – KOMIK PANEL)
 # ============================
 def ai_generate(prompt):
     try:
         response = client.chat.completions.create(
             model="llama-3.3-70b-versatile",
             messages=[{"role": "user", "content": prompt}],
-            max_tokens=600
+            max_tokens=800
         )
         return response.choices[0].message.content
     except Exception as e:
         return f"AI gagal menghasilkan teks. Error: {str(e)}"
 
-def generate_chapter_text(goal, style, tone, chapter_title):
+def generate_comic_panels(goal, style, tone, chapter_title, panels=PANELS_PER_CHAPTER):
     prompt = f"""
-    Buatkan 1 bab cerita anak Islami.
+    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}
-    Panjang: 4 paragraf.
-    Bahasa Indonesia.
+    Target: anak-anak SD, Islami, lembut, positif.
+
+    Format output HARUS seperti ini (tanpa penjelasan tambahan):
+
+    Panel 1: [deskripsi singkat adegan + dialog/narasi]
+    Panel 2: [deskripsi singkat adegan + dialog/narasi]
+    Panel 3: ...
+    Panel {panels}: ...
+
+    Jangan pakai bullet, jangan pakai tanda bintang, jangan pakai heading.
+    Hanya baris-baris Panel X: ...
     """
-    return ai_generate(prompt)
+    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]
 
 # ============================
 #  REPLICATE IMAGE MODELS
@@ -114,14 +134,14 @@ def download_image(url, filename):
         return ""
 
 # ============================
-#  PDF BUILDER
+#  PDF BUILDER (KOMIK)
 # ============================
 class EbookPDF(FPDF):
     def header(self):
         if self.page_no() == 1:
             return
         self.set_font("Helvetica", "I", 9)
-        self.cell(0, 8, "AIPromptLab E-Book", 0, 1, "R")
+        self.cell(0, 8, "AIPromptLab Comic", 0, 1, "R")
         self.ln(2)
 
     def footer(self):
@@ -136,8 +156,11 @@ def build_ebook_pdf_with_images(username, goal, genre, tone, style, length,
     pdf = EbookPDF(format="A4")
     pdf.set_margins(15, 15, 15)
 
-    # COVER IMAGE
-    cover_prompt = f"{cover_brief} | {goal} | {genre} | {style} | pastel, kid-friendly, Islamic"
+    # COVER IMAGE (lebih realistik tapi tetap ramah anak)
+    cover_prompt = (
+        f"{cover_brief} | {goal} | {genre} | {style} | "
+        "realistic comic cover, soft lighting, kid-friendly, Islamic, high detail"
+    )
     cover_url = generate_image_flux(cover_prompt)
     cover_path = download_image(cover_url, f"cover_{uuid.uuid4().hex}.png")
 
@@ -160,13 +183,13 @@ def build_ebook_pdf_with_images(username, goal, genre, tone, style, length,
     pdf.multi_cell(180, 10, safe_text(goal), 0)
 
     pdf.set_font("Helvetica", "", 12)
-    pdf.multi_cell(180, 7, safe_text(f"Genre: {genre}"), 0)
+    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", 0, 1)
+    pdf.cell(0, 10, "Daftar Isi (Komik)", 0, 1)
     pdf.ln(5)
 
     pdf.set_font("Helvetica", "", 12)
@@ -176,48 +199,56 @@ def build_ebook_pdf_with_images(username, goal, genre, tone, style, length,
     for i, ch in enumerate(chapters, start=1):
         pdf.cell(0, 8, safe_text(f"{i}. {ch}"), 0, 1)
 
-    # CHAPTERS (AI TEXT)
+    # CHAPTERS → KOMIK PANEL
     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)
 
-        # ILLUSTRATION
-        illus_prompt = (
-            f"Illustration for chapter {idx} of an Islamic children's story about {goal}. "
-            f"Scene: {ch}. Style: {style}, tone {tone}, pastel colors."
-        )
-        illus_url = generate_image_sdxl(illus_prompt)
-        illus_path = download_image(illus_url, f"chapter_{idx}_{uuid.uuid4().hex}.png")
-
-        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)
-
-        # AI CHAPTER TEXT (GROQ)
-        chapter_text = generate_chapter_text(goal, style, tone, ch)
-
-        pdf.set_font("Helvetica", "", 12)
-        pdf.multi_cell(180, 7, safe_text(chapter_text), 0)
-        pdf.ln(2)
-
-    filename = os.path.join("outputs", f"ebook_{uuid.uuid4().hex}.pdf")
+        # Generate panel texts (Groq)
+        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."
+            )
+            illus_url = generate_image_flux(illus_prompt)
+            illus_path = download_image(illus_url, f"chapter_{idx}_panel_{p_idx}_{uuid.uuid4().hex}.png")
+
+            pdf.set_font("Helvetica", "B", 12)
+            pdf.cell(0, 8, safe_text(f"Panel {p_idx}"), 0, 1)
+            pdf.ln(1)
+
+            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)
+
+            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")
     pdf.output(filename)
     return filename
 
 # ============================
-#  E-BOOK JSON GENERATOR
+#  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")
+    genre = normalize(genre, "Islamic Children Story (Comic)")
     tone = normalize(tone, "Friendly")
-    style = normalize(style, "Storybook style")
+    style = normalize(style, "Comic style")
     platform = normalize(platform, "KDP")
     length = normalize(length, "Medium (30–50 pages)")
 
@@ -231,7 +262,7 @@ def generate_ebook_json(username, tier,
     chapter_titles = [f"Bab {i}: Judul Bab {i}" for i in range(1, chapters + 1)]
 
     data = {
-        "type": "ebook",
+        "type": "comic",
         "generated_at": now_iso(),
         "user": username,
         "tier": tier,
@@ -242,6 +273,7 @@ def generate_ebook_json(username, tier,
         "length": length,
         "goal": goal,
         "chapters": chapter_titles,
+        "panels_per_chapter": PANELS_PER_CHAPTER,
     }
 
     return json.dumps(data, ensure_ascii=False, indent=2)
@@ -272,25 +304,25 @@ with gr.Blocks() as demo:
 
     with gr.Group(visible=False) as main_ui:
 
-        gr.Markdown("# 🌟 AIPromptLab — JSON & PDF E-Book Generator (Groq + Replicate)")
+        gr.Markdown("# 🌟 AIPromptLab — Comic JSON & PDF Generator (Groq + Replicate)")
 
         with gr.Tabs():
 
-            with gr.Tab("E-Book Prompt & PDF"):
+            with gr.Tab("Comic Prompt & PDF"):
 
-                ebook_goal = gr.Textbox(label="Goal / Ide E-book")
+                ebook_goal = gr.Textbox(label="Goal / Ide Komik")
                 ebook_genre = gr.Dropdown(
                     label="Genre",
                     choices=[
-                        "Islamic Children Story",
-                        "Parenting",
-                        "Motivation",
+                        "Islamic Children Comic",
+                        "Islamic Adventure",
+                        "Islamic Moral Story",
                         "Fantasy",
                         "Education",
                         "Adventure",
                         "Mystery"
                     ],
-                    value="Islamic Children Story"
+                    value="Islamic Children Comic"
                 )
                 ebook_tone = gr.Dropdown(
                     label="Tone",
@@ -299,62 +331,61 @@ with gr.Blocks() as demo:
                         "Inspirational",
                         "Educational",
                         "Storytelling",
-                        "Formal",
-                        "Casual"
+                        "Humorous",
+                        "Calm"
                     ],
                     value="Friendly"
                 )
                 ebook_style = gr.Dropdown(
-                    label="Writing Style",
+                    label="Comic Style",
                     choices=[
-                        "Storybook style",
-                        "Narrative",
-                        "Descriptive",
-                        "Persuasive",
-                        "Expository",
-                        "Dialog-based"
+                        "Semi-realistic comic",
+                        "Cartoon comic",
+                        "Pastel Islamic comic",
+                        "Manga-like",
+                        "Webtoon style"
                     ],
-                    value="Storybook style"
+                    value="Semi-realistic comic"
                 )
                 ebook_platform = gr.Dropdown(
                     label="Target Platform",
                     choices=[
                         "KDP",
-                        "Wattpad",
-                        "Google Play Books",
-                        "PDF / E-Learning"
+                        "Webtoon",
+                        "PDF / E-Learning",
+                        "Print"
                     ],
-                    value="KDP"
+                    value="PDF / E-Learning"
                 )
                 ebook_length = gr.Dropdown(
                     label="Length",
                     choices=[
-                        "Short (10–20 pages)",
-                        "Medium (30–50 pages)",
-                        "Long (80–120 pages)"
+                        "Short (1–2 chapters)",
+                        "Medium (3–5 chapters)",
+                        "Long (6–10 chapters)"
                     ],
-                    value="Medium (30–50 pages)"
+                    value="Medium (3–5 chapters)"
                 )
 
                 ebook_chapters = gr.Slider(
-                    label="Jumlah Bab (dengan ilustrasi)",
+                    label="Jumlah Bab Komik",
                     minimum=1,
                     maximum=MAX_CHAPTERS,
                     step=1,
-                    value=5
+                    value=1
                 )
 
                 ebook_cover_brief = gr.Textbox(
                     label="Cover Illustration Brief",
-                    value="Cute pastel Islamic illustration of children reading together under warm light, soft colors, friendly style."
+                    value="Realistic comic cover of Islamic children in adventure, soft colors, warm light, friendly faces."
                 )
 
-                ebook_generate_btn = gr.Button("🚀 Generate E-book JSON")
-                ebook_output = gr.Textbox(label="E-book JSON Output", lines=18)
+                ebook_generate_btn = gr.Button("🚀 Generate Comic JSON")
+                ebook_output = gr.Textbox(label="Comic JSON Output", lines=18)
 
-                gr.Markdown("### 📄 Generate PDF E-book (A4 + Cover + Ilustrasi Bab)")
-                ebook_pdf_btn = gr.Button("Generate PDF E-book")
-                ebook_pdf_file = gr.File(label="Download E-book PDF")
+                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")
 
     def handle_login(username, password):
         ok, tier, msg = login(username, password)
@@ -399,12 +430,12 @@ with gr.Blocks() as demo:
         if not login_status:
             return None
 
-        goal_ = normalize(goal, "Islamic children's story about good manners.")
-        genre_ = normalize(genre, "Islamic Children Story")
+        goal_ = normalize(goal, "Islamic children's comic about good manners.")
+        genre_ = normalize(genre, "Islamic Children Comic")
         tone_ = normalize(tone, "Friendly")
-        style_ = normalize(style, "Storybook style")
-        length_ = normalize(length, "Medium (30–50 pages)")
-        cover_ = normalize(cover_brief, "Pastel Islamic illustration of children reading together.")
+        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