tool.py

import os
import sys
import time
import platform
import subprocess
import pandas as pd
from datetime import datetime
from colorama import init, Fore

init()

ascii_art = '''
     ██╗███████╗███████╗     ██████╗ ██████╗ ██╗   ██╗███╗   ██╗███████╗███████╗██╗     ██╗      ██████╗ ██████╗ 
     ██║██╔════╝██╔════╝    ██╔════╝██╔═══██╗██║   ██║████╗  ██║██╔════╝██╔════╝██║     ██║     ██╔═══██╗██╔══██╗
     ██║█████╗  █████╗      ██║     ██║   ██║██║   ██║██╔██╗ ██║███████╗█████╗  ██║     ██║     ██║   ██║██████╔╝
██   ██║██╔══╝  ██╔══╝      ██║     ██║   ██║██║   ██║██║╚██╗██║╚════██║██╔══╝  ██║     ██║     ██║   ██║██╔══██╗
╚█████╔╝███████╗███████╗    ╚██████╗╚██████╔╝╚██████╔╝██║ ╚████║███████║███████╗███████╗███████╗╚██████╔╝██║  ██║
 ╚════╝ ╚══════╝╚══════╝     ╚═════╝ ╚═════╝  ╚═════╝ ╚═╝  ╚═══╝╚══════╝╚══════╝╚══════╝╚══════╝ ╚═════╝ ╚═╝  ╚═╝
                                                 - A command-line tool for JEE counselling assistance by Ksauraj.
'''

def pre_setup():
    os.system("cls" if os.name == "nt" else "clear")
    print(Fore.GREEN + ascii_art)
    print(Fore.YELLOW + "Select Counseling type:")
    print(Fore.GREEN + "1." + Fore.BLUE + "JOSAA")
    print(Fore.GREEN + "2." + Fore.BLUE + "CSAB" + Fore.RESET)
    option = input("Select Option (1 to 2) : ")
    if option == '1':
        josaa_rounds_year()
    elif option == '2':
        csab_rounds_year()
    else:
        pre_setup()


def josaa_rounds_year():
    os.system("cls" if os.name == "nt" else "clear")
    print(Fore.GREEN + ascii_art)
    print(Fore.YELLOW + "Select JOSAA round year")
    print(Fore.GREEN + "1." + Fore.BLUE + "2022")
    print(Fore.GREEN + "2." + Fore.BLUE + "2023" + Fore.RESET)
    josaa_round_year_sel = input("Select Option (1 to 2): ")
    if josaa_round_year_sel == "1":
        josaa_round_year = "2022"
    elif josaa_round_year_sel == "2":
        josaa_round_year = "2023"
    print(josaa_round_year)
    josaa_rounds(josaa_round_year)


def josaa_rounds(josaa_round_year):
    os.system("cls" if os.name == "nt" else "clear")
    print(Fore.GREEN + ascii_art)
    print("{Fore.YELLOW}Select JOSAA round {{josaa_round_year}}")

    menu_options = [
        "Round 1",
        "Round 2",
        "Round 3",
        "Round 4",
        "Round 5",
        "Round 6"
    ]

    for i, option in enumerate(menu_options, start=1):
        print(f"{Fore.GREEN}{i}. {Fore.RESET}{Fore.BLUE}{option}" + Fore.RESET)

    selected_round = input("Select Option (1 to 6): ")
    if int(selected_round) > 6:
        return josaa_rounds()
    csv_files("josaa", selected_round, josaa_round_year)


def csab_rounds_year():
    os.system("cls" if os.name == "nt" else "clear")
    print(Fore.GREEN + ascii_art)
    print(Fore.YELLOW + "Select CSAB round year")
    print(Fore.GREEN + "1." + Fore.BLUE + "2021")
    print(Fore.GREEN + "2." + Fore.BLUE + "2022")
    print(Fore.GREEN + "3." + Fore.BLUE + "2023" + Fore.RESET)
    csab_round_year_sel = input("Select Option (1 to 2): ")
    if csab_round_year_sel == "1":
        csab_round_year = "2021"
    elif csab_round_year_sel == "2":
        csab_round_year = "2022"
    elif csab_round_year_sel == "3":
        csab_round_year = "2023"
    print(csab_round_year)
    csab_rounds(csab_round_year)


def csab_rounds(csab_round_year):
    os.system("cls" if os.name == "nt" else "clear")
    print(Fore.GREEN + ascii_art)
    print(f"{Fore.YELLOW}Select CSAB round ({csab_round_year})" )
    print(Fore.GREEN + "1." + Fore.BLUE + "Round 1")
    print(Fore.GREEN + "2." + Fore.BLUE + "Round 2")
    csab_round = input(Fore.RESET + "Select Option (1 to 2) : ")
    csv_files("csab", csab_round, csab_round_year)

