change app add shutil for lance

app.py

@@ -26,48 +26,45 @@ import shutil
 def download_and_extract_gdrive(file_id, destination, extract_to):
     # Download the zip file
     gdown.download(f"https://drive.google.com/uc?id={file_id}", destination, quiet=False)
-    
-    # Use a temporary directory to avoid conflicts
+
+    # Use a temporary directory for extraction
     temp_dir = "./temp_extract"
     os.makedirs(temp_dir, exist_ok=True)
-    
+
     with zipfile.ZipFile(destination, 'r') as zip_ref:
         zip_ref.extractall(temp_dir)
 
-    # Ensure the main directory exists
+    # Ensure files are moved correctly
     if not os.path.exists(extract_to):
         os.makedirs(extract_to, exist_ok=True)
 
-    # Move extracted files into the target directory
     for item in os.listdir(temp_dir):
         item_path = os.path.join(temp_dir, item)
-        if os.path.isdir(item_path):
-            shutil.move(item_path, extract_to)
-        elif os.path.isfile(item_path):
-            shutil.move(item_path, os.path.join(extract_to, os.path.basename(item_path)))
-    
-    # Clean up temporary files
+        shutil.move(item_path, os.path.join(extract_to, item))
+
+    # Cleanup
     shutil.rmtree(temp_dir)
     os.remove(destination)
 
-# Download LanceDB directory
-st.info("Downloading and setting up LanceDB directory. This may take some time...")
+# Download LanceDB files
+st.info("Downloading and setting up LanceDB files. This may take a while...")
+
 download_and_extract_gdrive(
     file_id="your_lancedb_file_id",  # Replace with actual Google Drive file ID
     destination="lancedb_directory_main.zip",
     extract_to="./lancedb_directory_main"
 )
 
-# Ensure LanceDB structure is correct
-expected_directories = [
+# Validate extracted files
+expected_files = [
     "enhanced_papers_pretrained_1.lance",
     "enhanced_papers_pretrained_2.lance",
     "enhanced_papers_finetuned.lance"
 ]
-for subdir in expected_directories:
-    subdir_path = os.path.join("./lancedb_directory_main", subdir)
-    if not os.path.isdir(subdir_path):
-        raise FileNotFoundError(f"Missing expected directory: {subdir_path}")
+for file in expected_files:
+    file_path = os.path.join("./lancedb_directory_main", file)
+    if not os.path.isfile(file_path):
+        raise FileNotFoundError(f"Expected file is missing: {file_path}")
 
 # # --------------------------- Load the LanceDB Table and Models --------------------------- #