Update app.py

app.py

@@ -43,10 +43,7 @@ def ensure_outputs_dir():
 #  REPLICATE IMAGE HELPERS (MODEL GRATIS + FALLBACK)
 # ============================
 def generate_image_flux(prompt: str) -> str:
-    """
-    Generate image URL using Flux Schnell FREE on Replicate (for cover).
-    Jika gagal (kredit / model / jaringan), return "".
-    """
+    """Flux Schnell FREE (cover)"""
     try:
         output = replicate.run(
             "black-forest-labs/flux-schnell-free",
@@ -59,10 +56,7 @@ def generate_image_flux(prompt: str) -> str:
     return ""
 
 def generate_image_sdxl(prompt: str) -> str:
-    """
-    Generate image URL using SDXL LITE (gratis) on Replicate (for chapter illustrations).
-    Jika gagal, return "".
-    """
+    """SDXL Lite FREE (chapter illustrations)"""
     try:
         output = replicate.run(
             "stability-ai/sdxl-lite",
@@ -75,10 +69,7 @@ def generate_image_sdxl(prompt: str) -> str:
     return ""
 
 def download_image(url: str, filename: str) -> str:
-    """
-    Download image from URL to local file.
-    Jika gagal, return "".
-    """
+    """Download image safely"""
     if not url:
         return ""
     ensure_outputs_dir()
@@ -105,10 +96,12 @@ 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:
+    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)]
@@ -125,107 +118,6 @@ def generate_ebook_json(username, tier,
         "length": length,
         "goal": goal,
         "chapters": chapter_titles,
-        "prompts": {
-            "title_prompt": (
-                f"Generate 10 title ideas for an {genre} ebook about '{goal}'. "
-                f"Tone: {tone}. Style: {style}. Platform: {platform}. "
-                "Titles should be short, catchy, and suitable for children."
-            ),
-            "outline_prompt": (
-                f"Create a detailed chapter outline for an {genre} ebook about '{goal}'. "
-                f"Tone: {tone}. Style: {style}. Length: {length}. "
-                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}'. "
-                f"Tone: {tone}. Style: {style}. Length: {length}. "
-                "Use simple language suitable for children and include dialogues."
-            )
-        }
-    }
-
-    return json.dumps(data, ensure_ascii=False, indent=2)
-
-# ============================
-#  IMAGE DESIGN JSON GENERATOR
-# ============================
-def generate_image_json(username, tier,
-                        goal, design_style, color_palette,
-                        platform, aspect_ratio):
-
-    goal = normalize(goal, "Islamic children's story cover about kindness.")
-    design_style = normalize(design_style, "Pastel Islamic")
-    color_palette = normalize(color_palette, "Pastel")
-    platform = normalize(platform, "Midjourney")
-    aspect_ratio = normalize(aspect_ratio, "1:1")
-
-    prompt = (
-        f"{goal}, {design_style} style, {color_palette} colors, "
-        f"high quality, aspect ratio {aspect_ratio}, soft lighting, kid-friendly, Islamic theme."
-    )
-
-    negative_prompt = (
-        "no text, no watermark, no logo, no distortion, no blur, no low quality, no violence."
-    )
-
-    data = {
-        "type": "image_design",
-        "generated_at": now_iso(),
-        "user": username,
-        "tier": tier,
-        "platform": platform,
-        "goal": goal,
-        "style": design_style,
-        "color_palette": color_palette,
-        "aspect_ratio": aspect_ratio,
-        "prompt": prompt,
-        "negative_prompt": negative_prompt
-    }
-
-    return json.dumps(data, ensure_ascii=False, indent=2)
-
-# ============================
-#  VIDEO AI JSON GENERATOR
-# ============================
-def generate_video_json(username, tier,
-                        goal, video_style, camera_movement,
-                        platform, duration):
-
-    goal = normalize(goal, "Islamic children's story animation about honesty.")
-    video_style = normalize(video_style, "Pastel Kids Animation")
-    camera_movement = normalize(camera_movement, "Slow Zoom")
-    platform = normalize(platform, "Pika Labs")
-    duration = normalize(duration, "10 seconds")
-
-    scenes = [
-        {
-            "scene": 1,
-            "description": (
-                f"A child reading an Islamic storybook about '{goal}' in a cozy room, "
-                f"{video_style.lower()}, soft pastel colors, warm lighting."
-            ),
-            "camera": camera_movement
-        },
-        {
-            "scene": 2,
-            "description": (
-                "Close-up of the book pages glowing softly, Arabic patterns in the background, "
-                "gentle motion, peaceful atmosphere."
-            ),
-            "camera": "Static Shot"
-        }
-    ]
-
-    data = {
-        "type": "video_ai",
-        "generated_at": now_iso(),
-        "user": username,
-        "tier": tier,
-        "platform": platform,
-        "style": video_style,
-        "duration": duration,
-        "goal": goal,
-        "scenes": scenes
     }
 
     return json.dumps(data, ensure_ascii=False, indent=2)
