Update app.py

app.py

@@ -1,250 +1,63 @@
-import gradio as gr
-import requests
-import base64
-import uuid
-import os
-import json
-import time
-from groq import Groq
-from fpdf import FPDF, XPos, YPos
-
 # ============================
-# CONFIG (AMAN)
+# FIREWORKS AI MODULE (HYBRID MODE)
 # ============================
-HF_API_KEY = os.getenv("HF_API_KEY")
-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)
+import base64
+import requests
 
-# ============================
-# MODEL ENDPOINTS (FREE)
-# ============================
-MODEL_ENDPOINTS = {
-    "SD Turbo": "stabilityai/sd-turbo",
-    "Playground v2.5": "playgroundai/playground-v2.5-1024px-aesthetic",
-    "SDXL": "stabilityai/stable-diffusion-xl-base-1.0",
-    "PixArt-XL": "PixArt-alpha/PixArt-XL-2-1024-MS"
-}
+FIREWORKS_API_KEY = os.getenv("FIREWORKS_API_KEY")
 
-# ============================
-# ENGINE GAMBAR v3.2 (ULTRA STABLE)
-# ============================
-def hf_generate_image_ultra(prompt, log_callback):
-
-    fallback_rounds = [
-        ["SD Turbo", "Playground v2.5", "SDXL", "PixArt-XL"],
-        ["Playground v2.5", "SDXL", "PixArt-XL", "SD Turbo"],
-        ["SDXL", "PixArt-XL", "Playground v2.5", "SD Turbo"]
-    ]
-
-    delays = [30, 60, 120, 180, 300, 600]
-
-    attempt = 1
-    total_attempts = 12
-
-    for round_idx, models in enumerate(fallback_rounds, start=1):
-        for model_name in models:
-
-            log_callback(f"[Try {attempt}/{total_attempts}] Model: {model_name}")
-
-            model = MODEL_ENDPOINTS[model_name]
-            url = f"https://api-inference.huggingface.co/models/{model}"
-
-            payload = {
-                "inputs": prompt,
-                "options": {"wait_for_model": True}
-            }
-
-            try:
-                response = requests.post(url, headers=HEADERS, json=payload)
-
-                # Cek JSON error
-                try:
-                    data = response.json()
-                    if "error" in data:
-                        log_callback(f"  → HF Error: {data['error']}")
-                        time.sleep(delays[min(attempt-1, len(delays)-1)])
-                        attempt += 1
-                        continue
-                except:
-                    pass
-
-                # Cek MIME type
-                content_type = response.headers.get("Content-Type", "")
-
-                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)
-
-                    if os.path.getsize(filepath) > 10000:
-                        log_callback("  → SUCCESS (image bytes)")
-                        return filepath
-                    else:
-                        log_callback("  → FAILED (image too small)")
-
-                # Cek base64
-                try:
-                    data = response.json()
-                    if "generated_image" in data:
-                        b64 = data["generated_image"].split(",")[-1]
-                        img_bytes = base64.b64decode(b64)
-
-                        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(img_bytes)
-
-                        if os.path.getsize(filepath) > 10000:
-                            log_callback("  → SUCCESS (base64)")
-                            return filepath
-                        else:
-                            log_callback("  → FAILED (base64 too small)")
-                except:
-                    pass
-
-            except Exception as e:
-                log_callback(f"  → Exception: {str(e)}")
-
-            # Delay adaptif
-            delay = delays[min(attempt-1, len(delays)-1)]
-            log_callback(f"  → Waiting {delay} seconds before retry...")
-            time.sleep(delay)
-
-            attempt += 1
-
-    log_callback("❌ Gagal menghasilkan gambar setelah 12 percobaan.")
-    return None
+def fireworks_generate(prompt, model_name, log_callback):
+    """
+    Generate image using Fireworks AI.
+    Returns filepath or None.
+    """
+    url = "https://api.fireworks.ai/inference/v1/image/generate"
 
-# ============================
-# GROQ TEXT GENERATOR
-# ============================
-def ai(prompt):
-    res = client.chat.completions.create(
-        model="llama-3.3-70b-versatile",
-        messages=[{"role": "user", "content": prompt}],
-        max_tokens=800
-    )
-    return res.choices[0].message.content
+    payload = {
+        "model": model_name,
+        "prompt": prompt,
+        "size": "1024x1024",
+        "num_images": 1
+    }
 
