Update app.py

app.py

@@ -1,46 +1,59 @@
 import gradio as gr
-import replicate
+import requests
+import base64
 import uuid
 import os
 from groq import Groq
+from huggingface_hub import hf_hub_url, model_info
 from fpdf import FPDF
 
 # ============================
 # CONFIG
 # ============================
-os.environ["REPLICATE_API_TOKEN"] = "r8_Ap7YprjUOrZ7chAcXy2m1HAEDpFRbwV1dIM2Y"   # GANTI DENGAN TOKENMU
+HF_API_KEY = os.getenv("HF_API_KEY")
 client = Groq(api_key="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")
 
+HEADERS = {
+    "Authorization": f"Bearer {HF_API_KEY}",
+    "Content-Type": "application/json"
+}
+
 # ============================
-# NANO BANANA ENGINE
-# ============================
-def nano_banana(prompt, aspect_ratio="1:1", resolution="1K"):
-    output = replicate.run(
-        "google/nano-banana-2",
-        input={
-            "prompt": prompt,
-            "resolution": resolution,
-            "image_input": [],
-            "aspect_ratio": aspect_ratio,
-            "image_search": False,
-            "google_search": False,
-            "output_format": "jpg"
-        }
-    )
+# MODEL ENDPOINTS (FREE)
+# ============================
+MODEL_ENDPOINTS = {
+    "SDXL": "stabilityai/stable-diffusion-xl-base-1.0",
+    "Playground v2.5": "playgroundai/playground-v2.5-1024px-aesthetic",
+    "PixArt-XL": "PixArt-alpha/PixArt-XL-2-1024-MS",
+    "SD Turbo": "stabilityai/sd-turbo"
+}
+
+# ============================
+# HF INFERENCE API ENGINE
+# ============================
+def hf_generate_image(model, prompt):
+    url = f"https://api-inference.huggingface.co/models/{model}"
 
-    if not hasattr(output, "url"):
-        return None, None
+    payload = {
+        "inputs": prompt,
+        "options": {"wait_for_model": True}
+    }
 
-    url = output.url
+    response = requests.post(url, headers=HEADERS, json=payload)
+
+    if response.status_code != 200:
+        return None
+
+    img_bytes = response.content
+    filename = f"img_{uuid.uuid4().hex}.png"
 
     os.makedirs("outputs", exist_ok=True)
-    filename = f"nano_{uuid.uuid4().hex}.jpg"
     filepath = os.path.join("outputs", filename)
 
     with open(filepath, "wb") as f:
-        f.write(output.read())
+        f.write(img_bytes)
 
-    return url, filepath
+    return filepath
 
 # ============================
 # GROQ TEXT GENERATOR
@@ -82,8 +95,8 @@ def generate_panels(story, style, chapters):
 # ============================
 def build_visual_prompt(panel_text, style):
     return f"""
-    {style}, isometric 3D miniature diorama, soft pastel lighting,
-    cinematic depth, Nano Banana 2 optimized.
+    {style}, soft lighting, clean shapes, child-friendly,
+    cinematic depth, high detail.
     Scene: {panel_text}
     """
 
@@ -93,7 +106,7 @@ def build_visual_prompt(panel_text, style):
 class ComicPDF(FPDF):
     pass
 
-def build_pdf(story, style, chapters):
+def build_pdf(story, style, model, chapters):
     pdf = ComicPDF()
     pdf.set_auto_page_break(True, margin=15)
 
@@ -109,10 +122,10 @@ def build_pdf(story, style, chapters):
             pdf.cell(0, 8, f"Panel {p_idx}", 0, 1)
 
             visual_prompt = build_visual_prompt(panel, style)
-            url, path = nano_banana(visual_prompt)
+            img_path = hf_generate_image(model, visual_prompt)
 
-            if path:
-                pdf.image(path, x=20, w=170)
+            if img_path:
+                pdf.image(img_path, x=20, w=170)
 
             pdf.set_font("Helvetica", "", 11)
             pdf.multi_cell(0, 6, panel)
