Update app.py

app.py

@@ -20,7 +20,7 @@ KEYS = {
     "stablehorde": "56kkbwcSPfHIr2YsDkpNDQ",
     "modelslab": "42zL3I6pH7Mv94jhNf4RGyuKGGT3GoEhfXZv47WwmQsfyhgC8XNttfOpvMKL",
     "openrouter": "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",
-    "omniinfer": "sk_ffTeDYbT2PibvuC27eGFbeE4E-7wcCuWp4SVgFDFWHDU",
+    "omniinfer": "sk_fTeDYbT2PibvuC27eGFbeE4E-7wcCuWp4SVgFDFWHDU",
     "fireworks": "key_kH0svuL0"
 }
 
@@ -142,7 +142,7 @@ def generate_stable_horde(prompt):
     raise Exception("Error")
 
 # ==========================================
-# 3. Pipeline & Status Logic
+# 3. Pipeline & Status Logic (With Prompt Enhancer)
 # ==========================================
 API_PIPELINE = [
     ("Stability AI", generate_stability), 
@@ -164,29 +164,40 @@ def render_dashboard(statuses):
     html_content += "</div>"
     return html_content
 
-def generate_image_with_fallback(prompt, current_statuses):
+def generate_image_with_fallback(prompt, style, current_statuses):
     if not prompt.strip():
         yield None, "⚠️ **দয়া করে একটি প্রম্পট লিখুন!**", render_dashboard(current_statuses), current_statuses
         return
 
-    # জেনারেশন শুরুর আগে সব স্ট্যাটাস 'অপেক্ষমান' (সাদা বাতি) করে দেওয়া
+    # ম্যাজিক কিওয়ার্ড (যাতে ছবিগুলো হাই কোয়ালিটি এবং রিয়ালিস্টিক হয়)
+    style_modifiers = {
+        "ফটোরিয়ালিস্টিক (Photorealistic)": ", ultra realistic, highly detailed, photorealistic, 8k resolution, raw photo, masterpiece, soft cinematic lighting, stunning photography",
+        "এনিমে (Anime)": ", high quality anime style, studio ghibli, masterpiece, vibrant colors, detailed shading, gorgeous anime art",
+        "থ্রিডি আর্ট (3D)": ", 3d render, unreal engine 5, octane render, beautiful 3d masterpiece, highly detailed",
+        "ডিফল্ট (Default)": ""
+    }
+    
+    # ইউজার যা লিখবে, তার সাথে ম্যাজিক কিওয়ার্ড যুক্ত হয়ে যাবে
+    enhanced_prompt = prompt.strip() + style_modifiers.get(style, "")
+
     for name, _ in API_PIPELINE:
         current_statuses[name] = "⚪"
     
     yield None, "⏳ শুরু হচ্ছে...", render_dashboard(current_statuses), current_statuses
 
     for api_name, api_func in API_PIPELINE:
-        current_statuses[api_name] = "🟡" # হলুদ বাতি: কানেক্ট করার চেষ্টা করছে
+        current_statuses[api_name] = "🟡"
         yield None, f"⏳ **{api_name}**-এর সাথে কানেক্ট করা হচ্ছে...", render_dashboard(current_statuses), current_statuses
         
         try:
-            image = api_func(prompt)
+            # উন্নত করা প্রম্পটটি API তে পাঠানো হচ্ছে
+            image = api_func(enhanced_prompt)
             if image:
-                current_statuses[api_name] = "🟢" # সবুজ বাতি: ১০০% কাজ করেছে
+                current_statuses[api_name] = "🟢"
                 yield image, f"✅ **সফল!** {api_name} থেকে ছবি তৈরি হয়েছে।", render_dashboard(current_statuses), current_statuses
-                return # সফল হলে এখানেই থেমে যাবে
+                return 
         except Exception as e:
-            current_statuses[api_name] = "🔴" # লাল বাতি: লিমিট শেষ বা সার্ভার ডাউন
+            current_statuses[api_name] = "🔴" 
             yield None, f"❌ **{api_name}** ফেইল করেছে। পরেরটিতে যাচ্ছে...", render_dashboard(current_statuses), current_statuses
             continue
     
@@ -195,23 +206,26 @@ def generate_image_with_fallback(prompt, current_statuses):
 # ==========================================
 # 4. Gradio Web UI
 # ==========================================
-# শুরুতে সবগুলোতে সাদা বাতি (⚪) থাকবে
 initial_statuses = {name: "⚪" for name, _ in API_PIPELINE}
 
 with gr.Blocks(theme=gr.themes.Base()) as demo:
-    gr.Markdown("<h1 style='text-align: center;'>🎨 Pro AI Image Generator (Live Status)</h1>")
+    gr.Markdown("<h1 style='text-align: center;'>🎨 Pro AI Image Generator (Live Status & HD Quality)</h1>")
     
-    # টপ ড্যাশবোর্ড (লাইট প্যানেল)
     status_board = gr.HTML(render_dashboard(initial_statuses))
-    
-    # স্ট্যাটাস মনে রাখার জন্য State ভেরিয়েবল
     status_state = gr.State(initial_statuses)
-    
     gr.Markdown("<p style='text-align: center;'>⚪ অপেক্ষমান &nbsp; | &nbsp; ���� কানেক্টিং &nbsp; | &nbsp; 🟢 কানেক্টেড (সফল) &nbsp; | &nbsp; 🔴 লিমিট শেষ/ডাউন</p>")
 
     with gr.Row():
         with gr.Column(scale=1):
-            prompt_input = gr.Textbox(label="ইমেজের বর্ণনা লিখুন (Prompt)", placeholder="A highly detailed futuristic city...", lines=4)
+            prompt_input = gr.Textbox(label="ইমেজের বর্ণনা লিখুন (Prompt)", placeholder="Two beautiful red roses in a garden...", lines=3)
+            
+            # নতুন স্টাইল অপশন যুক্ত করা হয়েছে
+            style_input = gr.Radio(
+                choices=["ফটোরিয়ালিস্টিক (Photorealistic)", "এনিমে (Anime)", "থ্রিডি আর্ট (3D)", "ডিফল্ট (Default)"],
+                label="ছবি কেমন চাচ্ছেন? (Style)",
+                value="ফটোরিয়ালিস্টিক (Photorealistic)"
+            )
+            
             generate_btn = gr.Button("Generate Image", variant="primary")
             log_box = gr.Markdown("স্ট্যাটাস: অপেক্ষমান...")
         
@@ -220,7 +234,7 @@ with gr.Blocks(theme=gr.themes.Base()) as demo:
 
     generate_btn.click(
         fn=generate_image_with_fallback,
-        inputs=[prompt_input, status_state],
+        inputs=[prompt_input, style_input, status_state],
         outputs=[image_output, log_box, status_board, status_state]
     )