@@ -248,6 +140,7 @@ class EbookPDF(FPDF):
 
 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)
@@ -257,53 +150,57 @@ def build_ebook_pdf_with_images(username, goal, genre, tone, style, length,
     cover_url = generate_image_flux(cover_prompt)
     cover_path = download_image(cover_url, f"cover_{uuid.uuid4().hex}.png")
 
-    # COVER PAGE
+    # ===== COVER PAGE =====
     pdf.add_page()
+
     if cover_path:
         pdf.image(cover_path, x=0, y=0, w=210)
-        pdf.ln(200)
+        pdf.set_y(210)  # aman
     else:
         pdf.set_fill_color(230, 240, 255)
         pdf.rect(0, 0, 210, 297, "F")
+        pdf.set_y(50)
 
-    pdf.set_y(220)
     pdf.set_font("Helvetica", "B", 20)
-    pdf.set_text_color(20, 20, 40)
     pdf.multi_cell(0, 10, 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")
 
-    # TABLE OF CONTENTS
+    # ===== TABLE OF CONTENTS =====
     pdf.add_page()
     pdf.set_font("Helvetica", "B", 18)
     pdf.cell(0, 10, "Daftar Isi", 0, 1)
     pdf.ln(5)
-    pdf.set_font("Helvetica", "", 12)
 
+    pdf.set_font("Helvetica", "", 12)
     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)
 
-    # BODY TEXT TEMPLATE
+    # ===== BODY 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, "
+        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."
     )
 
-    # CHAPTERS WITH SDXL LITE IMAGES
+    # ===== 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.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, kid-friendly, Islamic."
         )
+
         illus_url = generate_image_sdxl(illus_prompt)
         illus_path = download_image(illus_url, f"chapter_{idx}_{uuid.uuid4().hex}.png")
 
@@ -347,7 +244,7 @@ with gr.Blocks() as demo:
         login_btn = gr.Button("Login")
         login_msg = gr.Markdown("")
 
-    # MAIN UI (hidden first)
+    # MAIN UI
     with gr.Group(visible=False) as main_ui:
 
         gr.Markdown("# 🌟 AIPromptLab — JSON & PDF E-Book Generator (Replicate FREE Models)")
@@ -431,122 +328,13 @@ with gr.Blocks() as demo:
                 )
 
                 ebook_generate_btn = gr.Button("🚀 Generate E-book JSON")
-                ebook_output = gr.Textbox(label="E-book JSON Output", lines=18, elem_id="ebook-json-output")
+                ebook_output = gr.Textbox(label="E-book JSON Output", lines=18)
                 ebook_copy_btn = gr.Button("Copy E-book JSON")
 
                 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")
 
