Update app.py

app.py

@@ -11,7 +11,7 @@ from gradio_client import Client
 import concurrent.futures
 
 # ==========================================
-# 1. API Keys (টোকেন এখন সিক্রেট ভল্ট থেকে আসবে)
+# 1. API Keys (টোকেন সিক্রেট ভল্ট থেকে আসবে)
 # ==========================================
 KEYS = {
     "segmind": "SG_8d03bc9ab924555c",
@@ -19,8 +19,6 @@ KEYS = {
     "google": "AIzaSyBdEqDB6SFI2uQ-5FBr-cCeRj-dkiv8pFY",
     "stablehorde": "56kkbwcSPfHIr2YsDkpNDQ",
     "modelslab": "42zL3I6pH7Mv94jhNf4RGyuKGGT3GoEhfXZv47WwmQsfyhgC8XNttfOpvMKL",
-    
-    # 🔒 টোকেন এখন সরাসরি কোডে নেই, সিক্রেট সেটিংস থেকে আসবে
     "huggingface": os.environ.get("HF_TOKEN") 
 }
 
@@ -36,33 +34,33 @@ def get_image_from_url(url):
 # ==========================================
 def generate_hercai(p, w, h, ar):
     url = f"https://hercai.onrender.com/v3/text2image?prompt={urllib.parse.quote(p)}"
-    res = requests.get(url, headers=HEADERS_DEFAULT, timeout=20)
+    res = requests.get(url, headers=HEADERS_DEFAULT, timeout=15)
     if res.status_code == 200: return get_image_from_url(res.json().get("url"))
     raise Exception(f"Hercai Error {res.status_code}")
 
 def generate_pollinations(p, w, h, ar):
     url = f"https://image.pollinations.ai/prompt/{urllib.parse.quote(p)}?width={w}&height={h}&nologo=true"
-    res = requests.get(url, headers=HEADERS_DEFAULT, timeout=20)
+    res = requests.get(url, headers=HEADERS_DEFAULT, timeout=15)
     if res.status_code == 200: return Image.open(BytesIO(res.content))
     raise Exception(f"Pollinations Error {res.status_code}")
 
 def generate_google_imagen(p, w, h, ar):
-    res = requests.post(f"https://generativelanguage.googleapis.com/v1beta/models/imagen-3.0-generate-001:predict?key={KEYS['google']}", json={"instances": [{"prompt": p}]}, timeout=20)
+    res = requests.post(f"https://generativelanguage.googleapis.com/v1beta/models/imagen-3.0-generate-001:predict?key={KEYS['google']}", json={"instances": [{"prompt": p}]}, timeout=15)
     if res.status_code == 200: return Image.open(BytesIO(base64.b64decode(res.json()['predictions'][0]['bytesBase64Encoded'])))
     raise Exception(f"Google Error {res.status_code}")
 
 def generate_segmind(p, w, h, ar):
-    res = requests.post("https://api.segmind.com/v1/sdxl1.0-txt2img", headers={"x-api-key": KEYS['segmind']}, json={"prompt": p, "samples": 1, "width": w, "height": h}, timeout=20)
+    res = requests.post("https://api.segmind.com/v1/sdxl1.0-txt2img", headers={"x-api-key": KEYS['segmind']}, json={"prompt": p, "samples": 1, "width": w, "height": h}, timeout=15)
     if res.status_code == 200: return Image.open(BytesIO(res.content))
     raise Exception(f"Segmind Error {res.status_code}")
 
 def generate_getimg(p, w, h, ar):
-    res = requests.post("https://api.getimg.ai/v1/essential/text-to-image", headers={"Authorization": f"Bearer {KEYS['getimg']}"}, json={"prompt": p, "width": w, "height": h}, timeout=20)
+    res = requests.post("https://api.getimg.ai/v1/essential/text-to-image", headers={"Authorization": f"Bearer {KEYS['getimg']}"}, json={"prompt": p, "width": w, "height": h}, timeout=15)
     if res.status_code == 200: return Image.open(BytesIO(base64.b64decode(res.json()["image"])))
     raise Exception(f"GetImg Error {res.status_code}")
 
 def generate_modelslab(p, w, h, ar):
-    res = requests.post("https://modelslab.com/api/v6/images/text2img", json={"key": KEYS['modelslab'], "prompt": p, "width": str(w), "height": str(h)}, timeout=20)
+    res = requests.post("https://modelslab.com/api/v6/images/text2img", json={"key": KEYS['modelslab'], "prompt": p, "width": str(w), "height": str(h)}, timeout=15)
     if res.status_code == 200 and "output" in res.json(): return get_image_from_url(res.json()["output"][0])
     raise Exception(f"ModelsLab Error {res.status_code}")
 
@@ -70,30 +68,19 @@ def generate_stable_horde(p, w, h, ar):
     res = requests.post("https://stablehorde.net/api/v2/generate/async", headers={"apikey": KEYS['stablehorde']}, json={"prompt": p, "params": {"n": 1, "width": 512, "height": 512}}, timeout=10)
     if res.status_code == 202:
         job_id = res.json()['id']
-        for _ in range(10): 
+        for _ in range(8): 
             time.sleep(2)
             check = requests.get(f"https://stablehorde.net/api/v2/generate/status/{job_id}", timeout=10)
             if check.status_code == 200 and check.json().get("done", False): return get_image_from_url(check.json()['generations'][0]['img'])
     raise Exception("Horde Timeout")
 
-def generate_hf_flux(p, w, h, ar):
-    client = Client("black-forest-labs/FLUX.1-schnell")
-    res = client.predict(prompt=p, seed=0, randomize_seed=True, width=w, height=h, num_inference_steps=4, api_name="/infer")
-    return Image.open(res[0])
-
-def generate_hf_sdxl_lightning(p, w, h, ar):
-    client = Client("ByteDance/SDXL-Lightning")
-    res = client.predict(p, "4-Step", api_name="/generate_image")
-    return Image.open(res)
-
-def generate_hf_sd3(p, w, h, ar):
-    client = Client("stabilityai/stable-diffusion-3.5-large")
-    res = client.predict(prompt=p, negative_prompt="", seed=0, randomize_seed=True, width=w, height=h, guidance_scale=4.5, num_inference_steps=40, api_name="/infer")
-    return Image.open(res[0])
-
 def generate_hf_api(prompt, model_id, w, h):
     url = f"https://api-inference.huggingface.co/models/{model_id}"
-    headers = {"Authorization": f"Bearer {KEYS['huggingface']}"} if KEYS.get('huggingface') else {}
+    # টোকেন চেক
+    if not KEYS.get('huggingface'):
+        raise Exception("Token Missing")
+        
+    headers = {"Authorization": f"Bearer {KEYS['huggingface']}"}
     payload = {"inputs": prompt, "parameters": {"width": w, "height": h}}
     res = requests.post(url, headers=headers, json=payload, timeout=20)
     
@@ -113,10 +100,7 @@ ALL_APIS = {
     "Segmind (Daily)": generate_segmind,
     "GetImg AI (Monthly)": generate_getimg,
     "ModelsLab (Daily)": generate_modelslab,
-    "Stable Horde (Unlimited)": generate_stable_horde,
-    "HF Space: FLUX.1": generate_hf_flux,
-    "HF Space: SDXL Light": generate_hf_sdxl_lightning,
-    "HF Space: SD3.5": generate_hf_sd3,
+    "Stable Horde (Unlimited)": generate_stable_horde
 }
 
 hf_models_list = [
@@ -189,7 +173,7 @@ for name, model_id in hf_models_list:
     ALL_APIS[f"HF: {name}"] = make_hf_func(model_id)
 
 # ==========================================
-# 4. Smart Random Routing Logic 
+# 4. Safe Smart Routing Logic (Batch = 3)
 # ==========================================
 def fetch_api_task(api_name, api_func, prompt, w, h, ar):
     try:
@@ -199,6 +183,11 @@ def fetch_api_task(api_name, api_func, prompt, w, h, ar):
         return None, api_name, str(e)
 
 def generate_image_logic(prompt, style, size, selected_api):
+    # টোকেন চেক: যদি টোকেন না থাকে, তবে ইউজারকে ওয়ার্নিং দেবে
+    if not KEYS.get("huggingface"):
+        yield None, "⚠️ **Secret Token পাওয়া যায়নি!** দয়া করে Settings > Factory Rebuild এ ক্লিক করুন।"
+        return
+
     if not prompt.strip():
         yield None, "⚠️ **দয়া করে একটি প্রম্পট লিখুন!**"
         return
@@ -216,16 +205,17 @@ def generate_image_logic(prompt, style, size, selected_api):
     enhanced_prompt = f"{prompt.strip()}{style_modifiers.get(style, '')}, aspect ratio {ar}"
 
     if selected_api == "অটোমেটিক (Auto Fallback)":
-        yield None, "⏳ ২০০টি সার্ভার থেকে লটারির মাধ্যমে ১০টি র‍্যান্ডম সার্ভার বেছে নেওয়া হচ্ছে..."
+        yield None, "⏳ নিরাপদভাবে ৩টি র‍্যান্ডম সার্ভার বেছে নেওয়া হচ্ছে..."
         
+        # আইপি ব্লক এড়াতে একসাথে ১০টার বদলে ৩টা সার্ভার বেছে নেবে
         all_api_items = list(ALL_APIS.items())
-        selected_batch = random.sample(all_api_items, 10)
+        selected_batch = random.sample(all_api_items, 3)
         
         successful_images = []
         successful_names = []
         error_logs = []
         
-        with concurrent.futures.ThreadPoolExecutor(max_workers=10) as executor:
+        with concurrent.futures.ThreadPoolExecutor(max_workers=3) as executor:
             futures = {executor.submit(fetch_api_task, name, func, enhanced_prompt, w, h, ar): name for name, func in selected_batch}
             
             for future in concurrent.futures.as_completed(futures):
@@ -241,7 +231,7 @@ def generate_image_logic(prompt, style, size, selected_api):
             yield successful_images, f"✅ **সফল!** {len(successful_images)} টি সার্ভার ছবি বানাতে পেরেছে।\n(উৎস: {source_text})"
             return 
             
-        yield None, f"🚫 **দুঃখিত!** লটারিতে ওঠা ১০টি সার্ভারই এখন ব্যস্ত। দয়া করে আবার 'Generate Image' এ ক্লিক করুন (এবার নতুন ১০টি সার্ভার খুঁজবে)।"
+        yield None, f"🚫 **দুঃখিত!** সার্ভারগুলো এখন ব্যস্ত। দয়া করে আবার 'Generate Image' এ ক্লিক করুন।"
 
     else:
         yield None, f"⏳ শুধুমাত্র **{selected_api}**-এ কল করা হচ্ছে..."
@@ -259,7 +249,7 @@ def generate_image_logic(prompt, style, size, selected_api):
 api_choices = ["অটোমেটিক (Auto Fallback)"] + list(ALL_APIS.keys())
 
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
-    gr.Markdown("<h1 style='text-align: center;'>🎨 Mega AI Image Generator (Smart Auto-Routing)</h1>")
+    gr.Markdown("<h1 style='text-align: center;'>🎨 Mega AI Image Generator (Safe Routing)</h1>")
     
     with gr.Row():
         with gr.Column(scale=1):
@@ -283,7 +273,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
             log_box = gr.Markdown("স্ট্যাটাস: অপেক্ষমান...")
         
         with gr.Column(scale=1):
-            image_output = gr.Gallery(label="জেনারেট হওয়া ছবি (একাধিক হতে পারে)", columns=2, object_fit="contain", height="auto")
+            image_output = gr.Gallery(label="জেনারেট হওয়া ছবি", columns=2, object_fit="contain", height="auto")
 
     generate_btn.click(
         fn=generate_image_logic,