|
@ -33,6 +33,7 @@ redis_ = redis.Redis(connection_pool=pool, decode_responses=True) |
|
|
|
|
|
|
|
|
db_key_query = 'query' |
|
|
db_key_query = 'query' |
|
|
db_key_querying = 'querying' |
|
|
db_key_querying = 'querying' |
|
|
|
|
|
db_key_error = 'error' |
|
|
|
|
|
|
|
|
@app.route("/search", methods=["POST"]) |
|
|
@app.route("/search", methods=["POST"]) |
|
|
def handle_query(): |
|
|
def handle_query(): |
|
@ -47,6 +48,9 @@ def handle_query(): |
|
|
code = result_dict["status_code"] |
|
|
code = result_dict["status_code"] |
|
|
texts = result_dict["texts"] |
|
|
texts = result_dict["texts"] |
|
|
probabilities = result_dict["probabilities"] |
|
|
probabilities = result_dict["probabilities"] |
|
|
|
|
|
if str(code) == 400: |
|
|
|
|
|
redis_.rpush(db_key_error, json.dumps({"id": id_})) |
|
|
|
|
|
return False |
|
|
result_text = {'code': code, 'text': texts, 'probabilities': probabilities} |
|
|
result_text = {'code': code, 'text': texts, 'probabilities': probabilities} |
|
|
else: |
|
|
else: |
|
|
querying_list = list(redis_.smembers("querying")) |
|
|
querying_list = list(redis_.smembers("querying")) |
|
@ -85,4 +89,4 @@ def handle_query(): |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
if __name__ == "__main__": |
|
|
if __name__ == "__main__": |
|
|
app.run(debug=False, host='0.0.0.0', port=14003) |
|
|
app.run(debug=False, host='0.0.0.0', port=14005) |
|
|