Update app.py

app.py

@@ -2,6 +2,9 @@
 import gradio as gr
 import json
 from datetime import datetime
+from fpdf import FPDF
+import os
+import uuid
 
 # ============================
 #  USER DATABASE (SIMPLE)
@@ -153,6 +156,88 @@ def generate_video_json(username, tier,
 
     return json.dumps(data, ensure_ascii=False, indent=2)
 
+# ============================
+#  PDF E-BOOK GENERATOR (A4)
+# ============================
+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.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_ebook_pdf(username, goal, genre, tone, style, length, cover_brief):
+    pdf = EbookPDF(format="A4")
+    pdf.set_auto_page_break(auto=True, margin=15)
+
+    # COVER
+    pdf.add_page()
+    pdf.set_fill_color(230, 240, 255)
+    pdf.rect(0, 0, 210, 297, "F")
+
+    pdf.set_font("Helvetica", "B", 28)
+    pdf.set_text_color(40, 40, 70)
+    pdf.ln(60)
+    pdf.multi_cell(0, 15, "E-Book", 0, "C")
+    pdf.ln(5)
+    pdf.set_font("Helvetica", "B", 20)
+    pdf.multi_cell(0, 12, goal, 0, "C")
+    pdf.ln(10)
+    pdf.set_font("Helvetica", "", 12)
+    pdf.multi_cell(0, 8, f"Genre: {genre}", 0, "C")
+    pdf.multi_cell(0, 8, f"Tone: {tone} | Style: {style}", 0, "C")
+    pdf.ln(15)
+    pdf.set_font("Helvetica", "I", 11)
+    pdf.multi_cell(0, 7, "Cover illustration idea:", 0, "C")
+    pdf.ln(3)
+    pdf.set_font("Helvetica", "", 11)
+    pdf.multi_cell(0, 7, cover_brief, 0, "C")
+
+    # TABLE OF CONTENTS
+    pdf.add_page()
+    pdf.set_font("Helvetica", "B", 18)
+    pdf.cell(0, 10, "Daftar Isi", 0, 1)
+    pdf.ln(5)
+    pdf.set_font("Helvetica", "", 12)
+    chapters = [
+        "Bab 1: Pembukaan",
+        "Bab 2: Konflik Utama",
+        "Bab 3: Perjalanan Tokoh",
+        "Bab 4: Puncak Cerita",
+        "Bab 5: Penutup & Hikmah"
+    ]
+    for i, ch in enumerate(chapters, start=1):
+        pdf.cell(0, 8, f"{i}. {ch}", 0, 1)
+
+    # CHAPTERS (SIMPLE GENERATED TEXT)
+    body_paragraph = (
+        f"Cerita ini mengisahkan tentang {goal.lower()}, disampaikan dengan gaya {style.lower()} "
+        f"dan nuansa {tone.lower()}. Setiap bagian dirancang agar mudah dipahami anak-anak, "
+        "mengandung nilai-nilai kebaikan, dan mengajak pembaca untuk merenungkan makna di balik setiap peristiwa."
+    )
+
+    for ch in chapters:
+        pdf.add_page()
+        pdf.set_font("Helvetica", "B", 16)
+        pdf.cell(0, 10, ch, 0, 1)
+        pdf.ln(4)
+        pdf.set_font("Helvetica", "", 12)
+        for _ in range(6):
+            pdf.multi_cell(0, 7, body_paragraph, 0, "J")
+            pdf.ln(2)
+
+    # SAVE TEMP FILE
+    os.makedirs("outputs", exist_ok=True)
+    filename = f"outputs/ebook_{uuid.uuid4().hex}.pdf"
+    pdf.output(filename)
+    return filename
+
 # ============================
 #  LOGIN HANDLER
 # ============================
@@ -182,14 +267,14 @@ with gr.Blocks() as demo:
     # MAIN UI (hidden first)
     with gr.Group(visible=False) as main_ui:
 
-        gr.Markdown("# 🌟 AIPromptLab — Multi Output JSON Prompt Generator")
+        gr.Markdown("# 🌟 AIPromptLab — Multi Output JSON & PDF E-Book Generator")
 
         with gr.Tabs():
 
             # ============================
             #  TAB 1: E-BOOK
             # ============================
-            with gr.Tab("E-Book Prompt"):
+            with gr.Tab("E-Book Prompt & PDF"):
 
                 ebook_goal = gr.Textbox(label="Goal / Ide E-book")
                 ebook_genre = gr.Dropdown(
@@ -249,10 +334,19 @@ with gr.Blocks() as demo:
                     value="Medium (30–50 pages)"
                 )
 
+                ebook_cover_brief = gr.Textbox(
+                    label="Cover Illustration Brief (diambil dari konsep image design)",
+                    value="Cute pastel Islamic illustration of children reading together under warm light, soft colors, friendly style."
+                )
+
                 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_output = gr.Textbox(label="E-book JSON Output", lines=18, elem_id="ebook-json-output")
                 ebook_copy_btn = gr.Button("Copy E-book JSON")
 
+                gr.Markdown("### 📄 Generate PDF E-book (A4)")
+                ebook_pdf_btn = gr.Button("Generate PDF E-book")
+                ebook_pdf_file = gr.File(label="Download E-book PDF")
+
             # ============================
             #  TAB 2: IMAGE DESIGN
             # ============================
@@ -305,7 +399,7 @@ with gr.Blocks() as demo:
                 )
 
                 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_output = gr.Textbox(label="Image Design JSON Output", lines=18, elem_id="image-json-output")
                 img_copy_btn = gr.Button("Copy Image JSON")
 
             # ============================
@@ -359,7 +453,7 @@ with gr.Blocks() as demo:
                 )
 
                 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_output = gr.Textbox(label="Video AI JSON Output", lines=18, elem_id="video-json-output")
                 vid_copy_btn = gr.Button("Copy Video JSON")
 
     # LOGIN LOGIC
@@ -380,7 +474,7 @@ with gr.Blocks() as demo:
         [login_msg, login_ui, main_ui, login_status, login_user, login_tier]
     )
 
-    # E-BOOK GENERATE
+    # E-BOOK GENERATE JSON
     def handle_ebook_generate(login_status, login_user, login_tier,
                               goal, genre, tone, style, platform, length):
         if not login_status:
@@ -394,6 +488,27 @@ with gr.Blocks() as demo:
         ebook_output
     )
 
+    # E-BOOK GENERATE PDF
+    def handle_ebook_pdf(login_status, login_user,
+                         goal, genre, tone, style, length, cover_brief):
+        if not login_status:
+            return None
+        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")
+        length_ = normalize(length, "Medium (30–50 pages)")
+        cover_ = normalize(cover_brief, "Pastel Islamic illustration of children reading together.")
+        pdf_path = build_ebook_pdf(login_user, goal_, genre_, tone_, style_, length_, cover_)
+        return pdf_path
+
+    ebook_pdf_btn.click(
+        handle_ebook_pdf,
+        [login_status, login_user,
+         ebook_goal, ebook_genre, ebook_tone, ebook_style, ebook_length, ebook_cover_brief],
+        ebook_pdf_file
+    )
+
     # IMAGE GENERATE
     def handle_image_generate(login_status, login_user, login_tier,
                               goal, design_style, color_palette, platform, aspect_ratio):