Update app.py

app.py

@@ -1,8 +1,13 @@
 import gradio as gr
-import gradio as gr
 import replicate
 import os
-import time
+
+# -------- সমস্যা সমাধানের কোড (FIX) --------
+# আপনার টোকেন থেকে অদৃশ্য স্পেস বা এন্টার (\n) মুছে ফেলা হচ্ছে
+if "REPLICATE_API_TOKEN" in os.environ:
+    raw_token = os.environ["REPLICATE_API_TOKEN"]
+    os.environ["REPLICATE_API_TOKEN"] = raw_token.strip()
+# ----------------------------------------
 
 # আপনার মডেল সেটআপ
 MODEL_ID = "lucataco/flux-dev-lora"
@@ -10,46 +15,45 @@ LORA_URL = "https://huggingface.co/SahinaKhatun123/SahinaThumbnails/resolve/main
 TRIGGER_WORD = "SahinaStyle" 
 
 def generate_image(prompt, aspect_ratio):
-    print("🚀 Starting generation process...") # Logs-এ এটা খুঁজবেন
-    full_prompt = f"{prompt}, {TRIGGER_WORD}, youtube thumbnail, 4k quality, highly detailed"
+    # ট্রিগার ওয়ার্ড যুক্ত করা
+    full_prompt = f"{prompt}, {TRIGGER_WORD}, youtube thumbnail, 4k quality, highly detailed, vibrant colors"
+    
+    print(f"🚀 Processing: {full_prompt}")
     
     try:
-        print(f"📡 Calling Replicate with prompt: {full_prompt}")
-        
-        # Replicate কল করা হচ্ছে
         output = replicate.run(
             MODEL_ID,
             input={
                 "hf_lora": LORA_URL,
-                "lora_scale": 2.0,
+                "lora_scale": 1.2, # থাম্বনেইলের জন্য ১.২ পারফেক্ট ব্যালেন্স
                 "prompt": full_prompt,
                 "aspect_ratio": aspect_ratio,
                 "num_outputs": 1,
                 "output_format": "webp"
             }
         )
-        
-        print(f"✅ Replicate Response: {output}") # রেজাল্ট আসলে এটা দেখাবে
-        return output[0], "Success! Image generated."
+        return output[0], "✅ Success! Thumbnail Generated."
         
     except Exception as e:
-        print(f"❌ ERROR HAPPENED: {str(e)}") # এরর হলে এটা দেখাবে
-        return None, f"Error: {str(e)}"
+        error_msg = str(e)
+        if "nsfw" in error_msg.lower():
+            return None, "❌ Error: NSFW content detected. Try a different prompt."
+        return None, f"❌ Error: {error_msg}"
 
-# UI ডিজাইন
+# অ্যাপের ডিজাইন
 with gr.Blocks(theme=gr.themes.Soft()) as app:
-    gr.Markdown("# 🎬 Debug Mode: Thumbnail Generator")
+    gr.Markdown("# 🎨 Sahina's Thumbnail Generator")
     
     with gr.Row():
         with gr.Column():
-            text_input = gr.Textbox(label="Video Title")
-            ratio = gr.Dropdown(["16:9", "1:1"], value="16:9", label="Ratio")
-            gen_btn = gr.Button("Generate", variant="primary")
+            text_input = gr.Textbox(label="Video Title / Topic", placeholder="Ex: A shocked man looking at a flying car...")
+            ratio = gr.Dropdown(["16:9", "1:1", "9:16"], value="16:9", label="Size")
+            gen_btn = gr.Button("Generate Thumbnail ✨", variant="primary")
         
         with gr.Column():
-            image_output = gr.Image(label="Result")
-            status_box = gr.Textbox(label="Status Log")
+            image_output = gr.Image(label="Generated Thumbnail")
+            status_output = gr.Textbox(label="Status", interactive=False)
 
-    gen_btn.click(fn=generate_image, inputs=[text_input, ratio], outputs=[image_output, status_box])
+    gen_btn.click(fn=generate_image, inputs=[text_input, ratio], outputs=[image_output, status_output])
 
 app.launch()