Update app.py

app.py

@@ -311,21 +311,36 @@ def setup_vectorstore(chunks, model_name):
         print("Persist directory:", vectorstore._persist_directory)
         print("Available methods in vectorstore:", dir(vectorstore))
         
-        # At this point, you can use your API upload method to upload the persisted vectorstore files
-        for root, _, files in os.walk(temp_dir):
-            for file_name in files:
-                file_path = os.path.join(root, file_name)
-                target_path_in_repo = os.path.relpath(file_path, temp_dir)
-                print(f"Uploading file: {file_path} -> {target_path_in_repo}")
+        # 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=file_path,
+                    path_or_fileobj=zip_file_path,
                     path_in_repo=target_path_in_repo,
                     repo_id=HF_SPACE_NAME,
                     repo_type="space"
                 )
-                print(f"Uploaded {file_path} to {target_path_in_repo}")
+                print(f"Uploaded {zip_file_path} as {target_path_in_repo}")
+                
+                break  # Stop after processing the first subfolder
         
-        print("All files uploaded successfully!")
+        print("Process completed successfully!")
+
 
 
 # Setup LLM