Update app.py

app.py

@@ -3,21 +3,24 @@ import requests
 import base64
 import uuid
 import os
+import json
 from groq import Groq
-from huggingface_hub import hf_hub_url, model_info
-from fpdf import FPDF
+from huggingface_hub import model_info
+from fpdf import FPDF, XPos, YPos
 
 # ============================
-# CONFIG
+# CONFIG (AMAN)
 # ============================
 HF_API_KEY = os.getenv("HF_API_KEY")
-client = Groq(api_key="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")
+GROQ_API_KEY = os.getenv("GROQ_API_KEY")
 
 HEADERS = {
     "Authorization": f"Bearer {HF_API_KEY}",
     "Content-Type": "application/json"
 }
 
+client = Groq(api_key=GROQ_API_KEY)
+
 # ============================
 # MODEL ENDPOINTS (FREE)
 # ============================
@@ -29,7 +32,7 @@ MODEL_ENDPOINTS = {
 }
 
 # ============================
-# HF INFERENCE API ENGINE
+# ENGINE GAMBAR v2.0 (ANTI-GAGAL)
 # ============================
 def hf_generate_image(model, prompt):
     url = f"https://api-inference.huggingface.co/models/{model}"
@@ -39,21 +42,54 @@ def hf_generate_image(model, prompt):
         "options": {"wait_for_model": True}
     }
 
-    response = requests.post(url, headers=HEADERS, json=payload)
+    # Retry 3x untuk cold start
+    for attempt in range(3):
+        response = requests.post(url, headers=HEADERS, json=payload)
+
+        # Jika model masih loading
+        try:
+            data = response.json()
+            if "error" in data:
+                if "loading" in data["error"].lower():
+                    continue
+        except:
+            pass
+
+        # Jika response adalah gambar bytes
+        if response.status_code == 200:
+            content_type = response.headers.get("Content-Type", "")
+
+            # Jika raw image
+            if "image" in content_type:
+                filename = f"img_{uuid.uuid4().hex}.png"
+                os.makedirs("outputs", exist_ok=True)
+                filepath = os.path.join("outputs", filename)
+
+                with open(filepath, "wb") as f:
+                    f.write(response.content)
+
+                return filepath
 
-    if response.status_code != 200:
-        return None
+            # Jika base64
+            try:
+                data = response.json()
+                if isinstance(data, dict):
+                    if "generated_image" in data:
+                        b64 = data["generated_image"].split(",")[-1]
+                        img_bytes = base64.b64decode(b64)
 
-    img_bytes = response.content
-    filename = f"img_{uuid.uuid4().hex}.png"
+                        filename = f"img_{uuid.uuid4().hex}.png"
+                        os.makedirs("outputs", exist_ok=True)
+                        filepath = os.path.join("outputs", filename)
 
-    os.makedirs("outputs", exist_ok=True)
-    filepath = os.path.join("outputs", filename)
+                        with open(filepath, "wb") as f:
+                            f.write(img_bytes)
 
-    with open(filepath, "wb") as f:
-        f.write(img_bytes)
+                        return filepath
+            except:
+                pass
 
-    return filepath
+    return None
 
 # ============================
 # GROQ TEXT GENERATOR
@@ -101,7 +137,7 @@ def build_visual_prompt(panel_text, style):
     """
 
 # ============================
-# PDF BUILDER
+# PDF BUILDER (AMAN)
 # ============================
 class ComicPDF(FPDF):
     pass
@@ -115,11 +151,11 @@ def build_pdf(story, style, model, chapters):
     for c_idx, chapter in enumerate(panels, start=1):
         pdf.add_page()
         pdf.set_font("Helvetica", "B", 16)
-        pdf.cell(0, 10, f"Bab {c_idx}", 0, 1)
+        pdf.cell(0, 10, f"Bab {c_idx}", new_x=XPos.LMARGIN, new_y=YPos.NEXT)
 
         for p_idx, panel in enumerate(chapter, start=1):
             pdf.set_font("Helvetica", "B", 12)
-            pdf.cell(0, 8, f"Panel {p_idx}", 0, 1)
+            pdf.cell(0, 8, f"Panel {p_idx}", new_x=XPos.LMARGIN, new_y=YPos.NEXT)
 
             visual_prompt = build_visual_prompt(panel, style)
             img_path = hf_generate_image(model, visual_prompt)
@@ -249,4 +285,4 @@ with gr.Blocks() as app:
 
             btn3.click(generate_storyboard, [vid_desc, vid_style, vid_model], frame_gallery)
 
-app.launch()
+app.launch(ssr_mode=False)