filename

main.py

@@ -53,6 +53,7 @@ class UserInput(BaseModel):
 class EnterpriseData(BaseModel):
     name: str
     id: Optional[str] = None
+    filename: Optional[str] = None
 
 
 
@@ -70,6 +71,11 @@ async def upload_file(file: UploadFile, enterprise_data: Json[EnterpriseData]):
 
         enterprise_name = enterprise_data.name.replace(" ","_").replace("-","_").replace(".","_").replace("/","_").replace("\\","_").strip()
 
+        if enterprise_data.filename is not None:
+            filename = enterprise_data.filename
+        else:
+            filename = file.filename
+
         # Assign a new UUID if id is not provided
         if enterprise_data.id is None:
             clean_name = remove_non_standard_ascii(enterprise_name)
@@ -87,11 +93,11 @@ async def upload_file(file: UploadFile, enterprise_data: Json[EnterpriseData]):
         text_chunks = get_text_chunks(text)
 
         # Create a vector store
-        vector_store = get_vectorstore(text_chunks, filename=file.filename, file_type="pdf", namespace=enterprise_data.id, index=index,enterprise_name=enterprise_name)
+        vector_store = get_vectorstore(text_chunks, filename=filename, file_type="pdf", namespace=enterprise_data.id, index=index,enterprise_name=enterprise_name)
 
         if vector_store:
             return {
-                "file_name":file.filename,
+                "file_name":filename,
                 "enterprise_id": enterprise_data.id,
                 "number_of_chunks": len(text_chunks),
                 "filename_id":vector_store["filename_id"],