changed colom name, addprint

This commit is contained in:
ishak jmilou.ishak
2025-01-15 15:46:39 +01:00
parent 1b3ccd1e72
commit fb12b20a0b

View File

@@ -163,10 +163,12 @@ def image():
def yolo_results_endpoint(): def yolo_results_endpoint():
global yolo_results global yolo_results
with lock: with lock:
print(f"YOLO Results: {yolo_results}") # Debug statement
db = get_db() db = get_db()
with db.cursor() as cursor: with db.cursor() as cursor:
sql_yolo = "INSERT INTO image (object, confidence) VALUES (%s, %s)" sql_yolo = "INSERT INTO image (class, confidence) VALUES (%s, %s)"
yolo_tuples = [(result["class"], result["confidence"]) for result in yolo_results] 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) cursor.executemany(sql_yolo, yolo_tuples)
db.commit() db.commit()
cursor.close() cursor.close()