Update app.py

app.py

@@ -9,7 +9,7 @@ from PIL import Image
 from gradio_client import Client
 
 # ==========================================
-# 1. API Keys
+# 1. API Keys (এখানে একটি নতুন অপশন যুক্ত হয়েছে)
 # ==========================================
 KEYS = {
     "deepai": "16e69995-00d4-4ffd-993c-a026632f6de6",
@@ -23,7 +23,10 @@ KEYS = {
     "modelslab": "42zL3I6pH7Mv94jhNf4RGyuKGGT3GoEhfXZv47WwmQsfyhgC8XNttfOpvMKL",
     "openrouter": "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",
     "omniinfer": "sk_fTeDYbT2PibvuC27eGFbeE4E-7wcCuWp4SVgFDFWHDU",
-    "fireworks": "key_kH0svuL0"
+    "fireworks": "key_kH0svuL0",
+    
+    # ⚠️ নিচে তোমার Hugging Face Token টি দিতে হবে! না দিলে Slow API গুলো কাজ করবে না।
+    "huggingface": "hf_xxxxxxxxxxxxxxxxxxxxxxxxx" 
 }
 
 # ==========================================
@@ -86,7 +89,8 @@ def generate_pollinations(p, w, h, ar):
     raise Exception("Error")
 
 # ==========================================
-# 3. HF & Slow API Functions (SLOW)
+# 3. HF & Slow API Functions (SLOW) 
+# (এখানে প্যারামিটারগুলো ফিক্স করা হয়েছে)
 # ==========================================
 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}})
@@ -98,21 +102,36 @@ def generate_stable_horde(p, w, h, ar):
             if check.status_code == 200 and check.json().get("done", False): return get_image_from_url(check.json()['generations'][0]['img'])
     raise Exception("Error")
 
-def generate_hf_space(client_name, api_name, p):
-    try:
-        client = Client(client_name)
-        result = client.predict(p, api_name=api_name)
-        return Image.open(result[0] if isinstance(result, (list, tuple)) else result)
-    except:
-        raise Exception("Error")
+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):
-    # এই ফাংশনে সাইজ (width, height) ফিক্স করে দেওয়া হয়েছে!
     url = f"https://api-inference.huggingface.co/models/{model_id}"
+    headers = {"Authorization": f"Bearer {KEYS['huggingface']}"} if KEYS.get('huggingface') and KEYS['huggingface'] != "hf_xxxxxxxxxxxxxxxxxxxxxxxxx" else {}
+    
+    # প্রথমে সাইজ দিয়ে ট্রাই করবে
     payload = {"inputs": prompt, "parameters": {"width": w, "height": h}}
-    res = requests.post(url, json=payload)
-    if res.status_code == 200: return Image.open(BytesIO(res.content))
-    raise Exception("Error")
+    res = requests.post(url, headers=headers, json=payload)
+    
+    # অনেক HF মডেল সাইজ রিকোয়েস্ট সাপোর্ট করে না বলে ক্র্যাশ করে, তাই ফেইল করলে সাইজ ছাড়া ট্রাই করবে
+    if res.status_code != 200:
+        res = requests.post(url, headers=headers, json={"inputs": prompt})
+        
+    if res.status_code == 200:
+        return Image.open(BytesIO(res.content))
+    raise Exception(f"HF Error: {res.status_code}")
 
 # ==========================================
 # 4. API Categorization
