Update app.py

app.py

@@ -1,23 +1,13 @@
 import os
 
-# ১. requirements.txt আপডেট করা (Inference Client এর জন্য)
+# ১. requirements.txt আপডেট করা (Inference API এর জন্য)
 with open("requirements.txt", "w") as f:
     f.write("gradio\nhuggingface_hub\n")
 
-# ২. app.py আপডেট করা (Public Access Version)
+# ২. app.py আপডেট করা (InferenceClient ব্যবহার করে)
 app_code = """
 import gradio as gr
 from huggingface_hub import InferenceClient
-import os
-
-# Space Settings-এর Secrets থেকে টোকেনটি নেবে
-hf_token = os.getenv('HF_TOKEN')
-
-# মডেল আইডি
-model_id = 'mx-llms/BLM'
-
-# Client ইনিশিয়ালাইজ করা (এটি আপনার টোকেন ব্যবহার করে কাজ করবে)
-client = InferenceClient(token=hf_token, model=model_id)
 
 def respond(
     message,
@@ -26,10 +16,15 @@ def respond(
     max_tokens,
     temperature,
     top_p,
+    hf_token: gr.OAuthToken,
 ):
+    # Initialize the Inference Client with the provided OAuth token
+    # Model: mx-llms/BLM
+    client = InferenceClient(token=hf_token.token, model='mx-llms/BLM')
+
     messages = [{"role": "system", "content": system_message}]
 
-    # চ্যাট হিস্ট্রি যোগ করা
+    # Add history to messages
     for h in history:
         messages.append(h)
 
@@ -37,7 +32,7 @@ def respond(
 
     response = ""
 
-    # ইনফারেন্স কল (স্ট্রিমিং সহ)
+    # Request streaming completion
     for message in client.chat_completion(
         messages,
         max_tokens=max_tokens,
@@ -50,8 +45,8 @@ def respond(
             response += token
             yield response
 
-# Gradio চ্যাট ইন্টারফেস
-demo = gr.ChatInterface(
+# Gradio Interface
+chatbot = gr.ChatInterface(
     respond,
     additional_inputs=[
         gr.Textbox(value="You are BLM. Created by MD Mushfiqur Rahim.", label="System message"),
@@ -59,11 +54,15 @@ demo = gr.ChatInterface(
         gr.Slider(minimum=0.1, maximum=2.0, value=0.1, step=0.1, label="Temperature"),
         gr.Slider(minimum=0.1, maximum=1.0, value=0.9, step=0.05, label="Top-p"),
     ],
-    type="messages",
-    title="BLM AI Assistant",
-    description="Created by MD Mushfiqur Rahim. This assistant is ready to help everyone!"
+    type="messages"
 )
 
+with gr.Blocks() as demo:
+    with gr.Sidebar():
+        gr.Markdown("### BLM Assistant Login")
+        gr.LoginButton()
+    chatbot.render()
+
 if __name__ == '__main__':
     demo.launch()
 """
@@ -71,4 +70,4 @@ if __name__ == '__main__':
 with open("app.py", "w") as f:
     f.write(app_code)
 
-print("✅ Public access app.py created successfully!")
+print("✅ app.py has been updated to use Inference API and OAuth!")