Update rag_routerv2.py

rag_routerv2.py

@@ -193,6 +193,40 @@ async def get_tables(user_id: str):
    
    return result
 
+
+@router.delete("/delete_table/{table_id}")
+async def delete_table(table_id: str, user_id: str):
+   try:
+       db = get_db()
+       
+       # Verify user owns the table
+       table = db.execute(
+           'SELECT * FROM tables WHERE table_id = ? AND user_id = ?',
+           (table_id, user_id)
+       ).fetchone()
+       
+       if not table:
+           raise HTTPException(status_code=404, detail="Table not found or unauthorized")
+
+       # Delete files from filesystem
+       table_path = f"./data/{table_id}"
+       index_path = f"./lancedb/index/{table_id}"
+       if os.path.exists(table_path):
+           shutil.rmtree(table_path)
+       if os.path.exists(index_path):
+           shutil.rmtree(index_path)
+
+       # Delete from database
+       db.execute('DELETE FROM table_files WHERE table_id = ?', (table_id,))
+       db.execute('DELETE FROM tables WHERE table_id = ?', (table_id,))
+       db.commit()
+
+       return {"message": "Table deleted successfully"}
+
+   except Exception as e:
+       raise HTTPException(status_code=500, detail=str(e))
+
+
 @router.get("/health")
 async def health_check():
     return {"status": "healthy"}