Update server.py

MP1/server.py

@@ -38,13 +38,15 @@ while True:
     print(f"Connected by {addr}")
 
     command = receive_message(conn)
-    # print("Received Grep Command:", command.decode())
+    if command is None:
+        conn.close()
+        continue
 
-    command_args = shlex.split(command.decode())
+    command_args = shlex.split(command)
 
     if not command_args or command_args[0] != "grep":
         response = "Invalid command. Please use the 'grep' command."
-        conn.sendall(response.encode())
+        send_message(conn, response)
         conn.close()
         continue