TahaFawzyElshrif commited on
Commit
5a8e898
·
1 Parent(s): acee99e

working on queue, finalization

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -79,7 +79,7 @@ def old_call(request: RequestModel):
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
 
@@ -90,7 +90,10 @@ def call(request: RequestAnswer):
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)}
96
 
 
79
 
80
  @app.post("/Message/Send/")
81
  def call(request: RequestModel):
82
+ redis_send(request.user_id,request.msg_id,{"status": "pending"})
83
  return send_message(json.dumps(request.model_dump()))
84
 
85
 
 
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
+ elif "status" in answer and json.loads(answer)["status"] == "pending":
94
+ return {"status": "processing"}
95
  else:
96
+
97
  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
98
  return {"status": "ready", "data": json.loads(answer)}
99