checking for error in db input

This commit is contained in:
ishak jmilou.ishak
2025-01-21 14:10:31 +01:00
parent ec44cb955b
commit 6597cb133a

View File

@@ -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)