Update app.py

app.py

@@ -3,9 +3,9 @@ from huggingface_hub import InferenceClient
 import os
 
 # ============================================================
-# ১. মডেল আইডি (সিক্রেট থেকে অথবা ডিফল্ট)
+# ১. মডেল আইডি (সিক্রেট বা ডিফল্ট)
 # ============================================================
-model_id = os.getenv("MODEL_ID", "mx-llms/BLM")  # আপনার মডেল আইডি
+model_id = os.getenv("MODEL_ID", "mx-llms/BLM")
 
 def respond(
     message,
@@ -14,28 +14,42 @@ def respond(
     max_tokens,
     temperature,
     top_p,
-    hf_token: gr.OAuthToken,
+    hf_token: gr.OAuthToken | None,
 ):
     """
-    Hugging Face Inference API ব্যবহার করে স্ট্রিমিং রেসপন্স।
+    Hugging Face Inference API - টোকেন অটো-ডিটেক্ট:
+    - ইউজার লগইন করলে OAuth Token ব্যবহার করবে
+    - না করলে Space Secret-এর HF_TOKEN ব্যবহার করবে
     """
-    client = InferenceClient(token=hf_token.token, model=model_id)
+    # টোকেন নির্বাচন
+    if hf_token is not None:
+        token = hf_token.token
+    else:
+        token = os.getenv("HF_TOKEN")
+        if token is None:
+            yield "⚠️ Please login or set HF_TOKEN in Space Secrets."
+            return
+
+    client = InferenceClient(token=token, model=model_id)
 
     messages = [{"role": "system", "content": system_message}]
     messages.extend(history)
     messages.append({"role": "user", "content": message})
 
     response = ""
-    for message in client.chat_completion(
-        messages,
-        max_tokens=max_tokens,
-        stream=True,
-        temperature=temperature,
-        top_p=top_p,
-    ):
-        token = message.choices[0].delta.content or ""
-        response += token
-        yield response
+    try:
+        for msg in client.chat_completion(
+            messages,
+            max_tokens=max_tokens,
+            stream=True,
+            temperature=temperature,
+            top_p=top_p,
+        ):
+            token_text = msg.choices[0].delta.content or ""
+            response += token_text
+            yield response
+    except Exception as e:
+        yield f"❌ Error: {str(e)}. Please check logs."
 
 # ============================================================
 # ২. ChatInterface UI
@@ -52,7 +66,7 @@ chatbot = gr.ChatInterface(
         gr.Slider(minimum=0.1, maximum=1.0, value=0.95, step=0.05, label="Top-p"),
     ],
     title="BLM AI Assistant",
-    description="Created by MD Mushfiqur Rahim. Powered by Hugging Face Inference API.",
+    description="Created by MD Mushfiqur Rahim. Please login (top-left) for full access.",
 )
 
 # ============================================================