Update app.py

app.py

@@ -7,17 +7,15 @@ from flask_cors import CORS
 from gradio_client import Client
 
 app = Flask(__name__)
-# Aktifkan CORS untuk semua origin agar frontend manapun bisa akses
+# Aktifkan CORS untuk semua 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" 
+# --- KONFIGURASI TARGET (SESUAI PERINTAH) ---
+# Menggunakan direct link Spaces Flux yang kamu berikan
+TARGET_SPACE = "https://black-forest-labs-flux-2-dev.hf.space"
 
-# --- TOKEN LIST (HARDCODED) ---
-# Tanda '+' disisipkan agar lolos filter keamanan HF saat upload code.
-# Code akan otomatis menghapus '+' sebelum request ke API.
+# --- TOKEN LIST ---
+# Token sudah ditanam dengan tanda '+' agar lolos deteksi
 RAW_TOKENS = [
     "hf_+PiRCDDtPcPFMLWkTkVaZmzoleHOunXnLIA",
     "hf_+BHvZXGICstaktSwycmwNmzHGrTNmKxnlRZ",
@@ -36,42 +34,37 @@ RAW_TOKENS = [
 # --- TOKEN MANAGER ---
 class TokenManager:
     def __init__(self, raw_tokens):
-        # Membersihkan token dari tanda '+' saat inisialisasi
+        # Hapus tanda '+' dan spasi
         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) ---
+# --- LOGIKA GENERATE ---
 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
-    
+    # Batasi retry maksimal 5 kali agar tidak stuck
+    max_retries = 5
     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:]}")
+        print(f"[LOG] Mencoba request dengan token akhiran ...{current_token[-5:]}")
 
         try:
-            # Init Client
+            # Langsung konek ke URL yang kamu kasih
             client = Client(TARGET_SPACE, hf_token=current_token)
             
-            # Request ke Flux
+            # Parameter request
             result = client.predict(
                 prompt=prompt,
                 seed=int(seed) if seed is not None else 0,
@@ -83,33 +76,32 @@ def process_generation_with_retry(prompt, width, height, guidance_scale, seed):
                 api_name="/infer"
             )
             
-            # Flux mengembalikan tuple, path gambar ada di index 0
+            # Flux biasanya return tuple path file di index 0
             return result[0]
 
         except Exception as e:
             error_msg = str(e)
-            print(f"[ERROR] Token ...{current_token[-6:]} gagal: {error_msg}")
+            print(f"[ERROR] Token ...{current_token[-5:]} gagal: {error_msg}")
             
-            # Cek apakah errornya karena kuota/limit
+            # Cek limit atau kuota habis
             if "429" in error_msg or "quota" in error_msg.lower():
-                print("[INFO] Rate limit terdeteksi, ganti token...")
+                print("[INFO] Rate limit kena, ganti token selanjutnya...")
             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
+            time.sleep(0.5)
 
-    raise Exception(f"Gagal generate setelah {attempt} percobaan. Terakhir: {last_error}")
+    raise Exception(f"Gagal generate gambar. Error terakhir: {last_error}")
 
 # --- ROUTES ---
 
 @app.route('/')
 def home():
     return jsonify({
-        "status": "active",
-        "backend": "CPU-Proxy-Flux",
-        "tokens_loaded": len(token_manager.tokens)
+        "status": "Ready",
+        "target": TARGET_SPACE,
+        "tokens_active": len(token_manager.tokens)
     })
 
 @app.route('/generate', methods=['POST'])
@@ -117,9 +109,8 @@ def generate():
     data = request.json
     
     if not data or 'prompt' not in data:
-        return jsonify({"error": "Prompt is required"}), 400
+        return jsonify({"error": "Prompt wajib diisi"}), 400
 
-    # Ambil parameter atau pakai default
     prompt = data.get('prompt')
     width = data.get('width', 1024)
     height = data.get('height', 1024)
@@ -127,17 +118,12 @@ def generate():
     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
+        # Mengembalikan file gambar hasil generate
         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)
-