From 6597cb133a81569febb5c29ebdf4d0f0bee67e74 Mon Sep 17 00:00:00 2001 From: "ishak jmilou.ishak" Date: Tue, 21 Jan 2025 14:10:31 +0100 Subject: [PATCH] checking for error in db input --- src/Python/flask/web/app.py | 23 ++++++++++++++--------- 1 file changed, 14 insertions(+), 9 deletions(-) diff --git a/src/Python/flask/web/app.py b/src/Python/flask/web/app.py index 6f89d0d..7a73f3d 100644 --- a/src/Python/flask/web/app.py +++ b/src/Python/flask/web/app.py @@ -162,16 +162,21 @@ def image(): def yolo_results_endpoint(): global yolo_results with lock: - db = get_db() - with db.cursor() as cursor: - sql_yolo = "INSERT INTO image (class, confidence) VALUES (%s, %s)" - yolo_tuples = [(result["class"], result["confidence"]) for result in yolo_results] - print(f"YOLO Tuples: {yolo_tuples}") # Debug statement - cursor.executemany(sql_yolo, yolo_tuples) - db.commit() - cursor.close() + try: + db = get_db() + with db.cursor() as cursor: + sql_yolo = "INSERT INTO image (class, confidence) VALUES (%s, %s)" + yolo_tuples = [(result["class"], result["confidence"]) for result in yolo_results] + print(f"YOLO Tuples: {yolo_tuples}") # Debug statement + cursor.executemany(sql_yolo, yolo_tuples) + db.commit() + cursor.close() + except mysql.connector.Error as err: + print(f"Database error: {err}") + except Exception as e: + print(f"Unexpected error: {e}") + return jsonify(yolo_results) - if __name__ == '__main__': app.run(debug=True, port=5000) \ No newline at end of file