Update app.py

app.py

@@ -2,7 +2,7 @@ import pickle
 from sklearn.feature_extraction.text import CountVectorizer
 import gradio as gr
 
-# Load the Naive Bayes model and vectorizer
+# Load Naive Bayes model and vectorizer
 with open("naive_bayes_model.pkl", "rb") as model_file:
     clf = pickle.load(model_file)
 
@@ -14,23 +14,7 @@ responses = {
     "activate_my_card": "To activate your card, log in to the app and navigate to the 'Activate Card' section.",
     "age_limit": "The minimum age to use this service is 18 years.",
     "apple_pay_or_google_pay": "Yes, you can use both Apple Pay and Google Pay with your card.",
-    "atm_support": "Your card is supported by ATMs that display the Visa or Mastercard logo.",
-    "automatic_top_up": "You can enable automatic top-up in the app under the 'Top-Up Settings' section.",
-    "balance_not_updated_after_bank_transfer": "If your balance hasn’t updated after a bank transfer, please wait for 24 hours. If the issue persists, contact support.",
-    "beneficiary_not_allowed": "Ensure that the beneficiary details are correct. Some accounts may have restrictions; check with customer support for clarification.",
-    "cancel_transfer": "To cancel a transfer, go to the 'Transaction History' section in the app and select the transfer you wish to cancel.",
-    "card_about_to_expire": "If your card is about to expire, a replacement will be sent automatically. Contact support if you haven’t received it.",
-    "card_arrival": "New cards usually arrive within 7-10 business days after being issued.",
-    "card_not_working": "If your card isn't working, ensure it is activated and has sufficient balance. Contact support if the issue persists.",
-    "change_pin": "You can change your PIN using the app or at any ATM with your card.",
-    "contactless_not_working": "Ensure your card supports contactless payments and check if the terminal accepts it. If the issue persists, contact support.",
-    "country_support": "Your card is supported in all countries where Visa/Mastercard is accepted. Check the app for restrictions.",
-    "declined_card_payment": "Card payments can be declined due to insufficient balance, incorrect PIN, or restrictions on the merchant. Check your app for details.",
-    "lost_or_stolen_card": "If your card is lost or stolen, block it immediately in the app under the 'Card Management' section and request a replacement.",
-    "pending_card_payment": "Pending payments are usually resolved within 2-3 business days. Contact support if the status doesn't update.",
-    "refund_not_showing_up": "Refunds can take up to 7 business days to appear. Check with the merchant or contact support if it takes longer.",
-    "top_up_failed": "Top-up failures may occur due to incorrect details or insufficient balance in the source account. Check your app for details.",
-    "transfer_not_received_by_recipient": "If the recipient hasn't received the transfer, ensure the details are correct. Contact support for further assistance.",
+    # Add remaining responses here...
 }
 
 # Define chatbot response logic
@@ -39,15 +23,35 @@ def chatbot_response(user_input):
     predicted_label = clf.predict(vectorized_input)[0]
     return responses.get(predicted_label, "Sorry, I couldn't understand your question.")
 
-# Define Gradio interface
+# Define UI description with sample questions
+description = """
+### FinTech Chatbot
+
+This chatbot can assist you with a variety of fintech-related queries. Below are some sample questions you can ask:
+
+- How do I activate my card?
+- What is the minimum age required to use this service?
+- Can I use Apple Pay with my card?
+- Which ATMs can I use my card with?
+- How can I set up automatic top-up for my account?
+- Why hasn't my balance updated after making a bank transfer?
+- Why can’t I add a beneficiary to my account?
+- Can I cancel a transfer I’ve made?
+- What should I do if my card is about to expire?
+- How long does it take for a new card to arrive?
+- My card has been stolen. What should I do?
+- Why hasn’t the recipient received my transfer?
+"""
+
+# Define the Gradio interface
 interface = gr.Interface(
     fn=chatbot_response,
     inputs=gr.Textbox(lines=2, placeholder="Ask your fintech question here..."),
     outputs="text",
-    title="Banking Chatbot",
-    description="A chatbot to handle banking-related queries using a Naive Bayes model."
+    title="FinTech Chatbot",
+    description=description
 )
 
 if __name__ == "__main__":
     # Launch the Gradio app
-    interface.launch()
+    interface.launch()