Update app.py

app.py

@@ -5,6 +5,16 @@ from datetime import datetime
 from fpdf import FPDF
 import os
 import uuid
+import requests
+import replicate
+
+# ============================
+#  CONFIG
+# ============================
+REPLICATE_API_TOKEN = "ISI_API_KEY_REPLICATE_KAMU_DI_SINI"
+os.environ["REPLICATE_API_TOKEN"] = REPLICATE_API_TOKEN
+
+MAX_CHAPTERS = 10  # batas maksimal bab
 
 # ============================
 #  USER DATABASE (SIMPLE)
@@ -26,12 +36,60 @@ def normalize(v, default=""):
 def now_iso():
     return datetime.utcnow().isoformat() + "Z"
 
+def ensure_outputs_dir():
+    os.makedirs("outputs", exist_ok=True)
+
+# ============================
+#  REPLICATE IMAGE HELPERS
+# ============================
+def generate_image_flux(prompt: str) -> str:
+    """
+    Generate image URL using Flux Schnell on Replicate (for cover).
+    """
+    output = replicate.run(
+        "black-forest-labs/flux-schnell",
+        input={"prompt": prompt}
+    )
+    # output biasanya list URL
+    if isinstance(output, list) and len(output) > 0:
+        return output[0]
+    return ""
+
+def generate_image_sdxl(prompt: str) -> str:
+    """
+    Generate image URL using SDXL on Replicate (for chapter illustrations).
+    """
+    output = replicate.run(
+        "stability-ai/sdxl",
+        input={"prompt": prompt}
+    )
+    if isinstance(output, list) and len(output) > 0:
+        return output[0]
+    return ""
+
+def download_image(url: str, filename: str) -> str:
+    """
+    Download image from URL to local file.
+    """
+    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 ""
+
 # ============================
 #  E-BOOK JSON GENERATOR
 # ============================
 def generate_ebook_json(username, tier,
                         goal, genre, tone, style,
-                        platform, length):
+                        platform, length, chapters):
 
     goal = normalize(goal, "Islamic children's story about good manners.")
     genre = normalize(genre, "Islamic Children Story")
