Update app.py

app.py

@@ -42,17 +42,22 @@ def ensure_outputs_dir():
     os.makedirs("outputs", exist_ok=True)
 
 # ============================
-#  AI TEXT GENERATOR (LLaMA‑3)
+#  AI TEXT GENERATOR (LLaMA‑3 FIXED)
 # ============================
 def ai_generate(prompt):
     try:
         output = replicate.run(
             "meta/meta-llama-3-8b-instruct",
-            input={"prompt": prompt, "max_tokens": 500}
+            input={
+                "messages": [
+                    {"role": "user", "content": prompt}
+                ],
+                "max_tokens": 500
+            }
         )
         return "".join(output)
-    except:
-        return "AI gagal menghasilkan teks."
+    except Exception as e:
+        return f"AI gagal menghasilkan teks. Error: {str(e)}"
 
 def generate_chapter_text(goal, style, tone, chapter_title):
     prompt = f"""
@@ -67,7 +72,7 @@ def generate_chapter_text(goal, style, tone, chapter_title):
     return ai_generate(prompt)
 
 # ============================
-#  REPLICATE (FREE MODELS)
+#  REPLICATE IMAGE MODELS
 # ============================
 def generate_image_flux(prompt):
     try:
@@ -170,7 +175,7 @@ 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 (LANGSUNG BAB 1–N, TANPA RINGKASAN/OUTLINE)
+    # CHAPTERS (AI TEXT)
     for idx, ch in enumerate(chapters, start=1):
         pdf.add_page()
         pdf.set_font("Helvetica", "B", 16)