Update app.py

app.py

@@ -25,7 +25,7 @@ KEYS = {
 }
 
 # ==========================================
-# 2. API Functions (আলাদা আলাদা ওয়েবসাইটের জন্য)
+# 2. API Functions
 # ==========================================
 
 def get_image_from_url(url):
@@ -33,7 +33,6 @@ def get_image_from_url(url):
     return Image.open(BytesIO(response.content))
 
 def generate_stability(prompt):
-    print("Trying Stability AI...")
     response = requests.post(
         "https://api.stability.ai/v2beta/stable-image/generate/core",
         headers={"Authorization": f"Bearer {KEYS['stability']}", "Accept": "image/*"},
@@ -44,8 +43,18 @@ def generate_stability(prompt):
         return Image.open(BytesIO(response.content)), "Stability AI"
     raise Exception("Stability Error")
 
+def generate_getimg(prompt):
+    response = requests.post(
+        "https://api.getimg.ai/v1/essential/text-to-image",
+        headers={"Authorization": f"Bearer {KEYS['getimg']}"},
+        json={"prompt": prompt}
+    )
+    if response.status_code == 200:
+        image_data = base64.b64decode(response.json()["image"])
+        return Image.open(BytesIO(image_data)), "GetImg AI"
+    raise Exception("GetImg Error")
+
 def generate_segmind(prompt):
-    print("Trying Segmind...")
     response = requests.post(
         "https://api.segmind.com/v1/sdxl1.0-txt2img",
         headers={"x-api-key": KEYS['segmind']},
@@ -55,20 +64,18 @@ def generate_segmind(prompt):
         return Image.open(BytesIO(response.content)), "Segmind (SDXL)"
     raise Exception("Segmind Error")
 
-def generate_getimg(prompt):
-    print("Trying GetImg AI...")
+def generate_fireworks(prompt):
     response = requests.post(
-        "https://api.getimg.ai/v1/essential/text-to-image",
-        headers={"Authorization": f"Bearer {KEYS['getimg']}"},
-        json={"prompt": prompt}
+        "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": prompt}]}
     )
     if response.status_code == 200:
-        image_data = base64.b64decode(response.json()["image"])
-        return Image.open(BytesIO(image_data)), "GetImg AI"
-    raise Exception("GetImg Error")
+        image_data = base64.b64decode(response.json()[0]["base64"])
+        return Image.open(BytesIO(image_data)), "Fireworks AI"
+    raise Exception("Fireworks Error")
 
 def generate_deepai(prompt):
-    print("Trying DeepAI...")
     response = requests.post(
         "https://api.deepai.org/api/text2img",
         data={'text': prompt},
@@ -79,8 +86,17 @@ def generate_deepai(prompt):
         return get_image_from_url(url), "DeepAI"
     raise Exception("DeepAI Error")
 
+def generate_google_imagen(prompt):
+    response = requests.post(
+        f"https://generativelanguage.googleapis.com/v1beta/models/imagen-3.0-generate-001:predict?key={KEYS['google']}",
+        json={"instances": [{"prompt": prompt}]}
+    )
+    if response.status_code == 200:
+        image_data = base64.b64decode(response.json()['predictions'][0]['bytesBase64Encoded'])
+        return Image.open(BytesIO(image_data)), "Google Imagen"
+    raise Exception("Google Imagen Error")
+
 def generate_edenai(prompt):
-    print("Trying Eden AI...")
     response = requests.post(
         "https://api.edenai.run/v2/image/generation",
         headers={"Authorization": f"Bearer {KEYS['edenai']}"},
@@ -92,20 +108,7 @@ def generate_edenai(prompt):
         return get_image_from_url(url), "Eden AI"
     raise Exception("Eden AI Error")
 
-def generate_fireworks(prompt):
-    print("Trying Fireworks AI...")
-    response = 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": prompt}]}
-    )
-    if response.status_code == 200:
-        image_data = base64.b64decode(response.json()[0]["base64"])
-        return Image.open(BytesIO(image_data)), "Fireworks AI"
-    raise Exception("Fireworks Error")
-
 def generate_modelslab(prompt):
-    print("Trying ModelsLab...")
     response = requests.post(
         "https://modelslab.com/api/v6/images/text2img",
         json={"key": KEYS['modelslab'], "prompt": prompt, "width": "512", "height": "512"}
@@ -115,19 +118,7 @@ def generate_modelslab(prompt):
         return get_image_from_url(url), "ModelsLab"
     raise Exception("ModelsLab Error")
 
-def generate_google_imagen(prompt):
-    print("Trying Google AI (Imagen)...")
-    response = requests.post(
-        f"https://generativelanguage.googleapis.com/v1beta/models/imagen-3.0-generate-001:predict?key={KEYS['google']}",
-        json={"instances": [{"prompt": prompt}]}
-    )
-    if response.status_code == 200:
-        image_data = base64.b64decode(response.json()['predictions'][0]['bytesBase64Encoded'])
-        return Image.open(BytesIO(image_data)), "Google Imagen"
-    raise Exception("Google Imagen Error")
-
 def generate_openrouter(prompt):
-    print("Trying OpenRouter (FLUX)...")
     response = requests.post(
         "https://openrouter.ai/api/v1/chat/completions",
         headers={"Authorization": f"Bearer {KEYS['openrouter']}"},
@@ -137,16 +128,13 @@ def generate_openrouter(prompt):
         }
     )
     if response.status_code == 200:
-        # Openrouter returns an image link inside the text response for Image models
         content = response.json()['choices'][0]['message']['content']
-        # Extracting markdown image link or raw link
         if "http" in content:
             url = content.split("(")[-1].split(")")[0] if "(" in content else content.strip()
             return get_image_from_url(url), "OpenRouter"
     raise Exception("OpenRouter Error")
 
 def generate_stable_horde(prompt):
-    print("Trying Stable Horde (Async)...")
     response = requests.post(
         "https://stablehorde.net/api/v2/generate/async",
         headers={"apikey": KEYS['stablehorde']},
@@ -154,7 +142,7 @@ def generate_stable_horde(prompt):
     )
     if response.status_code == 202:
         job_id = response.json()['id']
-        for _ in range(15): # 15 সেকেন্ড অপেক্ষা করবে
+        for _ in range(15):
             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):
