Update app.py

app.py

@@ -14,7 +14,6 @@ async def handle_client(websocket, path):
     try:
         # Receive the client's request (create a pair or connect to a key)
         request_type = await websocket.recv()
-        print(request_type)
 
         if request_type == "create_pair":
             # Generate a unique key
@@ -31,17 +30,17 @@ async def handle_client(websocket, path):
             # Receive the key from the client
             key_to_connect = await websocket.recv()
 
-            # Retrieve all clients associated with the key from the database
+            # Check if the key exists in the database
             cursor.execute('SELECT client_ws FROM pairs WHERE key=?', (key_to_connect,))
-            results = cursor.fetchall()
+            result = cursor.fetchone()
+
+            if result:
+                other_client_ws = result[0]
+
+                # Pair the clients
+                await websocket.send(f"Connected to key {key_to_connect}. You are paired with {other_client_ws}")
+                await other_client_ws.send(f"Connected to key {key_to_connect}. You are paired with {str(websocket)}")
 
-            if results:
-                # Pair the current client with all clients associated with the key
-                for result in results:
-                    other_client_ws = result[0]
-                    if other_client_ws != str(websocket):
-                        await websocket.send(f"Connected to key {key_to_connect}. You are paired with {other_client_ws}")
-                        await other_client_ws.send(f"Connected to key {key_to_connect}. You are paired with {str(websocket)}")
             else:
                 # Inform the client that the key does not exist
                 await websocket.send(f"Key {key_to_connect} does not exist.")
@@ -53,7 +52,7 @@ def generate_key():
     # Generate a random key (you might want to implement a more robust key generation logic)
     return ''.join(random.choices(string.ascii_uppercase + string.digits, k=8))
 
-start_server = websockets.serve(handle_client, "0.0.0.0", 7860)
+start_server = websockets.serve(handle_client, "localhost", 8765)
 
 async def main():
     print("WebSocket server is running. Waiting for connections...")