@@ -39,6 +97,13 @@ def generate_ebook_json(username, tier,
     style = normalize(style, "Storybook style")
     platform = normalize(platform, "KDP")
     length = normalize(length, "Medium (30–50 pages)")
+    try:
+        chapters = int(chapters)
+    except Exception:
+        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": "ebook",
@@ -51,6 +116,7 @@ def generate_ebook_json(username, tier,
         "writing_style": style,
         "length": length,
         "goal": goal,
+        "chapters": chapter_titles,
         "prompts": {
             "title_prompt": (
                 f"Generate 10 title ideas for an {genre} ebook about '{goal}'. "
@@ -60,7 +126,7 @@ def generate_ebook_json(username, tier,
             "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."
+                f"Use {chapters} chapters. Include chapter titles and 2–3 bullet points per chapter."
             ),
             "chapter_prompt": (
                 f"Write Chapter 1 of an {genre} ebook about '{goal}'. "
@@ -157,7 +223,7 @@ def generate_video_json(username, tier,
     return json.dumps(data, ensure_ascii=False, indent=2)
 
 # ============================
-#  PDF E-BOOK GENERATOR (A4)
+#  PDF E-BOOK GENERATOR (A4) + IMAGES
 # ============================
 class EbookPDF(FPDF):
     def header(self):
@@ -172,32 +238,34 @@ 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(username, goal, genre, tone, style, length, cover_brief):
+def build_ebook_pdf_with_images(username, goal, genre, tone, style, length,
+                                cover_brief, chapters_count):
+    ensure_outputs_dir()
     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")
+    # ===== COVER IMAGE VIA FLUX =====
+    cover_prompt = f"{cover_brief} | {goal} | {genre} | {style} | pastel, kid-friendly, Islamic"
+    cover_url = generate_image_flux(cover_prompt)
+    cover_path = download_image(cover_url, f"cover_{uuid.uuid4().hex}.png")
 
-    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)
+    # COVER PAGE
+    pdf.add_page()
+    if cover_path:
+        # full width cover
+        pdf.image(cover_path, x=0, y=0, w=210)
+        pdf.ln(200)
+    else:
+        pdf.set_fill_color(230, 240, 255)
+        pdf.rect(0, 0, 210, 297, "F")
+
+    pdf.set_y(220)
     pdf.set_font("Helvetica", "B", 20)
-    pdf.multi_cell(0, 12, goal, 0, "C")
-    pdf.ln(10)
+    pdf.set_text_color(20, 20, 40)
+    pdf.multi_cell(0, 10, goal, 0, "L")
     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")
+    pdf.multi_cell(0, 7, f"Genre: {genre}", 0, "L")
+    pdf.multi_cell(0, 7, f"Tone: {tone} | Style: {style}", 0, "L")
 
     # TABLE OF CONTENTS
     pdf.add_page()
@@ -205,36 +273,44 @@ def build_ebook_pdf(username, goal, genre, tone, style, length, cover_brief):
     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"
-    ]
+
+    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)]
     for i, ch in enumerate(chapters, start=1):
         pdf.cell(0, 8, f"{i}. {ch}", 0, 1)
 
-    # CHAPTERS (SIMPLE GENERATED TEXT)
+    # BODY TEXT TEMPLATE
     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:
+    # CHAPTERS WITH SDXL IMAGES
+    for idx, ch in enumerate(chapters, start=1):
         pdf.add_page()
         pdf.set_font("Helvetica", "B", 16)
         pdf.cell(0, 10, ch, 0, 1)
         pdf.ln(4)
+
+        # generate illustration for this chapter
+        illus_prompt = (
+            f"Illustration for chapter {idx} of an Islamic children's story about {goal}. "
+            f"Scene: {ch}. Style: {style}, tone {tone}, pastel colors, kid-friendly, Islamic."
+        )
+        illus_url = generate_image_sdxl(illus_prompt)
+        illus_path = download_image(illus_url, f"chapter_{idx}_{uuid.uuid4().hex}.png")
+
+        if illus_path:
+            pdf.image(illus_path, x=20, y=pdf.get_y(), w=170)
+            pdf.ln(80)
+
         pdf.set_font("Helvetica", "", 12)
-        for _ in range(6):
-            pdf.multi_cell(0, 7, body_paragraph, 0, "J")
+        for _ in range(5):
+            pdf.multi_cell(0, 7, body_paragraph, 0, "L")
             pdf.ln(2)
 
-    # SAVE TEMP FILE
-    os.makedirs("outputs", exist_ok=True)
-    filename = f"outputs/ebook_{uuid.uuid4().hex}.pdf"
+    filename = os.path.join("outputs", f"ebook_{uuid.uuid4().hex}.pdf")
     pdf.output(filename)
     return filename
 
@@ -267,7 +343,7 @@ with gr.Blocks() as demo:
     # MAIN UI (hidden first)
     with gr.Group(visible=False) as main_ui:
 
-        gr.Markdown("# 🌟 AIPromptLab — Multi Output JSON & PDF E-Book Generator")
+        gr.Markdown("# 🌟 AIPromptLab — JSON & PDF E-Book Generator (Replicate + Flux + SDXL)")
 
         with gr.Tabs():
 
@@ -334,8 +410,16 @@ with gr.Blocks() as demo:
                     value="Medium (30–50 pages)"
                 )
 
+                ebook_chapters = gr.Slider(
+                    label="Jumlah Bab (dengan ilustrasi)",
+                    minimum=1,
+                    maximum=MAX_CHAPTERS,
+                    step=1,
+                    value=5
+                )
+
                 ebook_cover_brief = gr.Textbox(
-                    label="Cover Illustration Brief (diambil dari konsep image design)",
+                    label="Cover Illustration Brief",
                     value="Cute pastel Islamic illustration of children reading together under warm light, soft colors, friendly style."
                 )
 
@@ -343,7 +427,7 @@ with gr.Blocks() as demo:
                 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)")
+                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")
 
@@ -476,21 +560,21 @@ with gr.Blocks() as demo:
 
     # E-BOOK GENERATE JSON
     def handle_ebook_generate(login_status, login_user, login_tier,
-                              goal, genre, tone, style, platform, length):
+                              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)
+        return generate_ebook_json(login_user, login_tier, goal, genre, tone, style, platform, length, chapters)
 
     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_goal, ebook_genre, ebook_tone, ebook_style, ebook_platform, ebook_length, ebook_chapters],
         ebook_output
     )
 
     # E-BOOK GENERATE PDF
     def handle_ebook_pdf(login_status, login_user,
-                         goal, genre, tone, style, length, cover_brief):
+                         goal, genre, tone, style, length, cover_brief, chapters):
         if not login_status:
             return None
         goal_ = normalize(goal, "Islamic children's story about good manners.")
@@ -499,13 +583,13 @@ with gr.Blocks() as demo:
         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_)
+        pdf_path = build_ebook_pdf_with_images(login_user, goal_, genre_, tone_, style_, length_, cover_, chapters)
         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_goal, ebook_genre, ebook_tone, ebook_style, ebook_length, ebook_cover_brief, ebook_chapters],
         ebook_pdf_file
     )
 
@@ -564,5 +648,5 @@ demo.launch(
     server_name="0.0.0.0",
     server_port=7860,
     theme=gr.themes.Soft(),
-    share=True
+    ssr_mode=True
 )