WELLBOT.ipynb

{
  "cells": [
    {
      "cell_type": "markdown",
      "source": [
        "# **WELLBOT: WELL.CA'S PHARMACY CHAT BOT**"
      ],
      "metadata": {
        "id": "NR9F-NBuK1NS"
      }
    },
    {
      "cell_type": "markdown",
      "source": [
        "\n",
        "*   Includes four primary scenarios: Prescription Ordering, Prescription Management, Mediction Information Index, and Feedback/Improvement.\n",
        "\n",
        "* The user_registration function creates a new user and stores the information in a JSON file to be later pulled by the user_Login function\n",
        "* The four scenarios are connected by the main pharmacy_chatbot function that prompts the user to login or register then takes the user to the main welcome menu which calls upon the main scenario functions\n",
        "\n"
      ],
      "metadata": {
        "id": "IuTXcQ9MK6sQ"
      }
    },
    {
      "cell_type": "markdown",
      "source": [
        "**Scenario 1: Prescription Ordering**\n",
        "* connected to the prescription JSON file\n",
        "* uses random to generate RX numbers  \n",
        "---\n",
        "Includes:\n",
        "* Initiate Prescription Transfer (Incoming)\n",
        "> *functions:* generate_prescription_number\n",
        "* Submit New Prescription\n",
        "> *functions:* new_prescription, generate_random_prescription_data\n",
        "* Get Drug Price\n",
        "> *functions:* get_drug_price,\n",
        "> * uses pandas dataframe to store drug pricing info\n",
        "\n",
        "\n",
        "---\n",
        "\n",
        "\n",
        "\n",
        "\n"
      ],
      "metadata": {
        "id": "LTF8BW-ELVt-"
      }
    },
    {
      "cell_type": "markdown",
      "source": [
        "**Scenario 2: Prescription Management**\n",
        "* connected to the prescription JSON file\n",
        "---\n",
        "Includes:\n",
        "* Refill Medication\n",
        "> *functions*: refill_prescription\n",
        "* Check Medication Availability\n",
        "> *functions*: check_availability\n",
        "\n",
        " > * connects to the stock pandas dataframe through the check_availability function to check if the medication the user entered is in stock. If it is the number of units available is displayed.\n",
        "\n",
        "* Check Order Status\n",
        "> *functions*: order_status\n",
        "  \n",
        "  > * connects to the delivery_status dictionary and JSON file to show the status of the order (pending, completed etc..)\n",
        "\n",
        "\n"
      ],
      "metadata": {
        "id": "e1fpwwlTOh_m"
      }
    },
    {
      "cell_type": "markdown",
      "source": [
        "**Scenario 3: Medication Information Index**\n",
        "* connects to the OpenFDA API for medication information\n",
        "* uses the fetch_medication_info function\n",
        "* provides drug information for the drug the user inputs\n",
        "---\n",
        "Includes:\n",
        "* Dosage\n",
        "* Allergy\n",
        "* General Information/Interactions\n",
        "---\n",
        "\n"
      ],
      "metadata": {
        "id": "yNMdSI-bQnYI"
      }
    },
    {
      "cell_type": "markdown",
      "source": [
        "**Scenario 4: Feedback/Improvement**\n",
        "* connected to the improvements and ratings JSON file (stored information for further analysis_|)\n",
        "* uses VADER for sentiment analysis of reviews stored in the improvements JSON file  \n",
        "* displays the polarity (postive,negative, neutral) after analysis\n",
        "---\n",
        "Includes:\n",
        "* Need further assistance\n",
        "> * email and phone number of Well.ca pharmacy for assistance\n",
        "* Rate our service\n",
        "> * *functions:* store_ratings\n",
        "* Review our service\n",
        ">* *functions*: store_improvement, analyze_sentiment\n",
        "  \n",
        "\n"
      ],
      "metadata": {
        "id": "aPWCcLf6RcEm"
      }
    },
    {
      "cell_type": "code",
      "source": [
        "import random\n",
        "import pandas as pd\n",
        "import requests\n",
        "import json\n",
        "import nltk\n",
        "from nltk.sentiment.vader import SentimentIntensityAnalyzer\n",
        "nltk.download('vader_lexicon')\n",
        "\n",
        "title = \"WellBot: Well.ca's Pharmacy Chatbot\"\n",
        "border = \"=\" * len(title)\n",
        "print(f\"\\n{border}\\n{title}\\n{border}\\n\")\n",
        "\n",
        "# Load prescriptions from JSON file\n",
        "def load_prescriptions():\n",
        "    try:\n",
        "        with open(\"prescriptions.json\", \"r\") as f:\n",
        "            return json.load(f)\n",
        "    except (FileNotFoundError, json.JSONDecodeError):\n",
        "        return {}\n",
        "\n",
        "# Save prescriptions to JSON file\n",
        "def save_prescriptions(prescriptions):\n",
        "    with open(\"prescriptions.json\", \"w\") as f:\n",
        "        json.dump(prescriptions, f, indent=4)\n",
        "\n",
        "def user_registration():\n",
        "    username = input(\"Enter a username: \")\n",
        "    password = input(\"Enter a password: \")\n",
        "\n",
        "    # Load existing users from JSON file\n",
        "    try:\n",
        "        with open(\"users.json\", \"r\") as f:\n",
        "            users = json.load(f)\n",
        "    except (FileNotFoundError, json.JSONDecodeError):\n",
        "        users = {}\n",
        "\n",
        "    # Check if username already exists\n",
        "    if username in users:\n",
        "        print(\"Username already exists. Please choose a different username.\")\n",
        "        return False\n",
        "\n",
        "    # Store the new user's credentials\n",
        "    users[username] = {\"password\": password, \"name\": username}\n",
        "\n",
        "    # Write the updated users to the JSON file\n",
        "    with open(\"users.json\", \"w\") as f:\n",
        "        json.dump(users, f, indent=4)\n",
        "\n",
        "    print(\"Registration successful!\")\n",
        "    return True\n",
        "\n",
        "# User Login\n",
        "def user_login():\n",
        "    username = input(\"Enter your username: \")\n",
        "    password = input(\"Enter your password: \")\n",
        "\n",
        "    # Load existing users from JSON file\n",
        "    try:\n",
        "        with open(\"users.json\", \"r\") as f:\n",
        "            users = json.load(f)\n",
        "    except (FileNotFoundError, json.JSONDecodeError):\n",
        "        print(\"Error: Unable to load user data.\")\n",
        "        return False\n",
        "\n",
        "    # Check if username and password match\n",
        "    if username in users and users[username][\"password\"] == password:\n",
        "        return True\n",
        "    else:\n",
        "        print(\"Invalid username or password.\")\n",
        "        return False\n",
        "\n",
        "# Prescription Ordering\n",
        "def prescription_ordering():\n",
        "    global prescriptions, delivery_status\n",
        "    prescriptions= {}\n",
        "    def generate_prescription_number():\n",
        "        return f\"RX{random.randint(1000, 9999)}\"\n",
        "\n",
        "    def new_prescription(medication, dosage, instructions, refills, fax_needed, telephone_number=None):\n",
        "        prescription_number = generate_prescription_number()\n",
        "        prescription = {\n",
        "            'prescription_number': prescription_number,\n",
        "            'medication': medication,\n",
        "            'dosage': dosage,\n",
        "            'instructions': instructions,\n",
        "            'refills': refills,\n",
        "            'sending_pharmacy': 'N/A',\n",
        "            'fax_needed': fax_needed,\n",
        "            'telephone_number': telephone_number,\n",
        "        }\n",
        "        prescriptions[prescription_number] = prescription\n",
        "        save_prescriptions(prescriptions)\n",
        "        # Add a default delivery status for the new prescription\n",
        "        delivery_status[prescription_number] = 'Pending'\n",
        "\n",
        "    def generate_random_prescription_data():\n",
        "        medication = f\"Medication_{random.randint(1, 100)}\"\n",
        "        dosage = f\"{random.randint(1, 20)}mg\"\n",
        "        instructions = f\"Take {random.randint(1, 3)} times daily\"\n",
        "        refills = random.randint(0, 5)\n",
        "        return medication, dosage, instructions, refills\n",
        "\n",
        "    data = {\n",
        "    'Medication': ['Amoxicillin', 'Ibuprofen', 'Lisinopril', 'Metformin', 'Levothyroxine', 'Atorvastatin', 'Amlodipine', 'Omeprazole', 'Losartan', 'Aspirin'],\n",
        "    'Brand_Public': [10.0, 12.0, 15.0, 18.0, 20.0, 22.0, 24.0, 26.0, 28.0, 30.0],\n",
        "    'Brand_Private': [8.0, 10.0, 12.0, 14.0, 16.0, 18.0, 20.0, 22.0, 24.0, 26.0],\n",
        "    'Brand_Manufacturer_Rebate': [1.0, 1.5, 1.8, 2.0, 2.2, 2.4, 2.6, 2.8, 3.0, 3.2],\n",
        "    'Generic_Public': [5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0],\n",
        "    'Generic_Private': [4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0],\n",
        "    'Generic_Manufacturer_Rebate': [0.5, 0.7, 0.8, 0.9, 1.0, 1.1, 1.2, 1.3, 1.4, 1.5]\n",
        "}\n",
        "\n",
        "    df = pd.DataFrame(data)  # Provided drug pricing data\n",
        "\n",
        "    def get_drug_price(medication, category, medication_type):\n",
        "        col_name = f'{medication_type}_{category}'\n",
        "        try:\n",
        "            price = df.loc[df['Medication'].str.lower() == medication.lower(), col_name].values[0]\n",
        "            return price\n",
        "        except IndexError:\n",
        "            return None\n",
        "\n",
        "    print(\"\\nWelcome to the ordering prescriptions tab!\")\n",
        "\n",
        "    while True:\n",
        "        print(\"1. Initiate Prescription Transfer (Incoming)\")\n",
        "        print(\"2. Submit New Prescription\")\n",
        "        print(\"3. Get Drug Price\")\n",
        "        print(\"4. Exit\")\n",
        "        choice = input(\"Select an option (1/2/3): \")\n",
        "        if choice == '1':\n",
        "            print(\"\\nInitiate Prescription Transfer (Incoming)\")\n",
        "            sending_pharmacy = input(\"Enter the name of the sending pharmacy: \")\n",
        "\n",
        "            # Generate a random prescription number for the incoming transfer\n",
        "            prescription_number = generate_prescription_number()\n",
        "\n",
        "            print(f\"Generated Prescription Number: {prescription_number}\")\n",
        "\n",
        "            fax_needed = True  # Fax is always needed for prescription transfer\n",
        "            telephone_number = input(\"Enter the telephone number of the sending pharmacy: \")\n",
        "\n",
        "            # Generate random prescription data for transfer\n",
        "            medication, dosage, instructions, refills = generate_random_prescription_data()\n",
        "            new_prescription(medication, dosage, instructions, refills, fax_needed, telephone_number)\n",
        "            print(f\"Prescription {prescription_number} initiated for transfer from {sending_pharmacy}. Telephone number: {telephone_number}\")\n",
        "\n",
        "        elif choice == '2':\n",
        "            medication = input(\"Enter the medication name: \")\n",
        "            dosage = input(\"Enter the dosage: \")\n",
        "            instructions = input(\"Enter the instructions: \")\n",
        "            refills = input(\"Enter the number of refills: \")\n",
        "\n",
        "            fax_needed = input(\"Fax of prescription from doctor's office needed? (y/n): \").lower() == 'y'\n",
        "            telephone_number = None\n",
        "            if fax_needed:\n",
        "                telephone_number = input(\"Enter the telephone number of the doctor's office: \")\n",
        "\n",
        "            new_prescription(medication, dosage, instructions, refills, fax_needed, telephone_number)\n",
        "            if fax_needed:\n",
        "                print(\"Prescription submitted successfully.\")\n",
        "            else:\n",
        "                print(\"Prescription submission pending. Upload a picture of your prescription to your account.\")\n",
        "\n",
        "        elif choice == '3':\n",
        "            medication = input(\"Enter the medication name: \").lower()\n",
        "            category = input(\"Select category (Public/Private Insurance): \").title()\n",
        "            medication_type = input(\"Select medication type (Brand/Generic): \").title()\n",
        "\n",
        "            manufacturer_rebate = input(\"Do you have a manufacturer rebate? (y/n): \")\n",
        "            if manufacturer_rebate.lower() == 'y':\n",
        "                has_rebate = True\n",
        "            else:\n",
        "                has_rebate = False\n",
        "\n",
        "            price = get_drug_price(medication, category, medication_type)\n",
        "\n",
        "            if has_rebate and price is not None:\n",
        "                manufacturer_rebate = df.loc[df['Medication'].str.lower() == medication, f'{medication_type}_Manufacturer_Rebate'].values[0]\n",
        "                price -= manufacturer_rebate\n",
        "\n",
        "            if price is not None:\n",
        "                print(f\"The price of {medication_type} {medication} under {category} Insurance is ${price:.2f}\")\n",
        "            else:\n",
        "                print(\"Invalid input or medication not found.\")\n",
        "        elif choice == '4':\n",
        "            break  # Exit the loop and return to the main menu\n",
        "        else:\n",
        "            print(\"Invalid choice. Please select 1, 2, 3 or 4.\")\n",
        "\n",
        "        another_action = input(\"Do you want to perform another action? (y/n): \")\n",
        "        if another_action.lower() != \"y\":\n",
        "            print(\"\\nPrescription Database:\")\n",
        "            for prescription_number, prescription in prescriptions.items():\n",
        "                print(f\"Prescription Number: {prescription_number}\")\n",
        "                print(f\"Medication: {prescription['medication']}\")\n",
        "                print(f\"Dosage: {prescription['dosage']}\")\n",
        "                print(f\"Instructions: {prescription['instructions']}\")\n",
        "                print(f\"Refills: {prescription['refills']}\")\n",
        "                if 'sending_pharmacy' in prescription:\n",
        "                    print(f\"Sending Pharmacy: {prescription['sending_pharmacy']}\")\n",
        "                if 'fax_needed' in prescription:\n",
        "                    print(f\"Fax Needed: {'Yes' if prescription['fax_needed'] else 'No'}\")\n",
        "                if prescription.get('fax_needed') and 'telephone_number' in prescription:\n",
        "                    print(f\"Telephone Number: {prescription['telephone_number']}\")\n",
        "                print(\"-\" * 20)\n",
        "            break\n",
        "\n",
        "# Prescription Management\n",
        "def prescription_management():\n",
        "\n",
        "    # The stock dict. match a medicine with its current stock\n",
        "    stock_df = pd.DataFrame({\n",
        "    'Medication': ['Amoxicillin', 'Ibuprofen', 'Lisinopril', 'Metformin', 'Levothyroxine', 'Atorvastatin', 'Amlodipine', 'Omeprazole', 'Losartan'],\n",
        "    'Stock': [100, 0, 150, 120, 80, 90, 110, 130, 140]  # Set Ibuprofen to 0 for demonstration\n",
        "})\n",
        "\n",
        "\n",
        "    # Create a function to ask for refills of medicines (use prescription_refill dict.)\n",
        "    def refill_prescription():\n",
        "        prescriptions = load_prescriptions()\n",
        "        orders = []\n",
        "        while True:\n",
        "            prescription_number = input(\"Please enter your prescription number: \")\n",
        "            while prescription_number not in prescriptions:\n",
        "                prescription_number = input(\"Invalid prescription number. Please try again or enter 'x' to exit WellBot: \")\n",
        "                if prescription_number.lower() == 'x':\n",
        "                    print(\"Thank you for using WellBot!\")\n",
        "                    return\n",
        "                continue\n",
        "            name = input(\"Please enter your full name: \")\n",
        "            medicine = prescriptions[prescription_number]\n",
        "            order = {\n",
        "                \"prescription_number\": prescription_number,\n",
        "                \"customer\": name,\n",
        "                \"medicine\": medicine\n",
        "            }\n",
        "            orders.append(order)\n",
        "            print(f\"\\nOrder Summary:\\nPrescription Number: {prescription_number}\\nCustomer: {name}\\nMedicine: {medicine}\")\n",
        "            print(\"\\nOrder placed! We will send you an email with the delivery information.\")\n",
        "            choice = input(\"\\nWould you like to place another order? (y/n): \")\n",
        "            if choice.lower() != 'y':\n",
        "                break\n",
        "\n",
        "    # Create a function to check if certain medicine is in stock (use stock dict.)\n",
        "    def check_availability(stock_df):\n",
        "        while True:\n",
        "            medicine_name = input(\"Please enter the name of the medicine you want to check: \").capitalize()\n",
        "            if medicine_name == \"X\":\n",
        "              print(\"Exiting WellBot. Have a great day!\")\n",
        "              break\n",
        "            if medicine_name not in   stock_df['Medication'].values:\n",
        "              print(f\"Sorry, we don't have {medicine_name} in our inventory. Please try again or enter 'x' to exit WellBot:\")\n",
        "              continue\n",
        "            available_stock = stock_df.loc[stock_df['Medication'] == medicine_name, 'Stock'].values[0]\n",
        "\n",
        "            if available_stock > 0:\n",
        "                print(f\"{medicine_name} is available, we have {available_stock} units in stock.\")\n",
        "            else:\n",
        "                print(f\"Sorry, {medicine_name} is currently out of stock.\")\n",
        "                # Recommend other available drugs using pandas\n",
        "                available_meds = stock_df[stock_df['Stock'] > 0]['Medication'].tolist()\n",
        "                if available_meds:\n",
        "                  print(\"Here are some other available medications:\")\n",
        "                  for med in available_meds:\n",
        "                    print(f\"- {med}\")\n",
        "                else:\n",
        "                   print(\"Unfortunately, we're currently out of stock for all medications.\")\n",
        "            choice = input(\"\\nWould you like to check another medicine? (y/n): \")\n",
        "            if choice.lower() != 'y':\n",
        "                print(\"Thank you for using WellBot!\")\n",
        "                break\n",
        "\n",
        "     # Create a function to inform about the order status of presciption number (use delivery_status dict.)\n",
        "    def order_status():\n",
        "        global delivery_status\n",
        "        prescriptions = load_prescriptions()\n",
        "\n",
        "        while True:\n",
        "            prescription_number = input(\"Please enter your prescription number: \")\n",
        "            if prescription_number in prescriptions:\n",
        "              status = delivery_status.get(prescription_number, \"Pending Delivery\")\n",
        "              print(f\"\\nThe order status for prescription number {prescription_number} is: {status}\")\n",
        "              prescription = prescriptions[prescription_number]\n",
        "              print(f\"\\nPrescription Details:\")\n",
        "              print(f\"Medication: {prescription['medication']}\")\n",
        "              print(f\"Dosage: {prescription['dosage']}\")\n",
        "              print(f\"Instructions: {prescription['instructions']}\")\n",
        "              print(f\"Refills: {prescription['refills']}\")\n",
        "              if 'sending_pharmacy' in prescription:\n",
        "                print(f\"Sending Pharmacy: {prescription['sending_pharmacy']}\")\n",
        "              if 'fax_needed' in prescription:\n",
        "                print(f\"Fax Needed: {'Yes' if prescription['fax_needed'] else 'No'}\")\n",
        "              if prescription.get('fax_needed') and 'telephone_number' in prescription:\n",
        "                print(f\"Telephone Number: {prescription['telephone_number']}\")\n",
        "\n",
        "              if status in ['Canceled', 'On Hold', 'Rescheduled']:\n",
        "                print('For further assistance, please contact our call center: 514 123 4567')\n",
        "            else:\n",
        "              print(\"Invalid prescription number. Please try again or enter 'x' to exit WellBot.\")\n",
        "              if prescription_number.lower() == 'x':\n",
        "                print(\"Thank you for using WellBot!\")\n",
        "                return\n",
        "            choice = input(\"\\nWould you like to check another order? (y/n): \")\n",
        "            if choice.lower() != 'y':\n",
        "              break\n",
        "\n",
        "    while True:\n",
        "        print(\"\\nPlease select an option: \")\n",
        "        print(\"1. Refill Medication\")\n",
        "        print(\"2. Check Medication Availability\")\n",
        "        print(\"3. Check Order Status\")\n",
        "        print(\"4. Exit\")\n",
        "\n",
        "        option = input(\"Select the number: \")\n",
        "\n",
        "        if option == '1':\n",
        "            refill_prescription()\n",
        "        elif option == '2':\n",
        "            check_availability(stock_df)\n",
        "        elif option == '3':\n",
        "            order_status()\n",
        "        elif option == '4':\n",
        "            break\n",
        "        else:\n",
        "            print(\"\\nInvalid option. Please choose a valid option.\")\n",
        "\n",
        "# Medication Information\n",
        "def medication_information():\n",
        "    def fetch_medication_info(medication_name):\n",
        "        api_url = f\"https://api.fda.gov/drug/label.json?search=openfda.brand_name:{medication_name}&limit=1\"\n",
        "        response = requests.get(api_url)\n",
        "\n",
        "        if response.status_code == 200:\n",
        "            data = response.json()\n",
        "            if \"results\" in data and data[\"results\"]:\n",
        "                medication_info = data[\"results\"][0]\n",
        "                return medication_info\n",
        "        return None\n",
        "    while True:\n",
        "      medication_name = input(\"Enter the medication name: \")\n",
        "      medication_info = fetch_medication_info(medication_name)\n",
        "      if medication_info:\n",
        "        print(f\"Medication Name: {medication_info['openfda']['brand_name'][0]}\")\n",
        "        print(f\"Manufacturer: {medication_info['openfda']['manufacturer_name'][0]}\")\n",
        "\n",
        "        while True:\n",
        "            print(\"\\nHello and welcome, how can I help you today?\")\n",
        "            print(\"\\nSelect the type of information you'd like to know:\")\n",
        "            print(\"1. Dosage Information\")\n",
        "            print(\"2. Allergy Information\")\n",
        "            print(\"3. General Information/Interactions\")\n",
        "            print(\"4. Go back to Medication Selection\")\n",
        "            print(\"5. Exit\")\n",
        "\n",
        "            info_choice = input(\"Enter your choice: \")\n",
        "\n",
        "            if info_choice == \"1\":\n",
        "                if \"dosage_and_administration\" in medication_info:\n",
        "                    print(\"Dosage and Administration:\")\n",
        "                    for dosage in medication_info[\"dosage_and_administration\"]:\n",
        "                        print(f\"  - {dosage}\")\n",
        "                else:\n",
        "                    print(\"Dosage information not available.\")\n",
        "\n",
        "            elif info_choice == \"2\":\n",
        "                if \"warnings\" in medication_info:\n",
        "                    print(\"Allergy Information:\")\n",
        "                    for warning in medication_info[\"warnings\"]:\n",
        "                        print(f\"  - {warning}\")\n",
        "                else:\n",
        "                    print(\"Allergy information not available.\")\n",
        "\n",
        "            elif info_choice == \"3\":\n",
        "                # Display general information or interactions\n",
        "                if \"indications_and_usage\" in medication_info:\n",
        "                    print(\"General Information:\")\n",
        "                    for indication in medication_info[\"indications_and_usage\"]:\n",
        "                        print(f\"  - {indication}\")\n",
        "                else:\n",
        "                    print(\"General information not available.\")\n",
        "\n",
        "                if \"drug_interactions\" in medication_info:\n",
        "                    print(\"\\nDrug Interactions:\")\n",
        "                    for interaction in medication_info[\"drug_interactions\"]:\n",
        "                        print(f\"  - {interaction}\")\n",
        "                else:\n",
        "                    print(\"Drug interactions information not available.\")\n",
        "\n",
        "            elif info_choice == \"4\":\n",
        "                break  # Exit the medication information loop and return to main menu\n",
        "\n",
        "            elif info_choice == \"5\":\n",
        "                return False\n",
        "            else:\n",
        "                print(\"Invalid choice.\")\n",
        "\n",
        "            more_info = input(\"Would you like more info? (yes/no): \")\n",
        "      if more_info.lower() == \"no\":\n",
        "                    break\n",
        "\n",
        "    else:\n",
        "        print(\"Medication not found.\")\n",
        "        return False\n",
        "\n",
        "\n",
        "# Feedback\n",
        "def feedback_improvement():\n",
        "    # Sentiment Analysis using VADER\n",
        "    def analyze_sentiment(feedback):\n",
        "        sia = SentimentIntensityAnalyzer()\n",
        "        sentiment_scores = sia.polarity_scores(feedback)\n",
        "\n",
        "        # Categorize sentiment based on compound score\n",
        "        if sentiment_scores['compound'] >= 0.05:\n",
        "            return \"Positive\"\n",
        "        elif sentiment_scores['compound'] <= -0.05:\n",
        "            return \"Negative\"\n",
        "        else:\n",
        "            return \"Neutral\"\n",
        "\n",
        "    def store_improvement(suggestion, sentiment):\n",
        "        # Load existing improvements from JSON file\n",
        "        try:\n",
        "            with open(\"improvements.json\", \"r\") as f:\n",
        "                improvements = json.load(f)\n",
        "        except (FileNotFoundError, json.JSONDecodeError):\n",
        "            improvements = []\n",
        "\n",
        "        # Add the new suggestion and sentiment\n",
        "        feedback_data = {\n",
        "            \"suggestion\": suggestion,\n",
        "            \"sentiment\": sentiment\n",
        "        }\n",
        "        improvements.append(feedback_data)\n",
        "\n",
        "        # Write the updated improvements to the JSON file\n",
        "        with open(\"improvements.json\", \"w\") as f:\n",
        "            json.dump(improvements, f, indent=4)\n",
        "\n",
        "    def store_rating(rating):\n",
        "        # Load existing ratings from JSON file\n",
        "        try:\n",
        "            with open(\"ratings.json\", \"r\") as f:\n",
        "                ratings = json.load(f)\n",
        "        except (FileNotFoundError, json.JSONDecodeError):\n",
        "            ratings = []\n",
        "\n",
        "        # Add the new rating\n",
        "        ratings.append(rating)\n",
        "\n",
        "        # Write the updated ratings to the JSON file\n",
        "        with open(\"ratings.json\", \"w\") as f:\n",
        "            json.dump(ratings, f, indent=4)\n",
        "    print(\"Thank you for providing feedback!\")\n",
        "    while True:\n",
        "        print(\"\\nFeedback/Improvement Options:\")\n",
        "        print(\"1. Need further assistance (email/call pharmacy support)\")\n",
        "        print(\"2. Rate our service (1-10)\")\n",
        "        print(\"3. Review our service\")\n",
        "        print(\"4. Exit\")\n",
        "\n",
        "\n",
        "        feedback_choice = input(\"Enter your choice: \")\n",
        "\n",
        "        if feedback_choice == \"1\":\n",
        "            # Implement the email/call functionality here\n",
        "            print(\"You selected: Need further assistance.\")\n",
        "            print(\"For further assistance call us toll-free at 1-866-640-3800 or send an email to pharmacysupport@well.ca\")\n",
        "            print('Monday-Friday: 9am - 10pm EST')\n",
        "            print('Saturday: 9am - 4pm EST')\n",
        "            print('Sunday: 10am - 5pm EST')\n",
        "            break\n",
        "        elif feedback_choice == \"2\":\n",
        "            rating = input(\"How would you rate our service (1-10)? \")\n",
        "            store_rating(rating)\n",
        "            # You can process the rating and take actions accordingly\n",
        "            print(f\"You rated our service: {rating}\")\n",
        "            break\n",
        "\n",
        "        elif feedback_choice == \"3\":\n",
        "            improvement = input(\"How was your experience using our service? \")\n",
        "            sentiment = analyze_sentiment(improvement)\n",
        "            print(f\"Feedback sentiment: {sentiment}\")\n",
        "            # Store the improvement suggestion in a JSON file\n",
        "            store_improvement(improvement,sentiment)\n",
        "            print(\"Thank you for your input!\")\n",
        "            break\n",
        "\n",
        "        elif feedback_choice == \"4\":\n",
        "            return False\n",
        "            break  # Return to the welcome menu\n",
        "\n",
        "        else:\n",
        "            print(\"Invalid choice.\")\n",
        "\n",
        "    return True\n",
        "# Main pharmacy chatbot function\n",
        "def pharmacy_chatbot():\n",
        "    print(\"Welcome to the Pharmacy Chatbot!\")\n",
        "    while True:\n",
        "        print(\"\\n1. Login\")\n",
        "        print(\"2. Register\")\n",
        "        print(\"3. Exit\")\n",
        "        choice = input(\"Enter your choice: \")\n",
        "\n",
        "        if choice == \"1\":\n",
        "            if user_login():\n",
        "                username = input(\"Enter your username again for verification: \")\n",
        "                break\n",
        "            else:\n",
        "                continue\n",
        "        elif choice == \"2\":\n",
        "            if user_registration():\n",
        "                username = input(\"Enter your username again for verification: \")\n",
        "                break\n",
        "            else:\n",
        "                continue\n",
        "        elif choice == \"3\":\n",
        "            print(\"Thank you for using the pharmacy chatbot. Goodbye!\")\n",
        "            exit()\n",
        "        else:\n",
        "            print(\"Invalid choice. Please choose a valid option.\")\n",
        "\n",
        "    print(f\"Welcome, {username}!\")\n",
        "    while True:\n",
        "        print(\"\\nWelcome menu options:\")\n",
        "        print(\"1. Medication Order\")\n",
        "        print(\"2. Prescription Management\")\n",
        "        print(\"3. Medication Information Index\")\n",
        "        print(\"4. Feedback/Improvement\")\n",
        "        print(\"5. Exit\")\n",
        "\n",
        "        choice = input(\"Enter your choice: \")\n",
        "\n",
        "        if choice == \"1\":\n",
        "            prescription_ordering()\n",
        "        elif choice == \"2\":\n",
        "            prescription_management()\n",
        "        elif choice == \"3\":\n",
        "            medication_information()\n",
        "        elif choice == \"4\":\n",
        "            feedback_improvement()\n",
        "        elif choice == \"5\":\n",
        "            print(\"Thank you for using the pharmacy chatbot. Goodbye!\")\n",
        "            exit()\n",
        "            break\n",
        "        else:\n",
        "            print(\"Invalid choice. Please choose a valid option.\")\n",
        "\n",
        "# Call the main function\n",
        "pharmacy_chatbot()"
      ],
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "id": "KL1btuBtnk-w",
        "outputId": "75153787-5e53-4c91-fe90-7ed2ed80c3c2"
      },
      "execution_count": null,
      "outputs": [
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "[nltk_data] Downloading package vader_lexicon to /root/nltk_data...\n",
            "[nltk_data]   Package vader_lexicon is already up-to-date!\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "\n",
            "===================================\n",
            "WellBot: Well.ca's Pharmacy Chatbot\n",
            "===================================\n",
            "\n",
            "Welcome to the Pharmacy Chatbot!\n",
            "\n",
            "1. Login\n",
            "2. Register\n",
            "3. Exit\n",
            "Enter your choice: 1\n",
            "Enter your username: mshamas\n",
            "Enter your password: xxxxxx\n",
            "Enter your username again for verification: mshamas\n",
            "Welcome, mshamas!\n",
            "\n",
            "Welcome menu options:\n",
            "1. Medication Order\n",
            "2. Prescription Management\n",
            "3. Medication Information Index\n",
            "4. Feedback/Improvement\n",
            "5. Exit\n",
            "Enter your choice: 1\n",
            "\n",
            "Welcome to the ordering prescriptions tab!\n",
            "1. Initiate Prescription Transfer (Incoming)\n",
            "2. Submit New Prescription\n",
            "3. Get Drug Price\n",
            "4. Exit to Main Menu\n",
            "Select an option (1/2/3): 4\n",
            "Returning to the main menu.\n",
            "\n",
            "Welcome menu options:\n",
            "1. Medication Order\n",
            "2. Prescription Management\n",
            "3. Medication Information Index\n",
            "4. Feedback/Improvement\n",
            "5. Exit\n",
            "Enter your choice: 2\n",
            "\n",
            "Please select an option: \n",
            "1. Refill Medication\n",
            "2. Check Medication Availability\n",
            "3. Check Order Status\n",
            "4. Exit\n",
            "Select the number: 4\n",
            "\n",
            "Welcome menu options:\n",
            "1. Medication Order\n",
            "2. Prescription Management\n",
            "3. Medication Information Index\n",
            "4. Feedback/Improvement\n",
            "5. Exit\n",
            "Enter your choice: 3\n",
            "Enter the medication name: advil\n",
            "Medication Name: Advil Dual Action with Acetaminophen\n",
            "Manufacturer: Lil' Drug Store Products, Inc.\n",
            "\n",
            "Hello and welcome, how can I help you today?\n",
            "\n",
            "Select the type of information you'd like to know:\n",
            "1. Dosage Information\n",
            "2. Allergy Information\n",
            "3. General Information/Interactions\n",
            "4. Go back to Medication Selection\n",
            "5. Exit\n",
            "Enter your choice: 5\n",
            "\n",
            "Welcome menu options:\n",
            "1. Medication Order\n",
            "2. Prescription Management\n",
            "3. Medication Information Index\n",
            "4. Feedback/Improvement\n",
            "5. Exit\n",
            "Enter your choice: 5\n",
            "Thank you for using the pharmacy chatbot. Goodbye!\n"
          ]
        }
      ]
    }
  ],
  "metadata": {
    "colab": {
      "provenance": []
    },
    "kernelspec": {
      "display_name": "Python 3",
      "name": "python3"
    },
    "language_info": {
      "name": "python"
    }
  },
  "nbformat": 4,
  "nbformat_minor": 0
}