Update app.py

app.py

@@ -1,6 +1,5 @@
 import gradio as gr
 import requests
-import json
 import time
 import base64
 import urllib.parse
@@ -9,25 +8,19 @@ from PIL import Image
 from gradio_client import Client
 
 # ==========================================
-# 1. API Keys (তোমার দেওয়া লিস্ট)
+# 1. API Keys (শুধু দরকারি এবং ডেইলি রিস্টার্ট হওয়াগুলো)
 # ==========================================
 KEYS = {
-    "deepai": "16e69995-00d4-4ffd-993c-a026632f6de6",
     "segmind": "SG_8d03bc9ab924555c",
     "getimg": "sk_6bdYCHa5p3E3dFJKIFuQI_8788b3203803717ab8980138a9a13625f1b4bc2419322bf1ad5e593a6734add2",
-    "edenai": "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJ1c2VyX2lkIjoiYWU0MzlhYTctMWYyYS00YjYwLWI1ZjctNDllY2Y5NTlhMWJlIiwidHlwZSI6ImFwaV90b2tlbiJ9.dr3_nbmwcjwIp0L19L2e11H2IoAvELRrFaTFNvSxS_8",
-    "stability": "sk-JmNBxpvE2AD5JWJEVbYSR0sjosJD4fLU4hh5L6j88SQ3gwEA",
-    "tensorart": "ak_tensor_uXCo6TvveNBDpXgKoiGCWG-6fmsDMYdi0kM2UH5rgUk",
     "google": "AIzaSyBdEqDB6SFI2uQ-5FBr-cCeRj-dkiv8pFY",
     "stablehorde": "56kkbwcSPfHIr2YsDkpNDQ",
     "modelslab": "42zL3I6pH7Mv94jhNf4RGyuKGGT3GoEhfXZv47WwmQsfyhgC8XNttfOpvMKL",
-    "openrouter": "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",
-    "omniinfer": "sk_fTeDYbT2PibvuC27eGFbeE4E-7wcCuWp4SVgFDFWHDU",
-    "fireworks": "key_kH0svuL0",
-    "huggingface": "hf_xxxxxxxxxxxxxxxxxxxxxxxxx" # স্লো মোডের জন্য তোমার HF Token এখানে দিও
+    
+    # ⚠️ তোমার Hugging Face Token এখানে দিতে ভুলবে না
+    "huggingface": "hf_xxxxxxxxxxxxxxxxxxxxxxxxx" 
 }
 
-# ব্রাউজার সিকিউরিটি (যাতে স্প্যাম ভেবে ব্লক না করে)
 HEADERS_DEFAULT = {"User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36"}
 
 def get_image_from_url(url):
@@ -36,70 +29,40 @@ def get_image_from_url(url):
     raise Exception(f"Image load failed: {res.status_code}")
 
 # ==========================================
-# 2. FAST APIs (তোমার প্রাইভেট + নতুন ফ্রি ফাস্ট)
+# 2. Core API Functions (Unlimited & Daily)
 # ==========================================
 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)
     if res.status_code == 200: return get_image_from_url(res.json().get("url"))
-    raise Exception(f"Hercai Error: {res.status_code}")
+    raise Exception("Hercai Error")
 
 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)
     if res.status_code == 200: return Image.open(BytesIO(res.content))
-    raise Exception(f"Pollinations Error: {res.status_code}")
-
-def generate_stability(p, w, h, ar):
-    res = requests.post("https://api.stability.ai/v2beta/stable-image/generate/core", headers={"Authorization": f"Bearer {KEYS['stability']}", "Accept": "image/*"}, files={"none": ''}, data={"prompt": p, "output_format": "jpeg", "aspect_ratio": ar})
-    if res.status_code == 200: return Image.open(BytesIO(res.content))
-    raise Exception(f"API 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})
-    if res.status_code == 200: return Image.open(BytesIO(base64.b64decode(res.json()["image"])))
-    raise Exception(f"API 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})
-    if res.status_code == 200: return Image.open(BytesIO(res.content))
-    raise Exception(f"API Error {res.status_code}")
-
-def generate_fireworks(p, w, h, ar):
-    res = requests.post("https://api.fireworks.ai/inference/v1/image_generation/accounts/fireworks/models/stable-diffusion-xl-1024-v1-0", headers={"Authorization": f"Bearer {KEYS['fireworks']}"}, json={"text_prompts": [{"text": p}], "width": w, "height": h})
-    if res.status_code == 200: return Image.open(BytesIO(base64.b64decode(res.json()[0]["base64"])))
-    raise Exception(f"API Error {res.status_code}")
+    raise Exception("Pollinations Error")
 
 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}]})
     if res.status_code == 200: return Image.open(BytesIO(base64.b64decode(res.json()['predictions'][0]['bytesBase64Encoded'])))
