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

working on queue, finalization

Browse files
Files changed (1) hide show
  1. app.py +3 -2
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", "message": "No answer found in Redis."}
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)}