Update app.py

app.py

@@ -2,39 +2,56 @@ import os
 import requests
 import random
 import base64
+import time
 from datetime import datetime
 from fastapi import FastAPI, HTTPException
-from fastapi.middleware.cors import CORSMiddleware # <--- PENTING: Import ini
+from fastapi.middleware.cors import CORSMiddleware
 from pydantic import BaseModel
 from typing import Optional
 from gradio_client import Client
+from requests.adapters import HTTPAdapter, Retry
 
 app = FastAPI()
 
 # ==========================================
-# 0. SETUP CORS (AGAR PERCHANCE BISA AKSES)
+# 0. SETUP CORS & ROBUST REQUESTS SESSION
 # ==========================================
 app.add_middleware(
     CORSMiddleware,
-    allow_origins=["*"],  # Mengizinkan akses dari semua domain (termasuk Perchance)
+    allow_origins=["*"],
     allow_credentials=True,
-    allow_methods=["*"],  # Mengizinkan semua method (GET, POST, dll)
-    allow_headers=["*"],  # Mengizinkan semua header
+    allow_methods=["*"],
+    allow_headers=["*"],
 )
 
+# Fungsi untuk membuat session yang otomatis mencoba lagi jika gagal koneksi
+def requests_retry_session(
+    retries=3,
+    backoff_factor=1,
+    status_forcelist=(500, 502, 503, 504),
+    session=None,
+):
+    session = session or requests.Session()
+    retry = Retry(
+        total=retries,
+        read=retries,
+        connect=retries,
+        backoff_factor=backoff_factor,
+        status_forcelist=status_forcelist,
+    )
+    adapter = HTTPAdapter(max_retries=retry)
+    session.mount("http://", adapter)
+    session.mount("https://", adapter)
+    return session
+
 # ==========================================
-# 1. KONFIGURASI DAN TOKEN (ANTI-DETEKSI)
+# 1. KONFIGURASI & TOKEN (ANTI-DETEKSI)
 # ==========================================
-
-# Database Config (JSONBin)
 JSONBIN_ID = "6962ebfe43b1c97be927b07d"
 JSONBIN_KEY = "$2a$10$khoKkhYEAUCG.O0xHDHiY.Ei88KlSV1l3olI2pxc86mUSsTaNpJx6"
 JSONBIN_URL = f"https://api.jsonbin.io/v3/b/{JSONBIN_ID}"
-
-# Target Engine (Flux.2 Dev Space)
 ENGINE_URL = "https://black-forest-labs-flux-2-dev.hf.space"
 
