Update app.py

app.py

@@ -41,6 +41,31 @@ def now_iso():
 def ensure_outputs_dir():
     os.makedirs("outputs", exist_ok=True)
 
+# ============================
+#  AI TEXT GENERATOR (LLaMA‑3)
+# ============================
+def ai_generate(prompt):
+    try:
+        output = replicate.run(
+            "meta/meta-llama-3-8b-instruct",
+            input={"prompt": prompt, "max_tokens": 500}
+        )
+        return "".join(output)
+    except:
+        return "AI gagal menghasilkan teks."
+
+def generate_chapter_text(goal, style, tone, chapter_title):
+    prompt = f"""
+    Buatkan 1 bab cerita anak Islami.
+    Judul bab: {chapter_title}
+    Tema utama: {goal}
+    Gaya: {style}
+    Tone: {tone}
+    Panjang: 4 paragraf.
+    Bahasa Indonesia.
+    """
+    return ai_generate(prompt)
+
 # ============================
 #  REPLICATE (FREE MODELS)
 # ============================
@@ -145,20 +170,14 @@ 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)
 
-    # BODY TEMPLATE
-    body_paragraph = (
-        f"Cerita ini mengisahkan tentang {goal.lower()}, disampaikan dengan gaya {style.lower()} "
-        f"dan nuansa {tone.lower()}. Cerita dirancang agar mudah dipahami anak-anak, "
-        "mengandung nilai-nilai kebaikan."
-    )
-
-    # CHAPTERS
+    # CHAPTERS (LANGSUNG BAB 1–N, TANPA RINGKASAN/OUTLINE)
     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."
@@ -171,10 +190,12 @@ def build_ebook_pdf_with_images(username, goal, genre, tone, style, length,
             pdf.image(illus_path, x=20, y=y_start, w=170)
             pdf.set_y(y_start + 80)
 
+        # AI CHAPTER TEXT
+        chapter_text = generate_chapter_text(goal, style, tone, ch)
+
         pdf.set_font("Helvetica", "", 12)
-        for _ in range(5):
-            pdf.multi_cell(180, 7, safe_text(body_paragraph), 0)
-            pdf.ln(2)
+        pdf.multi_cell(180, 7, safe_text(chapter_text), 0)
+        pdf.ln(2)
 
     filename = os.path.join("outputs", f"ebook_{uuid.uuid4().hex}.pdf")
     pdf.output(filename)