-            # ============================
-            #  TAB 2: IMAGE DESIGN
-            # ============================
-            with gr.Tab("Image Design Prompt"):
-
-                img_goal = gr.Textbox(label="Goal / Ide Desain Gambar")
-                img_style = gr.Dropdown(
-                    label="Design Style",
-                    choices=[
-                        "Pastel Islamic",
-                        "Flat Illustration",
-                        "3D Render",
-                        "Watercolor",
-                        "Minimalist",
-                        "Cartoon Kids",
-                        "Realistic Photography"
-                    ],
-                    value="Pastel Islamic"
-                )
-                img_palette = gr.Dropdown(
-                    label="Color Palette",
-                    choices=[
-                        "Pastel",
-                        "Vibrant",
-                        "Monochrome",
-                        "Earth Tone",
-                        "Neon"
-                    ],
-                    value="Pastel"
-                )
-                img_platform = gr.Dropdown(
-                    label="Target Platform",
-                    choices=[
-                        "Midjourney",
-                        "DALL·E",
-                        "Stable Diffusion",
-                        "Canva Design Brief"
-                    ],
-                    value="Midjourney"
-                )
-                img_ratio = gr.Dropdown(
-                    label="Aspect Ratio",
-                    choices=[
-                        "1:1",
-                        "9:16",
-                        "16:9",
-                        "4:5"
-                    ],
-                    value="1:1"
-                )
-
-                img_generate_btn = gr.Button("🚀 Generate Image Design JSON")
-                img_output = gr.Textbox(label="Image Design JSON Output", lines=18, elem_id="image-json-output")
-                img_copy_btn = gr.Button("Copy Image JSON")
-
-            # ============================
-            #  TAB 3: VIDEO AI
-            # ============================
-            with gr.Tab("Video AI Prompt"):
-
-                vid_goal = gr.Textbox(label="Goal / Ide Video AI")
-                vid_style = gr.Dropdown(
-                    label="Video Style",
-                    choices=[
-                        "Cinematic",
-                        "Pastel Kids Animation",
-                        "Anime",
-                        "Realistic",
-                        "Claymation",
-                        "Watercolor Motion"
-                    ],
-                    value="Pastel Kids Animation"
-                )
-                vid_camera = gr.Dropdown(
-                    label="Camera Movement",
-                    choices=[
-                        "Slow Zoom",
-                        "Pan Left",
-                        "Pan Right",
-                        "Dolly In",
-                        "Static Shot"
-                    ],
-                    value="Slow Zoom"
-                )
-                vid_platform = gr.Dropdown(
-                    label="Target Platform",
-                    choices=[
-                        "Pika Labs",
-                        "Runway Gen-2",
-                        "Kaiber",
-                        "Luma Dream Machine"
-                    ],
-                    value="Pika Labs"
-                )
-                vid_duration = gr.Dropdown(
-                    label="Duration",
-                    choices=[
-                        "5 seconds",
-                        "10 seconds",
-                        "15 seconds",
-                        "30 seconds"
-                    ],
-                    value="10 seconds"
-                )
-
-                vid_generate_btn = gr.Button("🚀 Generate Video AI JSON")
-                vid_output = gr.Textbox(label="Video AI JSON Output", lines=18, elem_id="video-json-output")
-                vid_copy_btn = gr.Button("Copy Video JSON")
-
     # LOGIN LOGIC
     def handle_login(username, password):
         ok, tier, msg = login(username, password)
@@ -600,56 +388,6 @@ with gr.Blocks() as demo:
         ebook_pdf_file
     )
 
-    # IMAGE GENERATE
-    def handle_image_generate(login_status, login_user, login_tier,
-                              goal, design_style, color_palette, platform, aspect_ratio):
-        if not login_status:
-            return "❌ Anda belum login."
-        return generate_image_json(login_user, login_tier, goal, design_style, color_palette, platform, aspect_ratio)
-
-    img_generate_btn.click(
-        handle_image_generate,
-        [login_status, login_user, login_tier,
-         img_goal, img_style, img_palette, img_platform, img_ratio],
-        img_output
-    )
-
-    # VIDEO GENERATE
-    def handle_video_generate(login_status, login_user, login_tier,
-                              goal, video_style, camera_movement, platform, duration):
-        if not login_status:
-            return "❌ Anda belum login."
-        return generate_video_json(login_user, login_tier, goal, video_style, camera_movement, platform, duration)
-
-    vid_generate_btn.click(
-        handle_video_generate,
-        [login_status, login_user, login_tier,
-         vid_goal, vid_style, vid_camera, vid_platform, vid_duration],
-        vid_output
-    )
-
-    # COPY BUTTONS (JS)
-    ebook_copy_btn.click(
-        None,
-        None,
-        None,
-        js="navigator.clipboard.writeText(document.getElementById('ebook-json-output').value)"
-    )
-
-    img_copy_btn.click(
-        None,
-        None,
-        None,
-        js="navigator.clipboard.writeText(document.getElementById('image-json-output').value)"
-    )
-
-    vid_copy_btn.click(
-        None,
-        None,
-        None,
-        js="navigator.clipboard.writeText(document.getElementById('video-json-output').value)"
-    )
-
 # LAUNCH
 demo.launch(
     server_name="0.0.0.0",