Update app.py

app.py

@@ -217,8 +217,11 @@ def create_db():
         files = request.files.getlist('folder')  # Folder uploads (multiple files)
         single_files = request.files.getlist('file')  # Single file uploads
 
-        # Check if any file is uploaded
-        if not files and not single_files:
+        print("==================folder==>", files)
+        print("==================single_files==>", single_files)
+
+        # Ensure at least one valid file is uploaded
+        if not any(file.filename.strip() for file in files) and not any(file.filename.strip() for file in single_files):
             return "No files uploaded", 400
 
         # Create upload directory
@@ -226,46 +229,36 @@ def create_db():
         print(f"Base Upload Path: {upload_base_path}")
         os.makedirs(upload_base_path, exist_ok=True)
 
-        # Process folder files (if any)
-        if files:
-            for file in files:
-                file_name = secure_filename(file.filename)  # Ensure the file name is safe
-                file_path = os.path.join(upload_base_path, file_name)
-
-                # Ensure the directory exists before saving the file
-                print(f"Saving to: {file_path}")
-                os.makedirs(os.path.dirname(file_path), exist_ok=True)
-
-                # Save the file
-                file.save(file_path)
-
-        # Process single files (if any)
-        if single_files:
+        # Process single file uploads first (if any exist)
+        if any(file.filename.strip() for file in single_files):
             for file in single_files:
-                if file.filename == '':
-                    print("Skipping empty single file")
-                    continue  # Skip empty uploads
-
-                # Create full file path for single file upload
-                file_name = secure_filename(file.filename)
-                file_path = os.path.join(upload_base_path, file_name)
-
-                # Ensure the directory exists before saving the file
-                print(f"Saving single file to: {file_path}")
-                os.makedirs(os.path.dirname(file_path), exist_ok=True)
+                if file.filename.strip():  # Ensure the file is valid
+                    file_name = secure_filename(file.filename)
+                    file_path = os.path.join(upload_base_path, file_name)
+                    print(f"Saving single file to: {file_path}")
+                    file.save(file_path)
+
+            # If single file is uploaded, skip folder processing
+            print("Single file uploaded, skipping folder processing.")
+            asyncio.run(generate_data_store(upload_base_path, db_name))
+            return redirect(url_for('list_dbs'))
 
-                # Save the file
-                file.save(file_path)
-                print("file------------->",file)
-                print("file_path------------->",file_path)
+        # Process folder files only if valid files exist
+        if any(file.filename.strip() for file in files):
+            for file in files:
+                if file.filename.strip():  # Ensure it's a valid file
+                    file_name = secure_filename(file.filename)
+                    file_path = os.path.join(upload_base_path, file_name)
+                    print(f"Saving folder file to: {file_path}")
+                    file.save(file_path)
 
-        # Generate datastore (example task, depending on your logic)
+        # Generate datastore
         asyncio.run(generate_data_store(upload_base_path, db_name))
-
         return redirect(url_for('list_dbs'))
 
     return render_template('create_db.html')
 
+
 @app.route('/list-dbs', methods=['GET'])
 def list_dbs():
     vector_dbs = [name for name in os.listdir(VECTOR_DB_FOLDER) if os.path.isdir(os.path.join(VECTOR_DB_FOLDER, name))]