Update app.py

app.py

@@ -11,7 +11,7 @@ import replicate
 # ============================
 #  CONFIG
 # ============================
-REPLICATE_API_TOKEN = "r8_OHfjZvEbVoU9fuioi4qfQ50OHdpAmqV4OAa5b"
+REPLICATE_API_TOKEN = "ISI_API_KEY_REPLICATE_KAMU_DI_SINI"
 os.environ["REPLICATE_API_TOKEN"] = REPLICATE_API_TOKEN
 
 MAX_CHAPTERS = 10  # batas maksimal bab
@@ -40,36 +40,44 @@ def ensure_outputs_dir():
     os.makedirs("outputs", exist_ok=True)
 
 # ============================
-#  REPLICATE IMAGE HELPERS
+#  REPLICATE IMAGE HELPERS (MODEL GRATIS + FALLBACK)
 # ============================
 def generate_image_flux(prompt: str) -> str:
     """
-    Generate image URL using Flux Schnell on Replicate (for cover).
+    Generate image URL using Flux Schnell FREE on Replicate (for cover).
+    Jika gagal (kredit / model / jaringan), return "".
     """
-    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]
+    try:
+        output = replicate.run(
+            "black-forest-labs/flux-schnell-free",
+            input={"prompt": prompt}
+        )
+        if isinstance(output, list) and len(output) > 0:
+            return output[0]
+    except Exception:
+        return ""
     return ""
 
 def generate_image_sdxl(prompt: str) -> str:
     """
-    Generate image URL using SDXL on Replicate (for chapter illustrations).
+    Generate image URL using SDXL LITE (gratis) on Replicate (for chapter illustrations).
+    Jika gagal, return "".
     """
-    output = replicate.run(
-        "stability-ai/sdxl",
-        input={"prompt": prompt}
-    )
-    if isinstance(output, list) and len(output) > 0:
-        return output[0]
+    try:
+        output = replicate.run(
+            "stability-ai/sdxl-lite",
+            input={"prompt": prompt}
+        )
+        if isinstance(output, list) and len(output) > 0:
+            return output[0]
+    except Exception:
+        return ""
     return ""
 
 def download_image(url: str, filename: str) -> str:
     """
     Download image from URL to local file.
+    Jika gagal, return "".
     """
     if not url:
         return ""
@@ -244,7 +252,7 @@ def build_ebook_pdf_with_images(username, goal, genre, tone, style, length,
     pdf = EbookPDF(format="A4")
     pdf.set_auto_page_break(auto=True, margin=15)
 
-    # ===== COVER IMAGE VIA FLUX =====
+    # ===== COVER IMAGE VIA FLUX (FREE) =====
     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")
@@ -252,7 +260,6 @@ def build_ebook_pdf_with_images(username, goal, genre, tone, style, length,
     # 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:
@@ -286,14 +293,13 @@ def build_ebook_pdf_with_images(username, goal, genre, tone, style, length,
         "mengandung nilai-nilai kebaikan, dan mengajak pembaca untuk merenungkan makna di balik setiap peristiwa."
     )
 
-    # CHAPTERS WITH SDXL IMAGES
+    # CHAPTERS WITH SDXL LITE 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."
@@ -302,8 +308,9 @@ def build_ebook_pdf_with_images(username, goal, genre, tone, style, length,
         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)
+            y_start = pdf.get_y()
+            pdf.image(illus_path, x=20, y=y_start, w=170)
+            pdf.set_y(y_start + 80)
 
         pdf.set_font("Helvetica", "", 12)
         for _ in range(5):
@@ -343,7 +350,7 @@ with gr.Blocks() as demo:
     # MAIN UI (hidden first)
     with gr.Group(visible=False) as main_ui:
 
-        gr.Markdown("# 🌟 AIPromptLab — JSON & PDF E-Book Generator (Replicate + Flux + SDXL)")
+        gr.Markdown("# 🌟 AIPromptLab — JSON & PDF E-Book Generator (Replicate FREE Models)")
 
         with gr.Tabs():