Update app.py

app.py

@@ -1,22 +1,29 @@
 import gradio as gr
+from transformers import pipeline
 
-# Define the chatbot function
+# Load a Hugging Face model for chat (choose an appropriate model)
+chatbot_pipeline = pipeline("text-generation", model="mistralai/Mistral-7B-Instruct", max_new_tokens=100)
+
+# Chatbot function using HF's transformer model
 def chat_response(history, user_input):
-    """Handles chatbot interactions."""
+    """Handles chatbot interactions using a Hugging Face model."""
     if not user_input.strip():
-        return history + [("User", user_input), ("Bot", "Please enter a valid message.")]
+        return history + [{"role": "user", "content": user_input}, {"role": "assistant", "content": "Please enter a valid message."}], ""
+
+    # Generate AI response
+    bot_reply = chatbot_pipeline(user_input)[0]["generated_text"]
 
-    response = f"You said: {user_input}"  # Replace this with actual chatbot logic (e.g., OpenAI API)
+    # Append messages to history
     history.append({"role": "user", "content": user_input})
-    history.append({"role": "assistant", "content": response})
+    history.append({"role": "assistant", "content": bot_reply})
 
     return history, ""
 
-# Initialize Gradio chatbot with the correct type
+# Initialize Gradio chatbot
 with gr.Blocks() as demo:
-    gr.Markdown("## 🚀 AI Bandar -")
+    gr.Markdown("## 🤖 AI Bandar - Powered by Hugging Face")
 
-    chatbot = gr.Chatbot(type="messages")  # ✅ Fix: Use 'messages' format
+    chatbot = gr.Chatbot(type="messages")  # Correct type
     user_input = gr.Textbox(label="Your Message")
     submit_button = gr.Button("Send")
 
@@ -24,6 +31,6 @@ with gr.Blocks() as demo:
     submit_button.click(chat_response, inputs=[chatbot, user_input], outputs=[chatbot, user_input])
     user_input.submit(chat_response, inputs=[chatbot, user_input], outputs=[chatbot, user_input])
 
-# Run the Gradio app
+# Run on Hugging Face Spaces
 if __name__ == "__main__":
-    demo.launch(share=True)  # 🚀 Allows sharing via public Gradio link
+    demo.launch()