import pickle
from sklearn.feature_extraction.text import CountVectorizer
import gradio as gr

# Load Naive Bayes model and vectorizer
with open("naive_bayes_model.pkl", "rb") as model_file:
    clf = pickle.load(model_file)

with open("vectorizer.pkl", "rb") as vectorizer_file:
    vectorizer = pickle.load(vectorizer_file)

# Define the responses dictionary
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.",
}

# Define chatbot response logic
def chatbot_response(user_input):
    vectorized_input = vectorizer.transform([user_input])
    predicted_label = clf.predict(vectorized_input)[0]
    return responses.get(predicted_label, "Sorry, I couldn't understand your question.")

# Define UI description with a table of sample questions
description = """
### Banking Chatbot

This chatbot can assist you with a variety of banking-related queries. Below are some sample questions you can ask:

<table>
    <tr>
        <th>Category</th>
        <th>Sample Questions</th>
    </tr>
    <tr>
        <td>Activate My Card</td>
        <td>How do I activate my card? <br> Where can I find the option to activate my card?</td>
    </tr>
    <tr>
        <td>Age Limit</td>
        <td>What is the minimum age required to use this service? <br> Can a 17-year-old open an account?</td>
    </tr>
    <tr>
        <td>Apple Pay or Google Pay</td>
        <td>Can I use Apple Pay with my card? <br> Does my card support Google Pay?</td>
    </tr>
    <tr>
        <td>ATM Support</td>
        <td>Which ATMs can I use my card with? <br> Are there fees for ATM withdrawals?</td>
    </tr>
    <tr>
        <td>Automatic Top-Up</td>
        <td>How can I set up automatic top-up for my account? <br> Does your app allow automatic balance top-ups?</td>
    </tr>
    <tr>
        <td>Balance Not Updated</td>
        <td>Why hasn't my balance updated after making a bank transfer? <br> How long does it take for updates to reflect?</td>
    </tr>
    <tr>
        <td>Cancel Transfer</td>
        <td>Can I cancel a transfer I’ve made? <br> What happens if I cancel a transaction?</td>
    </tr>
    <tr>
        <td>Card Issues</td>
        <td>Why isn’t my card working? <br> How do I fix issues with a non-functioning card?</td>
    </tr>
</table>
"""

# 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=description
)

if __name__ == "__main__":
    # Launch the Gradio app
    interface.launch()