Spaces:
Running
Running
TahaFawzyElshrif commited on
Commit ·
acee99e
1
Parent(s): ceaf913
working on queue, finalization
Browse files
app.py
CHANGED
|
@@ -11,7 +11,7 @@ import subprocess
|
|
| 11 |
from Queue_Producer import send_message
|
| 12 |
import redis
|
| 13 |
from utils import RequestModel, RequestAnswer
|
| 14 |
-
|
| 15 |
|
| 16 |
##################################################
|
| 17 |
# VARIABLES
|
|
@@ -79,6 +79,7 @@ def old_call(request: RequestModel):
|
|
| 79 |
|
| 80 |
@app.post("/Message/Send/")
|
| 81 |
def call(request: RequestModel):
|
|
|
|
| 82 |
return send_message(json.dumps(request.model_dump()))
|
| 83 |
|
| 84 |
|
|
@@ -88,7 +89,7 @@ def call(request: RequestAnswer):
|
|
| 88 |
try:
|
| 89 |
answer = redis_conn.get(f'ANSWER_FOR_USER_ID{request.user_id}_OF_{request.msg_id}')
|
| 90 |
if answer is None:
|
| 91 |
-
return {"status": "error"
|
| 92 |
else:
|
| 93 |
redis_conn.delete(f'ANSWER_FOR_USER_ID{request.user_id}_OF_{request.msg_id}') # Clean up after fetching for memory and better secure as double call is wrong
|
| 94 |
return {"status": "ready", "data": json.loads(answer)}
|
|
|
|
| 11 |
from Queue_Producer import send_message
|
| 12 |
import redis
|
| 13 |
from utils import RequestModel, RequestAnswer
|
| 14 |
+
from Consumer import redis_send
|
| 15 |
|
| 16 |
##################################################
|
| 17 |
# VARIABLES
|
|
|
|
| 79 |
|
| 80 |
@app.post("/Message/Send/")
|
| 81 |
def call(request: RequestModel):
|
| 82 |
+
redis_send(request.user_id,request.msg_id,{"status": "processing"})
|
| 83 |
return send_message(json.dumps(request.model_dump()))
|
| 84 |
|
| 85 |
|
|
|
|
| 89 |
try:
|
| 90 |
answer = redis_conn.get(f'ANSWER_FOR_USER_ID{request.user_id}_OF_{request.msg_id}')
|
| 91 |
if answer is None:
|
| 92 |
+
return {"status": "error"}
|
| 93 |
else:
|
| 94 |
redis_conn.delete(f'ANSWER_FOR_USER_ID{request.user_id}_OF_{request.msg_id}') # Clean up after fetching for memory and better secure as double call is wrong
|
| 95 |
return {"status": "ready", "data": json.loads(answer)}
|