-    raise Exception(f"API Error {res.status_code}")
+    raise Exception("Google Error")
 
-def generate_deepai(p, w, h, ar):
-    res = requests.post("https://api.deepai.org/api/text2img", data={'text': p}, headers={'api-key': KEYS['deepai']})
-    if res.status_code == 200: return get_image_from_url(res.json()['output_url'])
-    raise Exception(f"API 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})
+    if res.status_code == 200: return Image.open(BytesIO(res.content))
+    raise Exception("Segmind Error")
 
-def generate_edenai(p, w, h, ar):
-    res = requests.post("https://api.edenai.run/v2/image/generation", headers={"Authorization": f"Bearer {KEYS['edenai']}"}, json={"providers": "openai", "text": p, "resolution": "1024x1024"})
-    if res.status_code == 200: return get_image_from_url(res.json()['openai']['items'][0]['image_resource_url'])
-    raise Exception(f"API 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})
+    if res.status_code == 200: return Image.open(BytesIO(base64.b64decode(res.json()["image"])))
+    raise Exception("GetImg Error")
 
 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)})
     if res.status_code == 200 and "output" in res.json(): return get_image_from_url(res.json()["output"][0])
-    raise Exception(f"API Error {res.status_code}")
-
-def generate_openrouter(p, w, h, ar):
-    res = requests.post("https://openrouter.ai/api/v1/chat/completions", headers={"Authorization": f"Bearer {KEYS['openrouter']}"}, json={"model": "google/imagen-3-fast-generate", "messages": [{"role": "user", "content": p}]})
-    if res.status_code == 200:
-        content = res.json()['choices'][0]['message']['content']
-        if "http" in content: return get_image_from_url(content.split("(")[-1].split(")")[0] if "(" in content else content.strip())
-    raise Exception(f"API Error {res.status_code}")
+    raise Exception("ModelsLab Error")
 
-# ==========================================
-# 3. SLOW APIs (Hugging Face & Public)
-# ==========================================
 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}})
     if res.status_code == 202:
@@ -108,7 +71,7 @@ def generate_stable_horde(p, w, h, ar):
             time.sleep(2)
             check = requests.get(f"https://stablehorde.net/api/v2/generate/status/{job_id}")
             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")
+    raise Exception("Horde Error")
 
 def generate_hf_flux(p, w, h, ar):
     client = Client("black-forest-labs/FLUX.1-schnell")
@@ -133,48 +96,81 @@ def generate_hf_api(prompt, model_id, w, h):
     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}")
+    raise Exception("HF API Error")
 
 # ==========================================
-# 4. API Categorization (Total 30 Servers!)
+# 3. MEGA API DICTIONARY (Exactly 50 Servers)
 # ==========================================