@@ -132,29 +151,23 @@ FAST_APIS = {
 
 SLOW_APIS = {
     "Stable Horde (Crowdsourced)": generate_stable_horde,
-    "HF Space: FLUX.1": lambda p, w, h, ar: generate_hf_space("black-forest-labs/FLUX.1-schnell", "/infer", p),
-    "HF Space: SDXL Light": lambda p, w, h, ar: generate_hf_space("ByteDance/SDXL-Lightning", "/generate_image", p),
+    "HF Space: FLUX.1": generate_hf_flux,
+    "HF Space: SDXL Light": generate_hf_sdxl_lightning,
+    "HF Space: SD3.5": generate_hf_sd3,
     
-    # আগের ৪টি HF API
+    # 10 HF APIs
     "HF API: OpenJourney": lambda p, w, h, ar: generate_hf_api(p, "prompthero/openjourney", w, h),
     "HF API: RealVisXL": lambda p, w, h, ar: generate_hf_api(p, "SG161222/RealVisXL_V4.0", w, h),
     "HF API: Animagine XL": lambda p, w, h, ar: generate_hf_api(p, "cagliostrolab/animagine-xl-3.1", w, h),
     "HF API: Playground v2.5": lambda p, w, h, ar: generate_hf_api(p, "playgroundai/playground-v2.5-1024px-aesthetic", w, h),
-    
-    # তোমার চাওয়া নতুন ১০টি HF API! 
     "HF API: DreamShaper XL": lambda p, w, h, ar: generate_hf_api(p, "Lykon/dreamshaper-xl-1.0", w, h),
     "HF API: OpenDalle V1.1": lambda p, w, h, ar: generate_hf_api(p, "dataautogpt3/OpenDalleV1.1", w, h),
     "HF API: Anything V5 (Anime)": lambda p, w, h, ar: generate_hf_api(p, "stablediffusionapi/anything-v5", w, h),
-    "HF API: Disney Pixar Cartoon": lambda p, w, h, ar: generate_hf_api(p, "stablediffusionapi/disney-pixar-cartoon", w, h),
+    "HF API: Disney Pixar": lambda p, w, h, ar: generate_hf_api(p, "stablediffusionapi/disney-pixar-cartoon", w, h),
     "HF API: CyberRealistic": lambda p, w, h, ar: generate_hf_api(p, "stablediffusionapi/cyberrealistic", w, h),
-    "HF API: Absolute Reality": lambda p, w, h, ar: generate_hf_api(p, "stablediffusionapi/absolute-reality-v181", w, h),
-    "HF API: Juggernaut XL": lambda p, w, h, ar: generate_hf_api(p, "stablediffusionapi/juggernaut-xl-v5", w, h),
-    "HF API: SDXL Base 1.0": lambda p, w, h, ar: generate_hf_api(p, "stabilityai/stable-diffusion-xl-base-1.0", w, h),
-    "HF API: Realistic Vision V5": lambda p, w, h, ar: generate_hf_api(p, "SG161222/Realistic_Vision_V5.1_noVAE", w, h),
-    "HF API: Stable Diffusion 1.5": lambda p, w, h, ar: generate_hf_api(p, "runwayml/stable-diffusion-v1-5", w, h)
+    "HF API: Realistic Vision": lambda p, w, h, ar: generate_hf_api(p, "SG161222/Realistic_Vision_V5.1_noVAE", w, h)
 }
 
-# সব মিলিয়ে একটি মাস্টার ডিকশনারি (ম্যানুয়াল সিলেকশনের জন্য)
 ALL_APIS = {**FAST_APIS, **SLOW_APIS}
 
 # ==========================================
@@ -177,7 +190,6 @@ def generate_image_logic(prompt, style, size, speed_mode, selected_api):
     }
     enhanced_prompt = f"{prompt.strip()}{style_modifiers.get(style, '')}, aspect ratio {ar}"
 
-    # লজিক ১: অটোমেটিক (Fast বা Slow মোড অনুযায়ী)
     if selected_api == "অটোমেটিক (Auto Fallback)":
         target_apis = FAST_APIS if speed_mode == "Fast (ফাস্ট)" else SLOW_APIS
         yield None, f"⏳ {speed_mode} মোডে শুরু হচ্ছে..."
@@ -194,7 +206,6 @@ def generate_image_logic(prompt, style, size, speed_mode, selected_api):
                 continue
         yield None, "🚫 **দুঃখিত! এই মোডের সবগুলো সার্ভার বর্তমানে ব্যস্ত।**"
 
-    # লজিক ২: ম্যানুয়ালি নির্দিষ্ট API সিলেক্ট করলে
     else:
         yield None, f"⏳ শুধুমাত্র **{selected_api}**-এ কল করা হচ্ছে..."
         api_func = ALL_APIS[selected_api]
@@ -211,7 +222,7 @@ def generate_image_logic(prompt, style, size, speed_mode, selected_api):
 api_choices = ["অটোমেটিক (Auto Fallback)"] + list(ALL_APIS.keys())
 
 with gr.Blocks(theme=gr.themes.Base()) as demo:
-    gr.Markdown("<h1 style='text-align: center;'>🎨 Pro AI Image Generator (28 Servers)</h1>")
+    gr.Markdown("<h1 style='text-align: center;'>🎨 Pro AI Image Generator (24 Servers)</h1>")
     
     with gr.Row():
         with gr.Column(scale=1):
@@ -227,7 +238,6 @@ with gr.Blocks(theme=gr.themes.Base()) as demo:
                     label="ছবির সাইজ (Size)", value="স্কয়ার (1:1) - Instagram"
                 )
             
-            # নতুন Speed Mode অপশন
             speed_mode = gr.Radio(
                 choices=["Fast (ফাস্ট)", "Slow (স্লো) - High Quality"],
                 label="স্পিড মোড (অটোমেটিকের জন্য)", value="Fast (ফাস্ট)"