no ws

app.py

@@ -1,4 +1,4 @@
-from fastapi import FastAPI, HTTPException, WebSocket, WebSocketDisconnect
+from fastapi import FastAPI, HTTPException, Depends
 from pydantic import BaseModel
 from typing import Dict, List
 import bcrypt
@@ -36,7 +36,6 @@ def generate_rsa_key_pair():
         key_size=2048
     )
     public_key = private_key.public_key()
-    # Serialize the private key with encryption (optional)
     private_key_pem = private_key.private_bytes(
         encoding=serialization.Encoding.PEM,
         format=serialization.PrivateFormat.TraditionalOpenSSL,
@@ -125,39 +124,8 @@ async def delete_message(recipient: str, message: str):
     logger.info(f"Message deleted from {recipient}: {message}")
     return JSONResponse(content={"status": "Message deleted"})
 
-@app.websocket("/ws/{username}")
-async def websocket_endpoint(websocket: WebSocket, username: str):
-    if username not in users:
-        await websocket.close(code=4000)
-        return
-
-    await websocket.accept()
-    try:
-        while True:
-            data = await websocket.receive_text()
-            message_data = json.loads(data)
-            event_type = message_data.get("type")
-            recipient = message_data.get("recipient")
-            message = message_data.get("message")
-
-            if event_type == "send_message":
-                if recipient in messages:
-                    messages[recipient].append(message)
-                else:
-                    messages[recipient] = [message]
-
-                await websocket.send_text(json.dumps({"event": "message_received", "recipient": recipient, "message": message}))
-
-            elif event_type == "get_messages":
-                if username in messages:
-                    await websocket.send_text(json.dumps({"event": "messages_list", "messages": messages[username]}))
-                else:
-                    await websocket.send_text(json.dumps({"event": "messages_list", "messages": []}))
-
-            else:
-                await websocket.send_text(json.dumps({"event": "error", "message": "Unknown event type"}))
-    except WebSocketDisconnect:
-        logger.info(f"WebSocket disconnected: {username}")
-    except Exception as e:
-        logger.error(f"Error in WebSocket communication: {e}")
-        await websocket.close(code=1011)  # Internal server error
+@app.get("/poll_messages/{username}")
+async def poll_messages(username: str):
+    if username not in messages:
+        return JSONResponse(content={"messages": []})
+    return JSONResponse(content={"messages": messages[username]})