-FAST_APIS = {
-    "Hercai AI (Free Fast)": generate_hercai,
+ALL_APIS = {
+    # ১০টি কোর এবং আনলিমিটেড সার্ভার
     "Pollinations AI (Unlimited)": generate_pollinations,
-    "Google Imagen": generate_google_imagen,
-    "Stability AI": generate_stability, 
-    "GetImg AI": generate_getimg, 
-    "Segmind": generate_segmind, 
-    "Fireworks": generate_fireworks,
-    "OpenRouter": generate_openrouter,
-    "DeepAI": generate_deepai,
-    "Eden AI": generate_edenai,
-    "ModelsLab": generate_modelslab
-}
-
-SLOW_APIS = {
-    "Stable Horde (Crowdsourced)": generate_stable_horde,
+    "Hercai AI (Unlimited)": generate_hercai,
+    "Google Imagen (Daily)": generate_google_imagen,
+    "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,
-    "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: 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": 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: 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}
+# ৪০টি বেস্ট Hugging Face মডেলের লিস্ট
+hf_models_list = {
+    "HF: DreamShaper XL": "Lykon/dreamshaper-xl-1.0",
+    "HF: RealVisXL V4.0": "SG161222/RealVisXL_V4.0",
+    "HF: OpenJourney": "prompthero/openjourney",
+    "HF: Animagine XL 3.1": "cagliostrolab/animagine-xl-3.1",
+    "HF: Playground v2.5": "playgroundai/playground-v2.5-1024px-aesthetic",
+    "HF: Realistic Vision V5": "SG161222/Realistic_Vision_V5.1_noVAE",
+    "HF: Anything V5 (Anime)": "stablediffusionapi/anything-v5",
+    "HF: Disney Pixar": "stablediffusionapi/disney-pixar-cartoon",
+    "HF: CyberRealistic": "stablediffusionapi/cyberrealistic",
+    "HF: Absolute Reality": "stablediffusionapi/absolute-reality-v181",
+    "HF: Juggernaut XL": "stablediffusionapi/juggernaut-xl-v5",
+    "HF: SDXL Base 1.0": "stabilityai/stable-diffusion-xl-base-1.0",
+    "HF: Stable Diffusion 1.5": "runwayml/stable-diffusion-v1-5",
+    "HF: Stable Diffusion 2.1": "stabilityai/stable-diffusion-2-1",
+    "HF: EpicRealism": "emilianJR/epiCRealism",
+    "HF: Deliberate": "XpucT/Deliberate",
+    "HF: MeinaMix": "Meina/MeinaMix_V11",
+    "HF: DreamShaper 8": "Lykon/dreamshaper-8",
+    "HF: GhostMix": "GhostMix/GhostMix-V2.0",
+    "HF: Pastel-Mix": "andite/pastel-mix",
+    "HF: Protogen x3.4": "darkstorm2150/Protogen_x3.4_Official_Release",
+    "HF: RevAnimated": "stablediffusionapi/rev-animated",
+    "HF: Comic Babes": "stablediffusionapi/comic-babes",
+    "HF: 3D Animation Style": "stablediffusionapi/3d-animation-style",
+    "HF: NeverEnding Dream": "stablediffusionapi/neverending-dream",
+    "HF: Reliberate": "stablediffusionapi/reliberate",
+    "HF: Edge of Realism": "stablediffusionapi/edge-of-realism",
+    "HF: AOM3 (Anime)": "stablediffusionapi/abyssorangemix3",
+    "HF: AnyLoRA": "stablediffusionapi/anylora",
+    "HF: Counterfeit V3": "stablediffusionapi/counterfeit-v30",
+    "HF: Dark Sushi Mix": "stablediffusionapi/dark-sushi-mix",
+    "HF: Photoreal 2.0": "dreamlike-art/dreamlike-photoreal-2.0",
+    "HF: Dreamlike Anime": "dreamlike-art/dreamlike-anime-1.0",
+    "HF: Dreamlike Diffusion": "dreamlike-art/dreamlike-diffusion-1.0",
+    "HF: OpenDalle V1.1": "dataautogpt3/OpenDalleV1.1",
+    "HF: Nitro Diffusion": "nitrosocke/Nitro-Diffusion",
+    "HF: Arcane Diffusion": "nitrosocke/Arcane-Diffusion",
+    "HF: Mo-Di Diffusion": "nitrosocke/mo-di-diffusion",
+    "HF: Elden Ring Style": "nitrosocke/elden-ring-diffusion",
+    "HF: Spider-Verse": "nitrosocke/spider-verse-diffusion"
+}
+
+# ফাংশন জেনারেটর (কোড ছোট রাখার জন্য)
+def make_hf_func(model_id):
+    return lambda p, w, h, ar: generate_hf_api(p, model_id, w, h)
+
+for name, model_id in hf_models_list.items():
+    ALL_APIS[name] = make_hf_func(model_id)
+
 
 # ==========================================
-# 5. Generation Logic 
+# 4. Single Generation Logic
 # ==========================================
-def generate_image_logic(prompt, style, size, speed_mode, selected_api):
+def generate_image_logic(prompt, style, size, selected_api):
     if not prompt.strip():
         yield None, "⚠️ **দয়া করে একটি প্রম্পট লিখুন!**"
         return
@@ -192,10 +188,9 @@ def generate_image_logic(prompt, style, size, speed_mode, selected_api):
     enhanced_prompt = f"{prompt.strip()}{style_modifiers.get(style, '')}, aspect ratio {ar}"
 
     if selected_api == "অটোমেটিক (Auto Fallback)":
-        target_apis = FAST_APIS if speed_mode == "Fast (ফাস্ট)" else SLOW_APIS
-        yield None, f"⏳ {speed_mode} মোডে শুরু হচ্ছে..."
+        yield None, "⏳ ছবি তৈরি শুরু হচ্ছে..."
         
-        for api_name, api_func in target_apis.items():
+        for api_name, api_func in ALL_APIS.items():
             yield None, f"⏳ **{api_name}**-এ কল করা হচ্ছে..."
             try:
                 image = api_func(enhanced_prompt, w, h, ar)
@@ -203,11 +198,9 @@ def generate_image_logic(prompt, style, size, speed_mode, selected_api):
                     yield image, f"✅ **সফল!** অটোমেটিক {api_name} থেকে ছবি তৈরি হয়েছে।"
                     return 
             except Exception as e:
-                # 🔴 এখানে আসল কারণ দেখাবে (যেমন API Error 401 মানে লিমিট শেষ)
-                error_text = str(e)[:30]
-                yield None, f"❌ **{api_name}** ব্যর্থ ({error_text}). পরেরটিতে যাচ্ছে..."
+                yield None, f"❌ **{api_name}** ব্যস্ত বা লিমিট শেষ। পরেরটিতে যাচ্ছে..."
                 continue
-        yield None, f"🚫 **দুঃখিত! {speed_mode} মোডের সবগুলো সার্ভার বর্তমানে ব্যস্ত বা লিমিট শেষ।**"
+        yield None, "🚫 **দুঃখিত! বর্তমানে সবগুলো সার্ভার ব্যস্ত আছে।**"
 
     else:
         yield None, f"⏳ শুধুমাত্র **{selected_api}**-এ কল করা হচ্ছে..."
@@ -217,16 +210,15 @@ def generate_image_logic(prompt, style, size, speed_mode, selected_api):
             if image: yield image, f"✅ **সফল!** আপনি ম্যানুয়ালি {selected_api} ব্যবহার করেছেন।"
             else: yield None, f"❌ **দুঃখিত!** {selected_api} বর্তমানে ডাউন।"
         except Exception as e:
-            error_text = str(e)[:30]
-            yield None, f"❌ **দুঃখিত!** ব্যর্থ হয়েছে ({error_text})। অন্য একটি ট্রাই করুন।"
+            yield None, f"❌ **দুঃখিত!** ব্যর্থ হয়েছে। অন্য একটি সার্ভার ট্রাই করুন।"
 
 # ==========================================
-# 6. Gradio Web UI
+# 5. Clean Gradio Web UI
 # ==========================================
 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 (30 Servers)</h1>")
+with gr.Blocks(theme=gr.themes.Soft()) as demo:
+    gr.Markdown("<h1 style='text-align: center;'>🎨 Pro AI Image Generator (50 Servers)</h1>")
     
     with gr.Row():
         with gr.Column(scale=1):
@@ -242,13 +234,8 @@ with gr.Blocks(theme=gr.themes.Base()) as demo:
                     label="ছবির সাইজ (Size)", value="স্কয়ার (1:1) - Instagram"
                 )
             
-            speed_mode = gr.Radio(
-                choices=["Fast (ফাস্ট)", "Slow (স্লো) - High Quality"],
-                label="স্পিড মোড (অটোমেটিকের জন্য)", value="Fast (ফাস্ট)"
-            )
-            
             api_selector = gr.Dropdown(
-                choices=api_choices, label="সার্ভার ম্যানুয়ালি সিলেক্ট করুন", value="অটোমেটিক (Auto Fallback)", interactive=True
+                choices=api_choices, label="সার্ভার সিলেক্ট করুন (ডিফল্ট: অটোমেটিক)", value="অটোমেটিক (Auto Fallback)", interactive=True
             )
             
             generate_btn = gr.Button("Generate Image", variant="primary")
@@ -259,7 +246,7 @@ with gr.Blocks(theme=gr.themes.Base()) as demo:
 
     generate_btn.click(
         fn=generate_image_logic,
-        inputs=[prompt_input, style_input, size_input, speed_mode, api_selector],
+        inputs=[prompt_input, style_input, size_input, api_selector],
         outputs=[image_output, log_box]
     )