app.py

import dearpygui.dearpygui as dpg
import os


#client code starts#
import os
import socket
import time
import re


email_regex = '[A-Za-z0-9._%+-]+@ashoka\.edu\.in'
PORT = 2223
client = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
client.connect((socket.gethostname(), PORT))
print("[INFO] Connected to server")
 #client code ends#

width = 900
height = 750
pos = (0,-20)

client_dir = os.getcwd()

def login_callback(sender, app_data, user_data):
   
    print("Save Clicked")
    print(f"sender is {sender}")
    print(f"app_data is {app_data}")
    print(f"user data is {user_data}")
    email = dpg.get_value("username")
    pswd = dpg.get_value("password")
    if not re.search(email_regex, email):
        dpg.delete_item("original")
        with dpg.window(label="Example Window", width=width, height=height, tag="original", pos=pos):

            dpg.add_text("Email must end with '@ashoka.edu.in'.")
            dpg.add_text("Login again to continue.")

    else:

       
        client.send(email.encode())
        client.send(pswd.encode())

        msg = client.recv(1024).decode()
        print(msg)
    
        if msg == "Error!":
            dpg.delete_item("original")
            with dpg.window(label="Example Window", width=width, height=height, tag="original", pos=pos):
                dpg.add_text("Login Failed!!")
                dpg.add_text("Please re-run to login.")
        else:
            dpg.delete_item("original")
            
            with dpg.window(label="Example Window", width=width, height=height, tag="original", pos=pos):
                dpg.add_text("you have logged in")
                with dpg.file_dialog(directory_selector=False, show=False, callback=choose_file_callback, id="file_dialog_id"):
                    dpg.add_file_extension(".csv")
                    # dpg.add_file_extension("", color=(150, 255, 150, 255))
                    # dpg.add_file_extension("Source files (*.cpp *.h *.hpp){.cpp,.h,.hpp}", color=(0, 255, 255, 255))
                    # dpg.add_file_extension(".h", color=(255, 0, 255, 255), custom_text="[header]")
                    # dpg.add_file_extension(".py", color=(0, 255, 0, 255), custom_text="[Python]")
                dpg.add_button(label="File Selector", callback=lambda: dpg.show_item("file_dialog_id"))




def choose_file_callback(sender, app_data):
    print(f"Sender: {sender}")
    print(f"App Data: {app_data}")
    file_name = app_data["file_name"]
    file_path = app_data["file_path_name"]
    print(file_name)

    #============client code starts===========#
    file_size = os.path.getsize(file_path)
    print(file_size)
    print("[INFO] sending file size")
    client.send(str(file_size).encode())
    

    # Opening file and sending data.
    with open(file_name, "rb") as file:
        c = 0
        # Starting the time capture.
        start_time = time.time()

        # Running loop while c != file_size.
        while c <= file_size:
            data = file.read(1024)
            if not (data):
                break
            client.sendall(data)
            c += len(data)

        # Ending the time capture.
        end_time = time.time()

    print("[INFO] File Transfer Complete.Total time: ", end_time - start_time)
    #============client code ends==========#
    dpg.delete_item("original")
    with dpg.window(label="Example Window", width=width, height=height, tag="original", pos=pos):
        dpg.add_text("Your file has been selected and sent to server.")
        dpg.add_text("Enter the desired degree of your regression model. Eg-3")
        dpg.add_input_text(label="Regression Degree", tag="model_degree")
        dpg.add_text("Slide to choose the desired percentage of dataset for testing. Rest will be used for training")
        dpg.add_slider_int(label="Testing percentage", tag="model_test_percentage")
        dpg.add_button(label="Send to server", callback=send_model_data)
    

def send_model_data():
    model_degree = dpg.get_value("model_degree")
    client.send(model_degree.encode())
    print(f"model degree is {model_degree}")
    model_test_percentage = str(dpg.get_value("model_test_percentage"))
    print(dpg.get_value("model_degree"))
    print(dpg.get_value("model_test_percentage"))
    #======client code starts=====
    
    client.send(model_test_percentage.encode())
    #========client code ends=======


    dpg.delete_item("original")
    with dpg.window(label="Example Window", width=width, height=height, tag="original", pos=pos):
        dpg.add_text("Data sent to server")
        dpg.add_text("Model is getting trained by the server.....")

    #receive image
    file_size = client.recv(1024).decode()

    print(file_size)
    file_path = f'{client_dir}/client_data/receivedFitting.png'
    file = open(file_path, "wb")
    c = 0
    # Starting the time capture.
    start_time = time.time()

    # Running the loop while file is recieved.
    
    while c < int(file_size):
        data = client.recv(1024)
        if not (data):
            break
        file.write(data)
        c += len(data)
    file.close()

    # Ending the time capture.
    end_time = time.time()

    print("[INFO]: Fitting image file received. Total time: ", end_time - start_time)

    #recv msg that model is trained. Show to GUI with text and image.
    #=====need to adjust image size
    dpg.delete_item("original")
    with dpg.window(label="Example Window", width=900, height=750, tag="original", pos=pos):
        dpg.add_text("Your model is trained. ")
        #show image
        width1, height1, channels, data = dpg.load_image('./client_data/receivedFitting.png')
        
        with dpg.texture_registry(show=True):
            dpg.add_static_texture(width1, height1, data, tag="texture_tag")
        dpg.add_image("texture_tag")
        # with dpg.window(label="Tutorial"):
            
        
        
        
        #show msg "finalising report...."
        # msg = client.recv(1024).decode()

        dpg.add_text("finalising report...")



    
    



    #==============client code starts===========
            #Receive Report from server
    file_size = client.recv(1024).decode()
    print(file_size)
    file_path =  f'{client_dir}/client_data/sentreport.pdf'
    file = open(file_path, "wb")
    c = 0
    # Starting the time capture.
    start_time = time.time()

    # Running the loop while file is recieved.
    while c < int(file_size):
        data = client.recv(1024)
        if not (data):
            break
        file.write(data)
        c += len(data)
    file.close()

    # Ending the time capture.
    end_time = time.time()

    print("[INFO]: Report file received. Total time: ", end_time - start_time)




    #===========client code ends========


    #recreate the previous window along with received report's path
    # working_dir_path = os.getworkingdirectory()
    # new_file_path = working_dir_path + '/' + file_path

    dpg.delete_item("original")
    with dpg.window(label="Example Window", width=900, height=750, tag="original", pos=pos):
        dpg.add_text(f"Your final report downloaded at location: \n{client_dir}\\client_data\\sentreport.pdf" )



def get_model_inputs():
    pass

dpg.create_context()
dpg.create_viewport(title="Machine Learning on Server", width=width, height=height, x_pos=0, y_pos=-20)
dpg.setup_dearpygui()


with dpg.window(label="Example Window", width=width, height=height, tag="original", pos=pos):
    dpg.add_text("Enter your login credentials")
    dpg.add_input_text(label="User Name", tag="username")
    dpg.add_input_text(label="Password", tag="password", password=True)
    dpg.add_button(label="Login", callback=login_callback)

    # dpg.add_slider_float(label="float")

dpg.show_viewport()
dpg.start_dearpygui()
dpg.destroy_context()