# define the path of csv files for different types of colleges


def csv_files(type, round, year):
    # Get path to the temporary folder created by PyInstaller
    if getattr(sys, 'frozen', False):
        # If the script is running in a PyInstaller bundle
        cwd = sys._MEIPASS
    else:
        # If the script is running from the original Python file
        cwd = os.getcwd()
    if type == "josaa":
        josaa_rounds = round
        CSV_FILES = {
            "ALL": os.path.join(cwd, "josaa", f"{year}", f"round_{josaa_rounds}", "ranks_all.csv"),
            "IIITs": os.path.join(cwd, "josaa", f"{year}", f"round_{josaa_rounds}", "ranks_iiits.csv"),
            "IITs": os.path.join(cwd, "josaa", f"{year}", f"round_{josaa_rounds}", "ranks_iits.csv"),
            "NITs": os.path.join(cwd, "josaa", f"{year}", f"round_{josaa_rounds}", "ranks_nits.csv"),
            "GFTIs": os.path.join(cwd, "josaa", f"{year}", f"round_{josaa_rounds}", "ranks_gftis.csv")
        }
        josaa_institue_types(CSV_FILES)
    elif type == "csab":
        csab_rounds = round
        csv_path = os.path.join(
            cwd,
            "csab",
            f"{year}",
            f"round_{csab_rounds}",
            "ranks.csv")
        df = pd.read_csv(csv_path)
        csab_institue_types(df)


def csab_institue_types(df):
    # clear the screen
    os.system("cls" if os.name == "nt" else "clear")
    print(Fore.GREEN + ascii_art)

    # ask for user input for institute type
    print(Fore.YELLOW + "Select Institute type:")
    print(Fore.GREEN + "1." + Fore.BLUE + "ALL")
    print(Fore.GREEN + "2." + Fore.BLUE + "IIITs")
    print(Fore.GREEN + "3." + Fore.BLUE + "NITs")
    print(Fore.GREEN + "4." + Fore.BLUE + "GFTIs")
    option = input(Fore.RESET + "Select Option (1 to 4): ")

    # filter the dataframe based on the selected option
    if option == "1":
        df = df[~df['Institute'].str.contains(
            'Indian Institute of Technology')]
    elif option == "2":
        df = df[df['Institute'].str.contains(
            'Indian Institute of Information Technology')]
    elif option == "3":
        df = df[df['Institute'].str.contains(
            'National Institute of Technology')]
    elif option == "4":
        df = df[~df['Institute'].str.contains(
            'National Institute of Technology')]
        df = df[~df['Institute'].str.contains(
            'Indian Institute of Information Technology')]
    else:
        print("Invalid option. Please select again.")

    main(df)


def josaa_institue_types(CSV_FILES):
    # clear the screen
    os.system("cls" if os.name == "nt" else "clear")
    print(Fore.GREEN + ascii_art)

    # ask for user input for institute type
    print(Fore.YELLOW + "Select Institute type:")
    print(Fore.GREEN + "1." + Fore.BLUE + "ALL (No IITs Included)")
    print(Fore.GREEN + "2." + Fore.BLUE + "IIITs")
    print(Fore.GREEN + "3." + Fore.BLUE + "NITs")
    print(Fore.GREEN + "4." + Fore.BLUE + "GFTIs")
    print(Fore.GREEN + "5." + Fore.BLUE + "IITs")
    option = input(Fore.RESET + "Select Option (1 to 5): ")

    # filter the dataframe based on the selected option
    if option == "1":
        college_type = "ALL"
    elif option == "2":
        college_type = "IIITs"
    elif option == "3":
        college_type = "NITs"
    elif option == "4":
        college_type = "GFTIs"
    elif option == "5":
        college_type = "IITs"
    else:
        print("Invalid option. Please select again.")
        josaa_institue_types

    # read the csv file based on the selected college type
    csv_path = CSV_FILES[college_type]
    df = pd.read_csv(csv_path)
    if option == "1":
        # Remove all IITs occurance if "ALL" option was choosed.
        df = df[~df['Institute'].str.contains(
            'Indian Institute of Technology')]
        main(df)
    elif option == "5":
        # fix ranks with strings in it.
        df['Closing Rank'] = df['Closing Rank'].str.extract(
            r'(\d+)').astype(float)
        df['Opening Rank'] = df['Opening Rank'].str.extract(
            r'(\d+)').astype(int)
        main(df)
    else:
        main(df)


