Update app.py

app.py

@@ -1,4 +1,3 @@
-# app.py
 import gradio as gr
 import json
 from datetime import datetime
@@ -14,10 +13,10 @@ import replicate
 REPLICATE_API_TOKEN = "r8_OHfjZvEbVoU9fuioi4qfQ50OHdpAmqV4OAa5b"
 os.environ["REPLICATE_API_TOKEN"] = REPLICATE_API_TOKEN
 
-MAX_CHAPTERS = 10  # batas maksimal bab
+MAX_CHAPTERS = 10
 
 # ============================
-#  USER DATABASE (SIMPLE)
+#  USER DB
 # ============================
 USER_DB = {
     "akhmad": {"password": "12345", "tier": "super"},
@@ -33,6 +32,10 @@ def normalize(v, default=""):
         return default
     return str(v).strip()
 
+def safe_text(text):
+    # Hanya izinkan ASCII 32–126 (FPDF tidak support Unicode)
+    return ''.join(c if 32 <= ord(c) <= 126 else ' ' for c in text)
+
 def now_iso():
     return datetime.utcnow().isoformat() + "Z"
 
@@ -40,36 +43,33 @@ def ensure_outputs_dir():
     os.makedirs("outputs", exist_ok=True)
 
 # ============================
-#  REPLICATE IMAGE HELPERS (MODEL GRATIS + FALLBACK)
+#  REPLICATE (FREE MODELS)
 # ============================
-def generate_image_flux(prompt: str) -> str:
-    """Flux Schnell FREE (cover)"""
+def generate_image_flux(prompt):
     try:
-        output = replicate.run(
+        out = replicate.run(
             "black-forest-labs/flux-schnell-free",
             input={"prompt": prompt}
         )
-        if isinstance(output, list) and len(output) > 0:
-            return output[0]
-    except Exception:
+        if isinstance(out, list) and out:
+            return out[0]
+    except:
         return ""
     return ""
 
-def generate_image_sdxl(prompt: str) -> str:
-    """SDXL Lite FREE (chapter illustrations)"""
+def generate_image_sdxl(prompt):
     try:
-        output = replicate.run(
+        out = replicate.run(
             "stability-ai/sdxl-lite",
             input={"prompt": prompt}
         )
-        if isinstance(output, list) and len(output) > 0:
-            return output[0]
-    except Exception:
+        if isinstance(out, list) and out:
+            return out[0]
+    except:
         return ""
     return ""
 
-def download_image(url: str, filename: str) -> str:
-    """Download image safely"""
+def download_image(url, filename):
     if not url:
         return ""
     ensure_outputs_dir()
@@ -80,50 +80,11 @@ def download_image(url: str, filename: str) -> str:
         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, chapters):
-
-    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")
-    platform = normalize(platform, "KDP")
-    length = normalize(length, "Medium (30–50 pages)")
-
-    try:
-        chapters = int(chapters)
     except:
-        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",
-        "generated_at": now_iso(),
-        "user": username,
-        "tier": tier,
-        "platform": platform,
-        "genre": genre,
-        "tone": tone,
-        "writing_style": style,
-        "length": length,
-        "goal": goal,
-        "chapters": chapter_titles,
-    }
-
-    return json.dumps(data, ensure_ascii=False, indent=2)
+        return ""
 
 # ============================
-#  PDF E-BOOK GENERATOR (A4) + IMAGES
+#  PDF BUILDER
 # ============================
 class EbookPDF(FPDF):
     def header(self):
@@ -143,32 +104,40 @@ def build_ebook_pdf_with_images(username, goal, genre, tone, style, length,
 
     ensure_outputs_dir()
     pdf = EbookPDF(format="A4")
-    pdf.set_auto_page_break(auto=True, margin=15)
 
-    # ===== COVER IMAGE VIA FLUX (FREE) =====
+    # COVER IMAGE
     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")
 
-    # ===== COVER PAGE =====
     pdf.add_page()
 
+    # MATIKAN AUTO PAGE BREAK SAAT COVER
+    pdf.set_auto_page_break(False)
+
     if cover_path:
         pdf.image(cover_path, x=0, y=0, w=210)
-        pdf.set_y(210)  # aman
+        pdf.set_y(210)
     else:
         pdf.set_fill_color(230, 240, 255)
         pdf.rect(0, 0, 210, 297, "F")
         pdf.set_y(50)
 
+    # AKTIFKAN KEMBALI AUTO PAGE BREAK
+    pdf.set_auto_page_break(True, margin=15)
+
+    # PAKSA Y AMAN
+    pdf.set_y(max(pdf.get_y(), 50))
+
+    # COVER TEXT
     pdf.set_font("Helvetica", "B", 20)
-    pdf.multi_cell(0, 10, goal, 0, "L")
+    pdf.multi_cell(0, 10, safe_text(goal), 0, "L")
 
     pdf.set_font("Helvetica", "", 12)
-    pdf.multi_cell(0, 7, f"Genre: {genre}", 0, "L")
-    pdf.multi_cell(0, 7, f"Tone: {tone} | Style: {style}", 0, "L")
+    pdf.multi_cell(0, 7, safe_text(f"Genre: {genre}"), 0, "L")
+    pdf.multi_cell(0, 7, safe_text(f"Tone: {tone} | Style: {style}"), 0, "L")
 
-    # ===== TABLE OF CONTENTS =====
+    # TABLE OF CONTENTS
     pdf.add_page()
     pdf.set_font("Helvetica", "B", 18)
     pdf.cell(0, 10, "Daftar Isi", 0, 1)
@@ -179,28 +148,27 @@ def build_ebook_pdf_with_images(username, goal, genre, tone, style, length,
     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)
+        pdf.cell(0, 8, safe_text(f"{i}. {ch}"), 0, 1)
 
-    # ===== BODY TEMPLATE =====
+    # 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, dan mengajak pembaca untuk merenungkan makna di balik setiap peristiwa."
+        "mengandung nilai-nilai kebaikan."
     )
 
-    # ===== CHAPTERS =====
+    # CHAPTERS
     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.cell(0, 10, safe_text(ch), 0, 1)
         pdf.ln(4)
 
-        # ===== ILLUSTRATION =====
+        # 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, kid-friendly, Islamic."
+            f"Scene: {ch}. Style: {style}, tone {tone}, pastel colors."
         )
