Merge pull request #8 from JEENB/sendimage

everything works

Author: JEENB

__pycache__/models.cpython-38.pyc

@@ -102,7 +102,59 @@ def send_model_data():
     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 model.....")
+        dpg.add_text("Model is getting trained by the server.....")
+
+    #receive image
+    file_size = client.recv(1024).decode()
+
+    print(file_size)
+    file_path = '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('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
@@ -129,8 +181,21 @@ def send_model_data():
     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("Your final report downloaded at location: " + str(file_path))
+
+
+
 def get_model_inputs():
     pass
 

app.py

@@ -68,10 +68,7 @@ def reg_training_model(df, user_dir, degree = 1, split_ratio = 0.2):
 	plt.legend(loc='upper left')	
 	plt.title("Polynomial Fitting")
 	plt.savefig(f"{user_dir}/fitting.png")
-<<<<<<< HEAD
 	# plt.show()
-=======
->>>>>>> 65771cfe70696bdfb5aef65df8cf3f8aa223ab0b
 
 
 	##-------------------------------------------------

models.py

@@ -1,4 +1,3 @@
-
 import os
 import socket
 import time
@@ -76,6 +75,39 @@
 
     tot, maxi, mini, mean, medi, sd = stats_summary(df)
     print("model training done......")
+
+    #send 1 image to client
+    file_path = f'./server_data/{initial}/fitting.png'
+    file_size = os.path.getsize(file_path)
+    print(file_size)
+    conn.send(str(file_size).encode())
+
+
+    with open(file_path, "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
+            conn.sendall(data)
+            c += len(data)
+
+        # Ending the time capture.
+        end_time = time.time()
+    file.close()
+    print("[INFO] Fitting image Transfer Complete.Total time: ", end_time - start_time)
+
+
+    #client "finalising report......"
+    # conn.send("finalising report....".encode())
+
+    #file report downloaded. Local path"receivedreport.pdf"
+
+
     c_report(tot, maxi, mini, mean, medi, sd, training_points, testing_points, degree, split = training_ratio, user_dir = user_dir)
     print(str(tot))
     print("report generated now")
@@ -102,7 +134,7 @@
 
         # Ending the time capture.
         end_time = time.time()
-
+    file.close()
     print("[INFO] Report file Transfer Complete.Total time: ", end_time - start_time)