client end messages

client.py

@@ -28,17 +28,13 @@ user_input = input("Enter username: ")
 while connect:
     server_msg = send(user_input)
     print(server_msg)
-    user_input = input()
-    if user_input.upper() == "BYE" or server_msg == "ERR" or server_msg == "DONE" or server_msg == "END":
+    if server_msg == "ERR" or server_msg == "DONE" or server_msg == "END":
         connect = False
+    else:
+        user_input = input()
+        if user_input.upper() == "BYE":
+            connect = False
 
 
 # disconnect
 print(send("BYE"))
-
-
-
-
-
-
-