diff --git a/src/Python/flask/web/app.py b/src/Python/flask/web/app.py index d6e908c..549964e 100644 --- a/src/Python/flask/web/app.py +++ b/src/Python/flask/web/app.py @@ -163,6 +163,18 @@ def yolo_results_endpoint(): global yolo_results with lock: return jsonify(yolo_results) + try: + data = json.loads(yolo_results) + db = get_db() + with db.cursor() as cursor: + sql_yolo = "INSERT INTO image (class, confidence) VALUES (%s, %s)" + cursor.executemany(sql_yolo, data) + db.commit() + cursor.close() + except json.JSONDecodeError as e: + print(f"JSON decode error: {e}") + except mysql.connector.Error as err: + print(f"Database error: {err}") if __name__ == '__main__':