@@ -163,9 +151,8 @@ def generate_stable_horde(prompt):
     raise Exception("Stable Horde Error/Timeout")
 
 # ==========================================
-# 3. Fallback Logic (সিরিয়াল মেইনটেইন)
+# 3. Fallback Logic with Live Status (Yield)
 # ==========================================
-# প্রথমে সবচেয়ে ভালো API গুলো, তারপর লিমিট শেষ হলে নিচের দিকে যাবে
 API_PIPELINE = [
     generate_stability, 
     generate_getimg, 
@@ -179,39 +166,65 @@ API_PIPELINE = [
     generate_stable_horde
 ]
 
+def get_clean_name(func_name):
+    return func_name.replace("generate_", "").capitalize()
+
 def generate_image_with_fallback(prompt):
-    error_logs = []
+    if not prompt.strip():
+        yield None, "⚠️ **দয়া করে একটি প্রম্পট লিখুন!**"
+        return
+
+    live_status = "### 📡 লাইভ কানেকশন লগ:\n\n"
+    yield None, live_status  # শুরুতে স্ট্যাটাস ক্লিয়ার করবে
+
     for api_func in API_PIPELINE:
+        api_name = get_clean_name(api_func.__name__)
+        
+        # ১. কানেক্ট করার চেষ্টা করছে (UI তে দেখাবে)
+        current_status = live_status + f"⏳ **{api_name}**-এর সাথে কানেক্ট করা হচ্ছে...\n"
+        yield None, current_status
+        
         try:
+            # ২. API কল
             image, source = api_func(prompt)
             if image:
-                return image, f"✅ ছবি তৈরি হয়েছে: **{source}** এর মাধ্যমে"
+                # ৩. সফল হলে
+                final_status = live_status + f"✅ **সফল!** {source} থেকে ছবি জেনারেট হয়েছে।\n"
+                yield image, final_status
+                return # এখানেই থেমে যাবে
         except Exception as e:
-            error_logs.append(f"{api_func.__name__} failed.")
+            # ৪. ফেইল করলে বা লিমিট শেষ হলে
+            live_status += f"❌ **{api_name}:** লিমিট শেষ বা সার্ভার ডাউন।\n"
+            yield None, live_status # ফেইল মেসেজ আপডেট করে পরের লুপে যাবে
             continue
     
-    return None, f"❌ দুঃখিত! সবগুলোর লিমিট শেষ বা সার্ভার ডাউন।\nলগ: {', '.join(error_logs)}"
+    # ৫. সবগুলো ফেইল করলে
+    live_status += "\n🚫 **দুঃখিত! সবগুলো API-এর লিমিট শেষ বা কাজ করছে না।**"
+    yield None, live_status
 
 # ==========================================
 # 4. Gradio Web UI
 # ==========================================
 with gr.Blocks(theme=gr.themes.Base()) as demo:
     gr.Markdown("<h1 style='text-align: center;'>🎨 All-in-One AI Image Generator (10 APIs)</h1>")
-    gr.Markdown("<p style='text-align: center;'>একটি API এর লিমিট শেষ হলে স্বয়ংক্রিয়ভাবে অন্যটি থেকে ছবি তৈরি হবে।</p>")
+    gr.Markdown("<p style='text-align: center;'>একটি API এর লিমিট শেষ হলে স্বয়ংক্রিয়ভাবে অন্যটি থেকে ছবি তৈরি হবে এবং ���পনি লাইভ কানেকশন দেখতে পাবেন।</p>")
     
     with gr.Row():
         with gr.Column(scale=1):
-            prompt_input = gr.Textbox(label="ইমেজের বর্ণনা লিখুন (Prompt)", placeholder="A highly detailed futuristic city, cyberpunk style, 8k resolution...", lines=4)
+            prompt_input = gr.Textbox(label="ইমেজের বর্ণনা লিখুন (Prompt)", placeholder="A highly detailed futuristic city...", lines=3)
             generate_btn = gr.Button("Generate Image", variant="primary")
-            status_text = gr.Markdown("স্ট্যাটাস: অপেক্ষমান...")
+            
+            # লাইভ স্ট্যাটাস দেখানোর বক্স
+            status_box = gr.Markdown("স্ট্যাটাস: এখানে কানেকশন লগ দেখাবে...")
         
         with gr.Column(scale=1):
             image_output = gr.Image(label="জেনারেট হওয়া ছবি", type="pil")
 
+    # বাটনে ক্লিক করলে লাইভ আপডেট (yield) শুরু হবে
     generate_btn.click(
         fn=generate_image_with_fallback,
         inputs=[prompt_input],
-        outputs=[image_output, status_text]
+        outputs=[image_output, status_box]
     )
 
 if __name__ == "__main__":