Update app.py

app.py

@@ -150,17 +150,28 @@ def initialize_database(list_file_obj, chunk_size, chunk_overlap, vector_db, url
     # Create collection_name for vector database
     progress(0.1, desc="Creating collection name...")
     collection_name = Path(list_file_path[0]).stem
+    
     # Fix potential issues from naming convention
-    ## Remove space
-    collection_name = collection_name.replace(" ","-") 
-    ## Limit lenght to 50 characters
+    ## Remove spaces
+    collection_name = collection_name.replace(" ", "-")
+    
+    ## Ensure it meets the minimum length (3 characters)
+    if len(collection_name) < 3:
+        collection_name += "-XX"  # Append extra characters if too short
+    
+    ## Limit the length to 50 characters
     collection_name = collection_name[:50]
-    ## Enforce start and end as alphanumeric character
+    
+    ## Enforce that it starts with an alphanumeric character
     if not collection_name[0].isalnum():
-        collection_name[0] = 'A'
+        collection_name = 'A' + collection_name[1:]
+    
+    ## Enforce that it ends with an alphanumeric character
     if not collection_name[-1].isalnum():
-        collection_name[-1] = 'Z'
-    # print('list_file_path: ', list_file_path)
+        collection_name = collection_name[:-1] + 'Z'
+    
+    # Print the collection name for verification
+    print('Collection name:', collection_name)
     print('Collection name: ', collection_name)
     progress(0.25, desc="Loading document...")
     # Load document and create splits