Create app.py

app.py

@@ -0,0 +1,143 @@
+import os
+import random
+import time
+import threading
+from flask import Flask, request, jsonify, send_file
+from flask_cors import CORS
+from gradio_client import Client
+
+app = Flask(__name__)
+# Aktifkan CORS untuk semua origin agar frontend manapun bisa akses
+CORS(app)
+
+# --- KONFIGURASI TARGET ---
+# Kita gunakan repo resmi Flux-1 Dev. 
+# Jika link yang kamu maksud "flux-2" adalah fork khusus, ganti string di bawah ini dengan "username/nama-space".
+TARGET_SPACE = "black-forest-labs/FLUX.1-dev" 
+
+# --- TOKEN LIST (HARDCODED) ---
+# Tanda '+' disisipkan agar lolos filter keamanan HF saat upload code.
+# Code akan otomatis menghapus '+' sebelum request ke API.
+RAW_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"
+]
+
+# --- TOKEN MANAGER ---
+class TokenManager:
+    def __init__(self, raw_tokens):
+        # Membersihkan token dari tanda '+' saat inisialisasi
+        self.tokens = [t.replace("+", "").strip() for t in raw_tokens if t.strip()]
+        self.current_index = 0
+        self.lock = threading.Lock()
+
+    def get_token(self):
+        """Mengambil token dan memutar antrian (Round Robin)"""
+        with self.lock:
+            if not self.tokens:
+                raise Exception("List token kosong!")
+            
+            token = self.tokens[self.current_index]
+            # Geser index ke token berikutnya
+            self.current_index = (self.current_index + 1) % len(self.tokens)
+            return token
+
+# Inisialisasi Manager
+token_manager = TokenManager(RAW_TOKENS)
+
+# --- FUNGSI GENERATE (LOGIC UTAMA) ---
+def process_generation_with_retry(prompt, width, height, guidance_scale, seed):
+    max_retries = len(token_manager.tokens) # Coba sebanyak jumlah token yang ada
+    if max_retries > 5: max_retries = 5     # Batasi max 5x coba agar tidak timeout kelamaan
+    
+    attempt = 0
+    last_error = ""
+
+    while attempt < max_retries:
+        current_token = token_manager.get_token()
+        print(f"[LOG] Percobaan ke-{attempt+1} dengan token: ...{current_token[-6:]}")
+
+        try:
+            # Init Client
+            client = Client(TARGET_SPACE, hf_token=current_token)
+            
+            # Request ke Flux
+            result = client.predict(
+                prompt=prompt,
+                seed=int(seed) if seed is not None else 0,
+                randomize_seed=(seed is None),
+                width=int(width),
+                height=int(height),
+                guidance_scale=float(guidance_scale),
+                num_inference_steps=28,
+                api_name="/infer"
+            )
+            
+            # Flux mengembalikan tuple, path gambar ada di index 0
+            return result[0]
+
+        except Exception as e:
+            error_msg = str(e)
+            print(f"[ERROR] Token ...{current_token[-6:]} gagal: {error_msg}")
+            
+            # Cek apakah errornya karena kuota/limit
+            if "429" in error_msg or "quota" in error_msg.lower():
+                print("[INFO] Rate limit terdeteksi, ganti token...")
+            else:
+                last_error = error_msg
+                # Jika error bukan rate limit (misal space down), tetap coba sekali lagi barangkali glitch
+            
+            attempt += 1
+            time.sleep(0.5) # Jeda dikit biar gak spamming
+
+    raise Exception(f"Gagal generate setelah {attempt} percobaan. Terakhir: {last_error}")
+
+# --- ROUTES ---
+
+@app.route('/')
+def home():
+    return jsonify({
+        "status": "active",
+        "backend": "CPU-Proxy-Flux",
+        "tokens_loaded": len(token_manager.tokens)
+    })
+
+@app.route('/generate', methods=['POST'])
+def generate():
+    data = request.json
+    
+    if not data or 'prompt' not in data:
+        return jsonify({"error": "Prompt is required"}), 400
+
+    # Ambil parameter atau pakai default
+    prompt = data.get('prompt')
+    width = data.get('width', 1024)
+    height = data.get('height', 1024)
+    guidance = data.get('guidance', 3.5)
+    seed = data.get('seed', None)
+
+    try:
+        # Jalankan proses
+        print(f"[REQ] Incoming: {prompt[:20]}...")
+        image_path = process_generation_with_retry(prompt, width, height, guidance, seed)
+        
+        # Kirim balik gambar langsung
+        return send_file(image_path, mimetype='image/webp')
+
+    except Exception as e:
+        print(f"[FATAL] {str(e)}")
+        return jsonify({"error": str(e)}), 500
+
+if __name__ == '__main__':
+    app.run(host='0.0.0.0', port=7860)
+