Update baigiang.py

baigiang.py

@@ -344,7 +344,7 @@ def process_omr(image_bytes, all_keys):
         traceback.print_exc()
         return {"score": 0, "error": str(e)}
 
-# --- DB HELPERS & API (GIỮ NGUYÊN) ---
+# --- DB HELPERS & API ---
 def save_result_to_db(data, filename, session_id):
     if not supabase: return
     try:
@@ -386,6 +386,26 @@ async def create_session(name: str = Form(...), answer_key_json: str = Form(...)
             logger.error(f"Create Session DB Error: {e}")
     return {"session_id": session_id, "name": name}
 
+@router.delete("/delete_session/{session_id}")
+async def delete_session(session_id: str):
+    if session_id in sessions: del sessions[session_id]
+    if supabase:
+        try:
+            supabase.table("omr_results").delete().eq("session_id", session_id).execute()
+            supabase.table("omr_sessions").delete().eq("session_id", session_id).execute()
+            return {"status": "deleted"}
+        except Exception as e: return {"error": str(e)}
+    return {"status": "local_deleted"}
+
+@router.put("/update_session/{session_id}")
+async def update_session(session_id: str, name: str = Form(...)):
+    if supabase:
+        try:
+            supabase.table("omr_sessions").update({"session_name": name}).eq("session_id", session_id).execute()
+            return {"status": "updated", "name": name}
+        except Exception as e: return {"error": str(e)}
+    return {"status": "local_updated"}
+
 @router.post("/grade")
 async def grade_exam(file: UploadFile = File(...), answer_key_json: str = Form(...)):
     try: all_keys = json.loads(answer_key_json)