-# TOKEN OBFUSCATION (Agar tidak terdeteksi scanner HF)
 TOKEN_PREFIX = "hf_" 
 TOKEN_BODIES = [
     "PiRCDDtPcPFMLWkTkVaZmzoleHOunXnLIA", "BHvZXGICstaktSwycmwNmzHGrTNmKxnlRZ",
@@ -50,48 +67,49 @@ VALID_TOKENS = []
 # ==========================================
 # 2. FUNGSI BANTUAN (DATABASE & UTILITAS)
 # ==========================================
-
 def check_tokens():
-    """Mengecek validitas token saat server startup"""
     global VALID_TOKENS
-    print("--- MEMULAI PENGECEKAN TOKEN ---")
+    print("--- CHECKING TOKENS ---")
     valid_list = []
     test_url = "https://huggingface.co/api/whoami-v2"
+    http = requests_retry_session(retries=2, backoff_factor=0.5) # Retry session
     
     for token in RAW_HF_TOKENS:
         headers = {"Authorization": f"Bearer {token}"}
         try:
-            r = requests.get(test_url, headers=headers, timeout=3)
+            r = http.get(test_url, headers=headers, timeout=5)
             if r.status_code == 200:
                 print(f"✅ Valid: ...{token[-6:]}")
                 valid_list.append(token)
             else:
                 print(f"❌ Invalid: ...{token[-6:]} (Status: {r.status_code})")
-        except:
-            print(f"⚠️ Error checking: ...{token[-6:]}")
+        except Exception as e:
+             print(f"⚠️ Error checking ...{token[-6:]}: {e}")
     
     VALID_TOKENS = valid_list
-    print(f"--- SELESAI. Total Token Siap Pakai: {len(VALID_TOKENS)} ---")
+    print(f"--- DONE. Total Active Tokens: {len(VALID_TOKENS)} ---")
 
 check_tokens()
 
 def get_db():
     headers = {"X-Master-Key": JSONBIN_KEY}
+    http = requests_retry_session()
     try:
-        response = requests.get(JSONBIN_URL, headers=headers)
-        if response.status_code == 200:
-            data = response.json().get("record", {})
-            if not isinstance(data, dict): return {}
-            return data
-        return {}
+        response = http.get(JSONBIN_URL, headers=headers, timeout=10)
+        response.raise_for_status() # Raise error untuk status 4xx/5xx
+        data = response.json().get("record", {})
+        if not isinstance(data, dict): return {}
+        return data
     except Exception as e:
         print(f"Error DB Read: {e}")
+        # Kembalikan dict kosong jika gagal total, agar server tidak crash
         return {}
 
 def save_db(data):
     headers = {"X-Master-Key": JSONBIN_KEY, "Content-Type": "application/json"}
+    http = requests_retry_session()
     try:
-        requests.put(JSONBIN_URL, json=data, headers=headers)
+        http.put(JSONBIN_URL, json=data, headers=headers, timeout=10)
     except Exception as e:
         print(f"Error DB Save: {e}")
 
@@ -99,7 +117,7 @@ def get_random_token():
     if not VALID_TOKENS:
         check_tokens()
         if not VALID_TOKENS:
-            raise HTTPException(status_code=500, detail="Semua Token HF Mati/Limit.")
+            raise HTTPException(status_code=503, detail="No valid HF tokens available. Service temporarily unavailable.")
     return random.choice(VALID_TOKENS)
 
 def image_to_base64(image_path):
@@ -113,7 +131,6 @@ def image_to_base64(image_path):
 # ==========================================
 # 3. MODEL DATA & ENDPOINTS
 # ==========================================
-
 class UserLogin(BaseModel):
     username: str
     password: str
@@ -130,68 +147,63 @@ class GenerateReq(BaseModel):
 
 @app.get("/")
 def home():
-    return {"status": "Online", "service": "FLUX2-GEN Backend", "tokens": len(VALID_TOKENS)}
+    return {"status": "Online", "service": "FLUX2-GEN Backend (Intl)", "active_tokens": len(VALID_TOKENS)}
 
 @app.get("/ping")
 def ping():
-    return {"status": "awake", "time": datetime.now().strftime("%H:%M:%S")}
+    return {"status": "awake", "time": datetime.now().strftime("%H:%M:%S UTC")}
 
 @app.post("/auth/signup")
 def signup(user: UserLogin):
     db = get_db()
     if user.username in db:
-        raise HTTPException(status_code=400, detail="Username sudah dipakai.")
+        raise HTTPException(status_code=400, detail="Username already taken.")
     
     today_str = datetime.now().strftime("%Y-%m-%d")
     db[user.username] = {
         "password": user.password,
         "credits": 10,
-        "last_reset": today_str,
-        "history": []
+        "last_reset": today_str
     }
     save_db(db)
-    return {"message": "Akun berhasil dibuat! Silakan login."}
+    return {"message": "Account created successfully."}
 
 @app.post("/auth/login")
 def login(user: UserLogin):
     db = get_db()
     if user.username not in db:
-        raise HTTPException(status_code=404, detail="User tidak ditemukan.")
+        raise HTTPException(status_code=404, detail="User not found.")
     
     stored_user = db[user.username]
     if stored_user["password"] != user.password:
-        raise HTTPException(status_code=401, detail="Password salah.")
+        raise HTTPException(status_code=401, detail="Incorrect password.")
     
     today_str = datetime.now().strftime("%Y-%m-%d")
-    msg = "Login sukses."
-    
+    msg = "Login successful."
     if stored_user.get("last_reset") != today_str:
         stored_user["credits"] = 10 
         stored_user["last_reset"] = today_str
         save_db(db)
-        msg = "Login sukses. Kredit harian di-reset ke 10."
+        msg = "Login successful. Daily credits reset to 10."
         
-    return {
-        "message": msg,
-        "credits": stored_user["credits"],
-        "history": stored_user.get("history", [])
-    }
+    return {"message": msg, "credits": stored_user["credits"]}
 
 @app.post("/generate")
 def generate_image(req: GenerateReq):
     db = get_db()
     if req.username not in db:
-        raise HTTPException(status_code=404, detail="User tidak valid.")
-    
+        raise HTTPException(status_code=401, detail="Unauthorized user.")
     user_data = db[req.username]
     
     if user_data["credits"] <= 0:
-        raise HTTPException(status_code=403, detail="Kredit harian habis! Kembali besok.")
+        raise HTTPException(status_code=403, detail="Daily credits exhausted. Please return tomorrow.")
 
     hf_token = get_random_token()
     
     try:
         print(f"🚀 Generating for {req.username}...")
+        # Inisialisasi Client. Gradio client menangani retry secara internal untuk beberapa kasus,
+        # tapi kita bungkus dalam try-except untuk menangkap error koneksi awal.
         client = Client(ENGINE_URL, hf_token=hf_token)
         
         result = client.predict(
@@ -209,29 +221,25 @@ def generate_image(req: GenerateReq):
         img_b64 = image_to_base64(img_path)
         
         if not img_b64:
-            raise Exception("Gagal memproses file output.")
+            raise Exception("Failed to process output image file.")
 
+        # Kurangi kredit hanya jika sukses
         user_data["credits"] -= 1
-        
-        new_history = {
-            "prompt": req.prompt,
-            "date": datetime.now().strftime("%Y-%m-%d %H:%M"),
-            "status": "success"
-        }
-        user_data["history"] = [new_history] + user_data.get("history", [])[:19]
         save_db(db)
         
         return {
             "status": "success",
             "image_base64": img_b64,
-            "remaining_credits": user_data["credits"],
-            "warning": "NSFW content is not supported."
+            "remaining_credits": user_data["credits"]
         }
 
     except Exception as e:
-        print(f"❌ Error Generasi: {e}")
+        print(f"❌ Generation Error: {e}")
         error_msg = str(e)
+        # Cek error spesifik NSFW dari Flux
         if "NSFW" in error_msg or "safety" in error_msg:
-             return {"status": "error", "message": "Prompt terdeteksi NSFW atau melanggar aturan Safety Filter."}
-        raise HTTPException(status_code=500, detail=f"Engine Error: {error_msg}")
+             return {"status": "error", "message": "NSFW content detected and blocked by the engine safety filter."}
+        
+        # Error umum lainnya (koneksi, timeout, dll)
+        raise HTTPException(status_code=502, detail=f"Engine Error: Failed to communicate with generation engine. {error_msg}")