Update app.py

app.py

@@ -1,5 +1,4 @@
 import gradio as gr
-import json
 from datetime import datetime
 from fpdf import FPDF
 import os
@@ -32,16 +31,10 @@ USER_DB = {
 # ============================
 #  HELPERS
 # ============================
-def normalize(v, default=""):
-    if not v:
-        return default
-    return str(v).strip()
-
 def safe_text(text):
-    return ''.join(c if 32 <= ord(c) <= 126 else ' ' for c in text)
-
-def now_iso():
-    return datetime.utcnow().isoformat() + "Z"
+    if not text:
+        return ""
+    return ''.join(c if 32 <= ord(c) <= 126 else ' ' for c in str(text))
 
 def ensure_outputs_dir():
     os.makedirs("outputs", exist_ok=True)
@@ -98,86 +91,66 @@ def generate_comic_panels(goal, style, tone, chapter_title, panels=PANELS_PER_CH
 #  VISUAL PROMPT GENERATOR
 # ============================
 def convert_panel_to_visual_prompt(panel_text, character_mode, character_desc):
-    """
-    Mengubah teks panel menjadi prompt visual pastel cartoon.
-    """
     base_style = (
         "Cute pastel cartoon illustration, soft colors, round shapes, kid-friendly, "
         "Islamic-friendly, simple background, environmental setting, high quality."
     )
-
     if character_mode == "Custom" and character_desc.strip():
         return f"{base_style} Featuring: {character_desc}. Scene: {panel_text}."
-
     return f"{base_style} Scene: {panel_text}."
 
 # ============================
-#  IMAGE GENERATORS (3 MODEL)
+#  IMAGE MODEL ROUTER (16 MODEL)
 # ============================
-def generate_image_pastel_mix(prompt):
-    """
-    fofr/pastel-mix — paling cocok untuk komik pastel
-    """
-    try:
-        out = replicate.run(
-            "fofr/pastel-mix",
-            input={"prompt": prompt}
-        )
-        if isinstance(out, list) and out:
-            return out[0]
-    except:
-        return ""
-    return ""
 
-def generate_image_sdxl(prompt):
-    """
-    stability-ai/sdxl — detail tinggi
-    """
-    try:
-        out = replicate.run(
-            "stability-ai/sdxl",
-            input={
-                "prompt": prompt,
-                "cfg_scale": 7,
-                "steps": 30
-            }
-        )
-        if isinstance(out, dict) and "output" in out and out["output"]:
-            return out["output"][0]
-        if isinstance(out, list) and out:
-            return out[0]
-    except:
-        return ""
-    return ""
+MODEL_MAP = {
+    # Nama pendek -> ID model Replicate
+    "Pastel-Mix": "fofr/pastel-mix",
+    "SDXL": "stability-ai/sdxl",
+    "Flux Schnell": "black-forest-labs/flux-schnell",
+    "Flux Schnell Free": "black-forest-labs/flux-schnell-free",
+    "Flux Pro": "black-forest-labs/flux-1.1-pro",
+    "Flux Max": "black-forest-labs/flux-1.1-max",
+    "Flux Dev": "black-forest-labs/flux-1.1-dev",
+    "Flux Flex": "black-forest-labs/flux-1.1-flex",
+    "Seedream 4.5": "bytedance/seedream-4.5",
+    "Seedream 5 Lite": "bytedance/seedream-5-lite",
+    "Ideogram Turbo": "ideogram-ai/ideogram-v3-turbo",
+    "Ideogram Balanced": "ideogram-ai/ideogram-v3-balanced",
+    "Ideogram Quality": "ideogram-ai/ideogram-v3-quality",
+    "Recraft V4": "recraft-ai/recraft-v4",
+    "Recraft V4 SVG": "recraft-ai/recraft-v4-svg",
+    "Imagen Fast": "google/imagen-4-fast",
+}
 
-def generate_image_flux(prompt):
-    """
-    black-forest-labs/flux-schnell-free — cepat, tapi tidak selalu stabil
-    """
+def generate_image_generic(model_id, prompt):
     try:
-        out = replicate.run(
-            "black-forest-labs/flux-schnell-free",
-            input={"prompt": prompt}
-        )
+        out = replicate.run(model_id, input={"prompt": prompt})
+        # Beberapa model mengembalikan list, beberapa dict
         if isinstance(out, list) and out:
             return out[0]
-    except:
+        if isinstance(out, dict):
+            # coba beberapa kemungkinan key umum
+            for key in ["output", "image", "images"]:
+                if key in out and out[key]:
+                    if isinstance(out[key], list):
+                        return out[key][0]
+                    return out[key]
+    except Exception:
         return ""
     return ""
 
 def generate_image_by_model(prompt, model_name):
-    """
-    Pilih model berdasarkan dropdown user.
-    """
-    if model_name == "Pastel-Mix":
-        return generate_image_pastel_mix(prompt)
-    elif model_name == "SDXL":
-        return generate_image_sdxl(prompt)
-    elif model_name == "Flux":
-        return generate_image_flux(prompt)
-    else:
+    model_id = MODEL_MAP.get(model_name)
+    if not model_id:
         # fallback ke Pastel-Mix
-        return generate_image_pastel_mix(prompt)
+        model_id = MODEL_MAP["Pastel-Mix"]
+    url = generate_image_generic(model_id, prompt)
+    if not url:
+        # fallback kedua: Pastel-Mix
+        if model_name != "Pastel-Mix":
+            url = generate_image_generic(MODEL_MAP["Pastel-Mix"], prompt)
+    return url
 
 def download_image(url, filename):
     if not url:
@@ -190,7 +163,7 @@ def download_image(url, filename):
         with open(path, "wb") as f:
             f.write(r.content)
         return path
-    except:
+    except Exception:
         return ""
 
 # ============================
@@ -266,42 +239,34 @@ def build_comic_pdf(username, goal, genre, tone, style, cover_brief,
         pdf.cell(0, 10, safe_text(ch), 0, 1)
         pdf.ln(4)
 
-        # Generate panels
         panels_text = generate_comic_panels(goal, style, tone, ch, PANELS_PER_CHAPTER)
 
         for p_idx, panel in enumerate(panels_text, start=1):
-
-            # Visual prompt
             visual_prompt = convert_panel_to_visual_prompt(
                 panel,
                 character_mode,
                 character_desc
             )
 
-            # Generate image by selected model
             illus_url = generate_image_by_model(visual_prompt, image_model)
             illus_path = download_image(
                 illus_url,
                 f"chapter_{idx}_panel_{p_idx}_{uuid.uuid4().hex}.png"
             )
 
-            # Panel title
             pdf.set_font("Helvetica", "B", 12)
             pdf.cell(0, 8, safe_text(f"Panel {p_idx}"), 0, 1)
             pdf.ln(1)
 
-            # Image
             if illus_path:
                 y_start = pdf.get_y()
                 pdf.image(illus_path, x=20, y=y_start, w=170)
                 pdf.set_y(y_start + 80)
 
-            # Text
             pdf.set_font("Helvetica", "", 11)
             pdf.multi_cell(180, 6, safe_text(panel), 0)
             pdf.ln(3)
 
-    # OUTPUT FILE
     if comic_name.strip():
         filename = os.path.join("outputs", f"{comic_name}.pdf")
     else:
@@ -338,7 +303,7 @@ with gr.Blocks() as demo:
     # MAIN UI
     with gr.Group(visible=False) as main_ui:
 
-        gr.Markdown("# 🌟 AIPromptLab — Comic Generator 2.1 (Cartoon Pastel Edition)")
+        gr.Markdown("# 🌟 AIPromptLab — Comic Generator 2.3 (Multi Model Image)")
 
         with gr.Tabs():
 
@@ -395,7 +360,6 @@ with gr.Blocks() as demo:
                     value="Cute pastel cartoon cover of Islamic children adventure."
                 )
 
-                # CHARACTER MODE
                 character_mode = gr.Dropdown(
                     label="Mode Karakter",
                     choices=["Otomatis", "Custom"],
@@ -416,11 +380,11 @@ with gr.Blocks() as demo:
                     [character_desc]
                 )
 
-                # IMAGE MODEL DROPDOWN
+                # IMAGE MODEL DROPDOWN (16 MODEL, A–Z)
                 image_model = gr.Dropdown(
                     label="Model Gambar",
-                    choices=["Pastel-Mix", "SDXL", "Flux"],
-                    value="Pastel-Mix"  # default sesuai pilihanmu
+                    choices=sorted(list(MODEL_MAP.keys())),
+                    value="Pastel-Mix"
                 )
 
                 generate_pdf_btn = gr.Button("🚀 Generate Comic PDF")