@@ -127,7 +140,7 @@ def build_pdf(story, style, chapters):
 # ============================
 with gr.Blocks() as app:
 
-    gr.Markdown("# 🌟 AIPromptLab 3.0 — Nano Banana Edition")
+    gr.Markdown("# 🌟 AIPromptLab 3.1 — FREE MODE (HuggingFace Engine)")
 
     with gr.Tabs():
 
@@ -149,11 +162,19 @@ with gr.Blocks() as app:
                 ],
                 value="Pastel 3D Isometric"
             )
+            model = gr.Dropdown(
+                label="Model Gratis",
+                choices=list(MODEL_ENDPOINTS.keys()),
+                value="Playground v2.5"
+            )
             chapters = gr.Slider(1, 10, value=1, step=1, label="Jumlah Bab")
             btn = gr.Button("Generate Comic PDF")
             out = gr.File()
 
-            btn.click(build_pdf, [story, style, chapters], out)
+            def run_comic(story, style, model, chapters):
+                return build_pdf(story, style, MODEL_ENDPOINTS[model], chapters)
+
+            btn.click(run_comic, [story, style, model, chapters], out)
 
         # ============================
         # TAB 2 — IMAGE GENERATOR
@@ -173,20 +194,22 @@ with gr.Blocks() as app:
                 ],
                 value="Pastel 3D Isometric"
             )
-            aspect = gr.Dropdown(["1:1", "16:9", "9:16"], label="Aspect Ratio", value="1:1")
-            res = gr.Dropdown(["1K", "2K"], label="Resolution", value="1K")
+            img_model = gr.Dropdown(
+                label="Model Gratis",
+                choices=list(MODEL_ENDPOINTS.keys()),
+                value="Playground v2.5"
+            )
             btn2 = gr.Button("Generate Image")
             img_out = gr.Image()
 
-            def generate_image(p, s, a, r):
-                prompt = f"{s}, Nano Banana optimized. Scene: {p}"
-                url, path = nano_banana(prompt, a, r)
-                return path
+            def generate_image(p, s, m):
+                prompt = f"{s}. Scene: {p}"
+                return hf_generate_image(MODEL_ENDPOINTS[m], prompt)
 
-            btn2.click(generate_image, [img_prompt, img_style, aspect, res], img_out)
+            btn2.click(generate_image, [img_prompt, img_style, img_model], img_out)
 
         # ============================
-        # TAB 3 — VIDEO GENERATOR (STORYBOARD)
+        # TAB 3 — VIDEO STORYBOARD
         # ============================
         with gr.Tab("Video Storyboard"):
             vid_desc = gr.Textbox(label="Deskripsi Video")
@@ -203,22 +226,27 @@ with gr.Blocks() as app:
                 ],
                 value="Pastel 3D Isometric"
             )
+            vid_model = gr.Dropdown(
+                label="Model Gratis",
+                choices=list(MODEL_ENDPOINTS.keys()),
+                value="SD Turbo"
+            )
             btn3 = gr.Button("Generate Storyboard Frames")
             frame_gallery = gr.Gallery(label="Frames")
 
-            def generate_storyboard(desc, style):
-                prompt = f"Buatkan 8 scene untuk storyboard video. Deskripsi: {desc}"
+            def generate_storyboard(desc, style, model):
+                prompt = f"Buatkan 8 scene storyboard. Deskripsi: {desc}"
                 scenes = ai(prompt).split("\n")
                 frames = []
 
                 for s in scenes[:8]:
                     visual = build_visual_prompt(s, style)
-                    url, path = nano_banana(visual)
-                    if path:
-                        frames.append(path)
+                    img_path = hf_generate_image(MODEL_ENDPOINTS[model], visual)
+                    if img_path:
+                        frames.append(img_path)
 
                 return frames
 
-            btn3.click(generate_storyboard, [vid_desc, vid_style], frame_gallery)
+            btn3.click(generate_storyboard, [vid_desc, vid_style, vid_model], frame_gallery)
 
 app.launch()