def filter_programs(institute_df):
    print(Fore.GREEN + ascii_art)
    print("Note: Programs marked with '*' will display all the programs similar to them.")
    print("Select Program:")
    print(Fore.GREEN + "1." + Fore.BLUE + "All")
    print(Fore.GREEN + "2." + Fore.BLUE + "Computer Science and Engineering*")
    print(Fore.GREEN + "3." + Fore.BLUE +
          "Artificial Intelligence and Data Science*")
    print(Fore.GREEN + "4." + Fore.BLUE +
          "Electronics and Communication Engineering*")
    print(Fore.GREEN + "5." + Fore.BLUE + "Information Technology*")
    print(Fore.GREEN + "6." + Fore.BLUE + "Mechanical Engineering*")
    print(Fore.GREEN + "7." + Fore.BLUE + "Civil Engineering*")
    print(Fore.GREEN + "8." + Fore.BLUE + "Electrical Engineering*")
    print(Fore.GREEN + "9." + Fore.BLUE + "Data Science and Engineering*")
    print(Fore.GREEN + "10." + Fore.BLUE + "Biotechnology*")
    print(Fore.GREEN + "11." + Fore.BLUE + "Chemical Engineering*")
    print(Fore.GREEN + "12." + Fore.BLUE + "Smart Manufacturing*")
    print(Fore.GREEN + "13." + Fore.YELLOW + "Check Next Page" + Fore.RESET)
    print("")

    program_choices = {
        1: '',
        2: 'Computer Science and Engineering',
        3: 'Artificial Intelligence and Data Science',
        4: 'Electronics and Communication Engineering',
        5: 'Information Technology',
        6: 'Mechanical Engineering',
        7: 'Civil Engineering',
        8: 'Electrical Engineering',
        9: 'Data Science and Engineering',
        10: 'Biotechnology',
        11: 'Chemical Engineering',
        12: 'Smart Manufacturing'
    }

    program_input = input("Choose Option(s) (separated by space): ")
    program_choices_list = program_input.split()

    filtered_df = institute_df

    if '1' not in program_choices_list:
        filtered_df = institute_df[institute_df["Academic Program Name"].str.contains(
            '|'.join([program_choices.get(int(choice), '') for choice in program_choices_list]))]

    if len(filtered_df) == 0:
        print(
            Fore.RED +
            "No programs found matching the selected options." +
            Fore.RESET)
        return filtered_df

    if '13' in program_choices_list:
        os.system("cls" if os.name == "nt" else "clear")
        programs = filtered_df["Academic Program Name"].unique()
        for i, program in enumerate(programs, start=14):
            print(f"{Fore.GREEN}{i}. {Fore.BLUE}{program}{Fore.RESET}")
        program_choice = int(input("Choose Option: "))
        if program_choice >= 14 and program_choice < 14 + len(programs):
            program = programs[program_choice - 14]
            filtered_df = filtered_df[filtered_df["Academic Program Name"] == program]
        else:
            print(Fore.RED + "Invalid choice. Please try again." + Fore.RESET)
            return filtered_df

    return filtered_df


