Update app.py
Browse files
app.py
CHANGED
|
@@ -533,6 +533,17 @@ def handle_message(data):
|
|
| 533 |
emit('json_response', response, room=session_id)
|
| 534 |
session_store.pop(session_id, None)
|
| 535 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 536 |
# Home route
|
| 537 |
@app.route("/")
|
| 538 |
def index_view():
|
|
|
|
| 533 |
emit('json_response', response, room=session_id)
|
| 534 |
session_store.pop(session_id, None)
|
| 535 |
|
| 536 |
+
|
| 537 |
+
@socketio.on('example')
|
| 538 |
+
def handle_message(data):
|
| 539 |
+
img_url = data['img_url']
|
| 540 |
+
message = data['message']
|
| 541 |
+
session_id = request.sid
|
| 542 |
+
image_array = base64_to_numpy(img_url)
|
| 543 |
+
response = get_answer(image=image_array, message=message, sessionID=request.sid)
|
| 544 |
+
emit('response', response, room=session_id)
|
| 545 |
+
return response
|
| 546 |
+
|
| 547 |
# Home route
|
| 548 |
@app.route("/")
|
| 549 |
def index_view():
|