NER1.ipynb

{
  "nbformat": 4,
  "nbformat_minor": 0,
  "metadata": {
    "colab": {
      "provenance": [],
      "include_colab_link": true
    },
    "kernelspec": {
      "name": "python3",
      "display_name": "Python 3"
    },
    "language_info": {
      "name": "python"
    }
  },
  "cells": [
    {
      "cell_type": "markdown",
      "metadata": {
        "id": "view-in-github",
        "colab_type": "text"
      },
      "source": [
        "<a href=\"https://colab.research.google.com/github/basedsaksham/Improving-NER/blob/main/NER1.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": null,
      "metadata": {
        "id": "6qNzUpRqjnrS"
      },
      "outputs": [],
      "source": [
        "import re\n",
        "import spacy\n",
        "from datetime import datetime, timedelta\n",
        "\n",
        "nlp = spacy.load(\"en_core_web_sm\")\n",
        "\n",
        "def detect_email(sentence):\n",
        "    email_regex_pattern = '[A-Za-z0-9._%+-]*@[A-Za-z0-9.-]*\\.[A-Z|a-z]*'\n",
        "    emails_matches = re.findall(email_regex_pattern, sentence)\n",
        "    return emails_matches\n",
        "def detect_time(sentence):\n",
        "    time_regex = r'\\b(?:1[0-2]|0?[1-9])(?::[0-5][0-9])?(?:\\s?[ap]m)?\\b'\n",
        "    times = re.findall(time_regex, sentence, re.IGNORECASE)\n",
        "    return times\n",
        "def detect_phone_numbers(sentence):\n",
        "    phone_regex = r'(\\(?\\d{3}\\)?[\\s.-]?\\d{3}[\\s.-]?\\d{4})'\n",
        "    phone_numbers = re.findall(phone_regex, sentence)\n",
        "    return phone_numbers\n",
        "def detect_numbers_with_units(text):\n",
        "    #pattern = pattern = '(\\d+(?:\\.\\d+)?)(?:\\s+)(\\w+)'\n",
        "    #matches = re.findall(pattern, text)\n",
        "    #return matches\n",
        "    # Regex pattern for detecting phone numbers\n",
        "    phone_pattern = r'(\\(?\\d{3}\\)?[\\s.-]?\\d{3}[\\s.-]?\\d{4})'\n",
        "\n",
        "    # Find all phone numbers in the text\n",
        "    phone_numbers = re.findall(phone_pattern, text)\n",
        "\n",
        "    # Regex pattern for detecting numbers with units\n",
        "    pattern = r'(?<!\\d)(\\d+(?:\\.\\d+)?)(?:\\s+)(\\w+)(?!\\d)'\n",
        "\n",
        "    # Find all matches of numbers with units in the text\n",
        "    matches = re.findall(pattern, text)\n",
        "\n",
        "    # Filter out numbers that are not phone numbers\n",
        "    numbers_with_units = []\n",
        "    for match in matches:\n",
        "        number, unit = match\n",
        "        if number not in phone_numbers:\n",
        "            numbers_with_units.append((number, unit))\n",
        "\n",
        "    return numbers_with_units\n",
        "def detect_dates(text):\n",
        "    # Current date\n",
        "    today = datetime.now()\n",
        "\n",
        "    # Define regex patterns for relative date expressions\n",
        "    patterns = [\n",
        "        r\"(next|agle)\\s+(monday|tuesday|wednesday|thursday|friday|saturday|sunday|somvar|mangalwar|budhwar|guruwar|shukrawar|shaniwar|raviwar)\",\n",
        "        r\"(kal)\",\n",
        "        r\"(next|agle)\\s+(week|month|year|hafte|mahine|saal)\"\n",
        "    ]\n",
        "\n",
        "    # Initialize empty list to store detected dates\n",
        "    detected_dates = []\n",
        "\n",
        "    # Iterate through patterns and search for matches in text\n",
        "    for pattern in patterns:\n",
        "        matches = re.findall(pattern, text.lower())\n",
        "        for match in matches:\n",
        "            if match[0] in ['next', 'agle']:\n",
        "                if match[1] in ['monday', 'somvar']:\n",
        "                    # Find next Monday\n",
        "                    days_until_weekday = (today.weekday() - 1) % 7\n",
        "                    next_date = today + timedelta(days=days_until_weekday )\n",
        "                    detected_dates.append(next_date.strftime(\"%d-%m-%Y\"))\n",
        "                elif match[1] in ['tuesday', 'mangalwar']:\n",
        "                    # Find next Tuesday\n",
        "                    days_until_weekday = (today.weekday() - 0) % 7\n",
        "                    next_date = today + timedelta(days=days_until_weekday )\n",
        "                    detected_dates.append(next_date.strftime(\"%d-%m-%Y\"))\n",
        "                elif match[1] in ['wednesday', 'budhwar']:\n",
        "                    # Find next Wednesday\n",
        "                    days_until_weekday = (today.weekday() +1) % 7\n",
        "                    next_date = today + timedelta(days=days_until_weekday )\n",
        "                    detected_dates.append(next_date.strftime(\"%d-%m-%Y\"))\n",
        "                elif match[1] in ['thursday', 'guruwar']:\n",
        "                    # Find next Thursday\n",
        "                    days_until_weekday = (today.weekday() +2) % 7\n",
        "                    next_date = today + timedelta(days=days_until_weekday )\n",
        "                    detected_dates.append(next_date.strftime(\"%d-%m-%Y\"))\n",
        "                elif match[1] in ['friday', 'shukrawar']:\n",
        "                    # Find next Friday\n",
        "                    days_until_weekday = (today.weekday() +3) % 7\n",
        "                    next_date = today + timedelta(days=days_until_weekday )\n",
        "                    detected_dates.append(next_date.strftime(\"%d-%m-%Y\"))\n",
        "                elif match[1] in ['saturday', 'shaniwar']:\n",
        "                    # Find next Saturday\n",
        "                    days_until_weekday = (today.weekday() +4) % 7\n",
        "                    next_date = today + timedelta(days=days_until_weekday )\n",
        "                    detected_dates.append(next_date.strftime(\"%d-%m-%Y\"))\n",
        "                elif match[1] in ['sunday', 'raviwar']:\n",
        "                    # Find next Sunday\n",
        "                    days_until_weekday = (today.weekday() +5) % 7\n",
        "                    next_date = today + timedelta(days=days_until_weekday )\n",
        "                    detected_dates.append(next_date.strftime(\"%d-%m-%Y\"))\n",
        "                elif match[1] in ['week', 'hafte']:\n",
        "                    # Find next week\n",
        "                    next_date = today + timedelta(days=(7 - today.weekday())+6)\n",
        "                    detected_dates.append(next_date.strftime(\"%d-%m-%Y\"))\n",
        "                elif match[1] in ['month', 'mahine']:\n",
        "                    # Find next month\n",
        "                    next_date = today.replace(day=1, month=today.month+1)\n",
        "                    detected_dates.append(next_date.strftime(\"%d-%m-%Y\"))\n",
        "                elif match[1] in ['year', 'saal']:\n",
        "                    # Find next year\n",
        "                    next_date = today.replace(day=1, month=1, year=today.year+1)\n",
        "                    detected_dates.append(next_date.strftime(\"%d-%m-%Y\"))\n",
        "            elif match[0] == 'kal':\n",
        "                # Find tomorrow's date\n",
        "                next_date = today + timedelta(1)\n",
        "                detected_dates.append(next_date.strftime(\"%d-%m-%Y\"))\n",
        "\n",
        "    return detected_dates\n",
        "\n",
        "def main():\n",
        "    while True:\n",
        "        user_input = input(\"Enter a sentence (or 'exit' to quit): \")\n",
        "        if user_input.lower() == 'exit':\n",
        "            break\n",
        "        else:\n",
        "            detected_emails = detect_email(user_input)\n",
        "            detected_time =detect_time(user_input)\n",
        "            detected_phone_numbers = detect_phone_numbers(user_input)\n",
        "            detected_number_units = detect_numbers_with_units(user_input)\n",
        "            detected_dates = detect_dates(user_input)\n",
        "            if detected_emails:\n",
        "                print(\"Detected email(s):\", \", \".join(detected_emails))\n",
        "            else:\n",
        "                print(\"No email address detected.\")\n",
        "            if detected_time:\n",
        "                print(\"Detected time(s):\", \", \".join(detected_time))\n",
        "            else:\n",
        "                print(\"No time detected.\")\n",
        "            if detected_phone_numbers:\n",
        "                print(\"Detected phone number(s):\", \", \".join(detected_phone_numbers))\n",
        "            else:\n",
        "                print(\"No phone number detected.\")\n",
        "            if detected_number_units:\n",
        "                print(\"Detected number(s) with respective units:\")\n",
        "                for match in detected_number_units:\n",
        "                    number, unit = match\n",
        "                    print(f\"Number: {number}, Unit: {unit}\")\n",
        "            else:\n",
        "                print(\"No numbers with respective units detected.\")\n",
        "            if detected_dates:\n",
        "                print(\"Detected date(s):\", \", \".join(detected_dates))\n",
        "            else:\n",
        "                print(\"No dates detected.\")\n",
        "\n",
        "\n",
        "\n",
        "if __name__ == \"__main__\":\n",
        "    main()"
      ]
    },
    {
      "cell_type": "code",
      "source": [],
      "metadata": {
        "id": "_mZXqqt8jsNf"
      },
      "execution_count": null,
      "outputs": []
    }
  ]
}