-# ============================
-# COMIC PANEL GENERATOR
-# ============================
-def generate_panels(story, style, chapters):
-    panels_per_chapter = 8
-    all_chapters = []
-
-    for c in range(1, chapters+1):
-        prompt = f"""
-        Buatkan {panels_per_chapter} panel komik.
-        Cerita: {story}
-        Style visual: {style}
-        Format:
-        Panel 1: ...
-        Panel 2: ...
-        ...
-        Panel {panels_per_chapter}: ...
-        """
-        raw = ai(prompt)
-        lines = [l for l in raw.split("\n") if l.strip().startswith("Panel")]
-        all_chapters.append(lines[:panels_per_chapter])
-
-    return all_chapters
+    headers = {
+        "Authorization": f"Bearer {FIREWORKS_API_KEY}",
+        "Content-Type": "application/json"
+    }
 
-# ============================
-# VISUAL PROMPT BUILDER
-# ============================
-def build_visual_prompt(panel_text, style):
-    return f"""
-    {style}, soft lighting, clean shapes, child-friendly,
-    cinematic depth, high detail.
-    Scene: {panel_text}
-    """
+    try:
+        response = requests.post(url, json=payload, headers=headers, timeout=40)
+        data = response.json()
 
-# ============================
-# PDF BUILDER
-# ============================
-class ComicPDF(FPDF):
-    pass
+        if "images" not in data:
+            log_callback(f"  → Fireworks Error: {data}")
+            return None
 
-def build_pdf(story, style, chapters, log_callback):
-    pdf = ComicPDF()
-    pdf.set_auto_page_break(True, margin=15)
+        b64 = data["images"][0]["image_base64"]
+        img_bytes = base64.b64decode(b64)
 
-    panels = generate_panels(story, style, chapters)
+        filename = f"fw_{uuid.uuid4().hex}.png"
+        os.makedirs("outputs", exist_ok=True)
+        filepath = os.path.join("outputs", filename)
 
-    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}", new_x=XPos.LMARGIN, new_y=YPos.NEXT)
+        with open(filepath, "wb") as f:
+            f.write(img_bytes)
 
-        for p_idx, panel in enumerate(chapter, start=1):
-            pdf.set_font("Helvetica", "B", 12)
-            pdf.cell(0, 8, f"Panel {p_idx}", new_x=XPos.LMARGIN, new_y=YPos.NEXT)
+        if os.path.getsize(filepath) < 10000:
+            log_callback("  → Fireworks FAILED (image too small)")
+            return None
 
-            visual_prompt = build_visual_prompt(panel, style)
+        log_callback("  → Fireworks SUCCESS")
+        return filepath
 
-            log_callback(f"\n=== GENERATING PANEL {p_idx} ===")
-            img_path = hf_generate_image_ultra(visual_prompt, log_callback)
+    except Exception as e:
+        log_callback(f"  → Fireworks Exception: {str(e)}")
+        return None
 
-            if img_path:
-                pdf.image(img_path, x=20, w=170)
 
-            pdf.set_font("Helvetica", "", 11)
-            pdf.multi_cell(0, 6, panel)
-            pdf.ln(4)
+def fireworks_schnell(prompt, log_callback):
+    return fireworks_generate(prompt, "black-forest-labs/FLUX.1-schnell", log_callback)
 
-    filename = f"comic_{uuid.uuid4().hex}.pdf"
-    pdf.output(filename)
-    return filename
 
-# ============================
-# GRADIO UI
-# ============================
-def run_comic(story, style, chapters):
-    logs = []
-
-    def log_callback(msg):
-        logs.append(msg)
-
-    pdf_file = build_pdf(story, style, chapters, log_callback)
-    return pdf_file, "\n".join(logs)
-
-
-with gr.Blocks() as app:
-
-    gr.Markdown("# 🌟 AIPromptLab 3.2 — Ultra Stable Free Mode (HuggingFace Engine)")
-
-    with gr.Tab("Comic Generator"):
-        story = gr.Textbox(label="Ide Cerita Komik")
-        style = gr.Dropdown(
-            label="Style Visual",
-            choices=[
-                "Pastel 3D Isometric",
-                "Cute 3D Cartoon",
-                "Realistic Cinematic",
-                "Lowpoly Diorama",
-                "Claymation 3D",
-                "Toy Photography",
-                "Anime 3D Soft Light"
-            ],
-            value="Pastel 3D Isometric"
-        )
-        chapters = gr.Slider(1, 10, value=1, step=1, label="Jumlah Bab")
-        btn = gr.Button("Generate Comic PDF")
-        out_pdf = gr.File()
-        out_log = gr.Textbox(label="Log Proses", lines=30)
-
-        btn.click(run_comic, [story, style, chapters], [out_pdf, out_log])
-
-app.launch(ssr_mode=False)
+def fireworks_dev(prompt, log_callback):
+    return fireworks_generate(prompt, "black-forest-labs/FLUX.1-dev", log_callback)