-
         illus_url = generate_image_sdxl(illus_prompt)
         illus_path = download_image(illus_url, f"chapter_{idx}_{uuid.uuid4().hex}.png")
 
@@ -211,7 +179,7 @@ def build_ebook_pdf_with_images(username, goal, genre, tone, style, length,
 
         pdf.set_font("Helvetica", "", 12)
         for _ in range(5):
-            pdf.multi_cell(0, 7, body_paragraph, 0, "L")
+            pdf.multi_cell(0, 7, safe_text(body_paragraph), 0, "L")
             pdf.ln(2)
 
     filename = os.path.join("outputs", f"ebook_{uuid.uuid4().hex}.pdf")
@@ -219,7 +187,46 @@ def build_ebook_pdf_with_images(username, goal, genre, tone, style, length,
     return filename
 
 # ============================
-#  LOGIN HANDLER
+#  E-BOOK JSON GENERATOR
+# ============================
+def generate_ebook_json(username, tier,
+                        goal, genre, tone, style,
+                        platform, length, chapters):
+
+    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")
+    platform = normalize(platform, "KDP")
+    length = normalize(length, "Medium (30–50 pages)")
+
+    try:
+        chapters = int(chapters)
+    except:
+        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",
+        "generated_at": now_iso(),
+        "user": username,
+        "tier": tier,
+        "platform": platform,
+        "genre": genre,
+        "tone": tone,
+        "writing_style": style,
+        "length": length,
+        "goal": goal,
+        "chapters": chapter_titles,
+    }
+
+    return json.dumps(data, ensure_ascii=False, indent=2)
+
+# ============================
+#  LOGIN
 # ============================
 def login(username, password):
     if username in USER_DB and USER_DB[username]["password"] == password:
@@ -335,7 +342,9 @@ with gr.Blocks() as demo:
                 ebook_pdf_btn = gr.Button("Generate PDF E-book")
                 ebook_pdf_file = gr.File(label="Download E-book PDF")
 
-    # LOGIN LOGIC
+    # ============================
+    #  LOGIN HANDLER
+    # ============================
     def handle_login(username, password):
         ok, tier, msg = login(username, password)
         return (
@@ -353,42 +362,64 @@ with gr.Blocks() as demo:
         [login_msg, login_ui, main_ui, login_status, login_user, login_tier]
     )
 
-    # E-BOOK GENERATE JSON
+    # ============================
+    #  E-BOOK GENERATE JSON
+    # ============================
     def handle_ebook_generate(login_status, login_user, login_tier,
                               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, chapters)
+        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_chapters],
+        [
+            login_status, login_user, login_tier,
+            ebook_goal, ebook_genre, ebook_tone,
+            ebook_style, ebook_platform, ebook_length,
+            ebook_chapters
+        ],
         ebook_output
     )
 
-    # E-BOOK GENERATE PDF
+    # ============================
+    #  E-BOOK GENERATE PDF
+    # ============================
     def handle_ebook_pdf(login_status, login_user,
                          goal, genre, tone, style, length, cover_brief, chapters):
         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_with_images(login_user, goal_, genre_, tone_, style_, length_, cover_, chapters)
+
+        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_chapters],
+        [
+            login_status, login_user,
+            ebook_goal, ebook_genre, ebook_tone,
+            ebook_style, ebook_length, ebook_cover_brief,
+            ebook_chapters
+        ],
         ebook_pdf_file
     )
 
-# LAUNCH
+# ============================
+#  LAUNCH APP
+# ============================
 demo.launch(
     server_name="0.0.0.0",
     server_port=7860,