def display_df_web(df, heading, subheading):
    output_dir = 'output'
    if not os.path.exists(output_dir):
        os.makedirs(output_dir)
    # create a unique filename based on the current date and time
    filename = os.path.join(
        output_dir,
        f"{datetime.now().strftime('%Y-%m-%d_%H-%M-%S')}.html")

    # convert the DataFrame to an HTML table
    html_table = df.to_html(index=False, classes='table')

    # Generate the complete HTML content with headings, CSS styles, and the
    # table
    html_content = f'''
    <!DOCTYPE html>
    <html>
    <head>
        <title>{heading}</title>
        <style>
        html, body {{
            height: 100%;
            text-align: center;
            font-family: Tahoma, sans-serif;
        }}

        body {{
            margin: 30px;
        }}

        table.table {{
            border-collapse: separate;
            border-spacing: 0;
            min-width: 350px;
            border: none;  /* Remove the border attribute */
        }}

        table.table tr th,
        table.table tr td {{
            border-right: 1px solid #bbb;
            border-bottom: 1px solid #bbb;
            padding: 5px;
        }}

        table.table tr th:first-child,
        table.table tr td:first-child {{
            border-left: 1px solid #bbb;
        }}

        table.table tr th {{
            background: #eee;
            border-top: 1px solid #bbb;
            text-align: center;
        }}

        /* top-left border-radius */
        table.table tr:first-child th:first-child {{
            border-top-left-radius: 6px;
        }}

        /* top-right border-radius */
        table.table tr:first-child th:last-child {{
            border-top-right-radius: 6px;
        }}

        /* bottom-left border-radius */
        table.table tr:last-child td:first-child {{
            border-bottom-left-radius: 6px;
        }}

        /* bottom-right border-radius */
        table.table tr:last-child td:last-child {{
            border-bottom-right-radius: 6px;
        }}

        a {{
            text-decoration: none;
            color: #18272F;
            font-weight: 700;
        position: relative;
        }}

        a::before {{
        content: '';
        background-color: hsla(0, 100%, 50%, 0.236);
        position: absolute;
        left: 0;
        bottom: 3px;
        width: 100%;
        height: 8px;
        z-index: -1;
        transition: all .3s ease-in-out;
        }}

        a:hover::before {{
        bottom: 0;
        height: 100%;
        background-color: hsla(142, 61%, 58%, 0.359);
        }}

        table.table tbody tr:hover {{
            background-color: rgba(159, 131, 86, 0.3);
            filter: none;
            transform: scale(1.005);
        }}
        </style>
    </head>
    <body>
        <h1><a href="https://github.com/ksauraj/jee_counsellor">{heading}</a></h1>
        <h2><a href="https://sauraj.eu.org">{subheading}</a></h2>
        <div class="container">
            {html_table}
        </div>
    </body>
    </html>
    '''

    # Save the HTML content to the file
    with open(filename, "w") as file:
        file.write(html_content)

    print(filename)
    # open the file in the default web browser

    if platform.system() == "Windows":
        subprocess.Popen(["start",
                          filename],
                         stdout=subprocess.DEVNULL,
                         stderr=subprocess.DEVNULL,
                         shell=True)
    elif platform.system() == "Darwin":
        subprocess.Popen(["open", filename],
                         stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
    else:
        subprocess.Popen(["xdg-open", filename],
                         stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)


# main function to run the CLI tool
def main(df):
    while True:
        institute_df = df
        os.system("cls" if os.name == "nt" else "clear")
        filtered_df = filter_programs(institute_df)

        def filter_by_choices(choices, column_name):
            unique_choices = institute_df[column_name].unique()

            os.system("cls" if os.name == "nt" else "clear")
            print(Fore.GREEN + ascii_art)
            print(f"{Fore.YELLOW}Select {column_name}:")

            print(Fore.GREEN + "1." + Fore.BLUE +" All")
            for i, choice in enumerate(unique_choices, start=2):
                print(f"{Fore.GREEN}{i}. {Fore.BLUE}{choice}{Fore.RESET}")

            print(f"{Fore.YELLOW}")
            print(f"You are Selecting {column_name}")
            choices_input = input("Choose Options ((space-separated, e.g., 2 3 4) & 1 for all choices) : ")
            selected_choices = list(map(int, choices_input.split()))

            if 1 in selected_choices:
                return filtered_df
            else:
                selected_choices = [choice - 2 for choice in selected_choices]
                filtered_choices = [unique_choices[i] for i in selected_choices]
                return filtered_df[filtered_df[column_name].isin(filtered_choices)]

        filtered_df = filter_by_choices(institute_df["Institute"], "Institute")
        filtered_df = filter_by_choices(institute_df["Quota"], "Quota")
        filtered_df = filter_by_choices(institute_df["Seat Type"], "Seat Type")
        filtered_df = filter_by_choices(institute_df["Gender"], "Gender")

        os.system("cls" if os.name == "nt" else "clear")
        rank = int(input(Fore.YELLOW + "Enter your rank: " + Fore.RESET))
        filtered_df["Closing Rank"] = filtered_df["Closing Rank"].astype(
            str).str.extract(r"(\d+)").astype(int)
        filtered_df = filtered_df[filtered_df["Closing Rank"] > rank].sort_values(
            by=["Closing Rank"], ascending=True)

        os.system("cls" if os.name == "nt" else "clear")
        display_df_web(filtered_df, "JEE Counsellor", "-By Ksauraj")
        print(
            Fore.GREEN +
            "Congratulations! File successfully opened in browser." +
            Fore.RESET)
        time.sleep(3)
        os.system("cls" if os.name == "nt" else "clear")
        print(Fore.GREEN + ascii_art)
        choice = input(
            f"{Fore.BLUE}1. {Fore.YELLOW}Go back to main menu.\n"
            f"{Fore.BLUE}2. {Fore.YELLOW}Continue with same settings.\n"
            f"{Fore.BLUE}3. {Fore.YELLOW}Exit from tool.\n"
            f"{Fore.RESET}Select option (1-3): ")
        if choice == "1":
            pre_setup()
        elif choice == "3":
            break


# run the main function
pre_setup()