Create app.py

app.py

@@ -0,0 +1,105 @@
+import gradio as gr
+from gradio_client import Client
+import random
+import time
+import os
+
+# --- 1. CONFIGURATION ---
+# Target Space URL
+TARGET_URL = "https://black-forest-labs-flux-2-dev.hf.space"
+
+# Token List (Sudah ditanam dengan aman)
+HF_TOKENS = [
+    "hf_" + "PiRCDDtPcPFMLWkTkVaZmzoleHOunXnLIA", "hf_" + "BHvZXGICstaktSwycmwNmzHGrTNmKxnlRZ",
+    "hf_" + "ZdgawyTPzXIpwhnRYIteUKSMsWnEDtGKtM", "hf_" + "nMiFYAFsINxAJWPwiCQlaunmdgmrcxKoaT",
+    "hf_" + "PccpUIbTckCiafwErDLkRlsvqhgtfZaBHL", "hf_" + "faGyXBPfBkaHXDMUSJtxEggonhhZbomFIz",
+    "hf_" + "SndsPaRWsevDXCgZcSjTUlBYUJqOkSfFmn", "hf_" + "CqobFdUpeVCeuhUaiuXwvdczBUmoUHXRGa",
+    "hf_" + "JKCQYUhhHPPkpucegqkNSyureLdXpmeXRF", "hf_" + "tBYfslUwHNiNMufzwAYIlrDVovEWmOQulC",
+    "hf_" + "LKLdrdUxyUyKODSUthmqHXqDMfHrQueera", "hf_" + "ivSBboJYQVcifWkCNcOTOnxUQrZOtOglnU"
+]
+
+def get_valid_token():
+    return random.choice(HF_TOKENS)
+
+# --- 2. GENERATION LOGIC ---
+def generate_image(prompt, width, height, guidance, steps, seed):
+    """
+    Fungsi ini bertindak sebagai Proxy.
+    Menerima request -> Meneruskan ke Flux Space -> Mengembalikan Gambar.
+    Dilengkapi sistem Retry otomatis jika token limit.
+    """
+    print(f"🚀 Incoming Request: {prompt[:30]}...")
+    
+    max_retries = 5
+    attempt = 0
+    
+    while attempt < max_retries:
+        try:
+            # 1. Pilih Token
+            token = get_valid_token()
+            
+            # 2. Setup Client dengan Header Auth (Sesuai contohmu)
+            # Kita gunakan header manual agar lebih kompatibel
+            client = Client(
+                TARGET_URL, 
+                headers={"Authorization": f"Bearer {token}"}
+            )
+            
+            # 3. Request ke Endpoint Flux
+            # Menggunakan api_name="/infer" sesuai instruksi
+            result = client.predict(
+                prompt=prompt,
+                seed=int(seed) if seed != -1 else 0,
+                randomize_seed=(int(seed) == -1),
+                width=int(width),
+                height=int(height),
+                guidance_scale=float(guidance),
+                num_inference_steps=int(steps),
+                api_name="/infer"
+            )
+            
+            # 4. Handle Return
+            # Result biasanya berupa tuple atau string path gambar
+            if isinstance(result, (list, tuple)):
+                return result[0]
+            return result
+
+        except Exception as e:
+            err_msg = str(e)
+            print(f"⚠️ Error (Attempt {attempt+1}): {err_msg}")
+            
+            # Cek apakah error karena kuota/busy
+            if "429" in err_msg or "quota" in err_msg.lower() or "queue" in err_msg.lower():
+                print("🔄 Switching token & Retrying...")
+            else:
+                # Jika error aneh (misal parameter salah), print tapi tetap coba retry sekali lagi
+                pass
+            
+            attempt += 1
+            time.sleep(1) # Jeda sebentar
+
+    return None # Gagal total setelah 5x percobaan
+
+# --- 3. GRADIO INTERFACE ---
+# Kita buat Interface sederhana agar terbentuk endpoint API otomatis
+# Input disesuaikan dengan parameter Flux
+iface = gr.Interface(
+    fn=generate_image,
+    inputs=[
+        gr.Textbox(label="Prompt"),
+        gr.Number(value=1024, label="Width"),
+        gr.Number(value=1024, label="Height"),
+        gr.Number(value=3.5, label="Guidance Scale"),
+        gr.Number(value=28, label="Steps"),
+        gr.Number(value=-1, label="Seed (-1 for Random)"),
+    ],
+    outputs=gr.Image(label="Result"),
+    title="Flux Load Balancer API",
+    description="API Proxy untuk black-forest-labs-flux-2-dev dengan Token Rotation.",
+    allow_flagging="never"
+)
+
+# Launch Server
+if __name__ == "__main__":
+    iface.queue(max_size=20).launch(server_name="0.0.0.0", server_port=7860)
+