Create database.py

database.py

@@ -0,0 +1,16 @@
+import lancedb
+
+def get_db():
+    db = lancedb.connect("lancedb")
+    return db
+
+def insert_video_data(video_data):
+    db = get_db()
+    table = db.create_table("videos", video_data)
+    table.add(video_data)
+
+def search_similar_videos(query, limit=5):
+    db = get_db()
+    table = db.open_table("videos")
+    results = table.search(query).limit(limit).to_list()
+    return results