Upload 2 files

main.py

@@ -1,29 +1,32 @@
-import logging
-import uvicorn
-from fastapi import FastAPI
-from pydantic import BaseModel
-import os
-
-logging.basicConfig(
-    format='%(asctime)s.%(msecs)03d %(levelname)-8s %(message)s',
-    level=logging.DEBUG,
-    datefmt='%Y-%m-%d %H:%M:%S'
-)
-
+from fastapi import FastAPI, WebSocket
 
 app = FastAPI()
+worker_ws = None
+client_ws = None
+
+
+@app.websocket("/ws")
+async def websocket_endpoint(websocket: WebSocket):
+    global client_ws
+    await websocket.accept()
+    client_ws = websocket
+    while True:
+        data = await websocket.receive_text()
+        if type(worker_ws) is WebSocket:
+            await worker_ws.send_json(data)
+        else:
+            await websocket.send_text(f"Worker is sleeping now!")
+
+
+@app.websocket("/worker/ws")
+async def websocket_endpoint(websocket: WebSocket):
+    global worker_ws
+    await websocket.accept()
+    worker_ws = websocket
+    while True:
+        data = await websocket.receive_json()
+        if type(client_ws) is WebSocket:
+            await client_ws.send_json(data)
+        else:
+            await websocket.send_text(f"Client is sleeping now!")
 
-requests = []
-
-@app.post("/sign", tags=["Classificator"])
-async def sign(data: str):
-    requests.append(data)
-    return result
-
-@app.get("/ping", tags=["TEST"])
-async def ping():
-    return "pong"
-
-
-if __name__ == "__main__":
-    uvicorn.run(app, host="127.0.0.1", port=8000)

requirements.txt

@@ -0,0 +1,19 @@
+annotated-types==0.6.0
+anyio==4.3.0
+click==8.1.7
+exceptiongroup==1.2.0
+fastapi==0.110.0
+h11==0.14.0
+httptools==0.6.1
+idna==3.6
+pydantic==2.6.4
+pydantic_core==2.16.3
+python-dotenv==1.0.1
+PyYAML==6.0.1
+sniffio==1.3.1
+starlette==0.36.3
+typing_extensions==4.10.0
+uvicorn==0.29.0
+uvloop==0.19.0
+watchfiles==0.21.0
+websockets==12.0