Update app.py

app.py

@@ -293,6 +293,7 @@ def split_into_chunks(texts, references, chunk_size, chunk_overlap):
   #  return vectorstore
 
 
+
 def setup_vectorstore(chunks, model_name):
     print("Start setup_vectorstore_function")
     
@@ -311,33 +312,26 @@ def setup_vectorstore(chunks, model_name):
         print("Persist directory:", vectorstore._persist_directory)
         print("Available methods in vectorstore:", dir(vectorstore))
         
-        # Get the first subfolder and zip its contents
-        for root, dirs, files in os.walk(temp_dir):
-            if dirs:  # Check for subfolders
-                next_subfolder = os.path.join(root, dirs[0])  # Take the first subfolder
-                zip_file_path = os.path.join(temp_dir, "subfolder_archive.zip")
-                
-                # Create a zip file
-                with zipfile.ZipFile(zip_file_path, "w", zipfile.ZIP_DEFLATED) as zipf:
-                    for folder_root, _, folder_files in os.walk(next_subfolder):
-                        for file_name in folder_files:
-                            file_path = os.path.join(folder_root, file_name)
-                            arcname = os.path.relpath(file_path, next_subfolder)  # Preserve relative paths
-                            zipf.write(file_path, arcname)
-                
-                print(f"Created zip file: {zip_file_path}")
-                
-                # Upload the zip file
-                target_path_in_repo = "subfolder_archive.zip"  # Define the target name in the repository
-                api.upload_file(
-                    path_or_fileobj=zip_file_path,
-                    path_in_repo=target_path_in_repo,
-                    repo_id=HF_SPACE_NAME,
-                    repo_type="space"
-                )
-                print(f"Uploaded {zip_file_path} as {target_path_in_repo}")
-                
-                break  # Stop after processing the first subfolder
+        # Zip the entire folder
+        zip_file_path = os.path.join(temp_dir, "folder_archive.zip")
+        with zipfile.ZipFile(zip_file_path, "w", zipfile.ZIP_DEFLATED) as zipf:
+            for folder_root, _, folder_files in os.walk(temp_dir):
+                for file_name in folder_files:
+                    file_path = os.path.join(folder_root, file_name)
+                    arcname = os.path.relpath(file_path, temp_dir)  # Preserve relative paths
+                    zipf.write(file_path, arcname)
+        
+        print(f"Created zip file: {zip_file_path}")
+        
+        # Upload the zip file
+        target_path_in_repo = "folder_archive.zip"  # Define the target name in the repository
+        api.upload_file(
+            path_or_fileobj=zip_file_path,
+            path_in_repo=target_path_in_repo,
+            repo_id=HF_SPACE_NAME,
+            repo_type="space"
+        )
+        print(f"Uploaded {zip_file_path} as {target_path_in_repo}")
         
         print("Process completed successfully!")