@@ -455,10 +455,9 @@ def merge_chunks_local(file_name, total_chunks, chunk_dir, merged_dir):
455
455
shutil .copyfileobj (chunk_file , write_stream )
456
456
os .unlink (chunk_file_path ) # Delete the individual chunk file after merging
457
457
logging .info ("Chunks merged successfully and return file size" )
458
- file_name , pages , file_extension = get_documents_from_file_by_path (merged_file_path ,file_name )
459
- pdf_total_pages = pages [0 ].metadata ['total_pages' ]
458
+
460
459
file_size = os .path .getsize (merged_file_path )
461
- return pdf_total_pages , file_size
460
+ return file_size
462
461
463
462
464
463
@@ -484,9 +483,8 @@ def upload_file(graph, model, chunk, chunk_number:int, total_chunks:int, origina
484
483
# If this is the last chunk, merge all chunks into a single file
485
484
if gcs_file_cache == 'True' :
486
485
file_size = merge_file_gcs (BUCKET_UPLOAD , originalname , folder_name , int (total_chunks ))
487
- total_pages = 1
488
486
else :
489
- total_pages , file_size = merge_chunks_local (originalname , int (total_chunks ), chunk_dir , merged_dir )
487
+ file_size = merge_chunks_local (originalname , int (total_chunks ), chunk_dir , merged_dir )
490
488
491
489
logging .info ("File merged successfully" )
492
490
file_extension = originalname .split ('.' )[- 1 ]
@@ -496,12 +494,11 @@ def upload_file(graph, model, chunk, chunk_number:int, total_chunks:int, origina
496
494
obj_source_node .file_size = file_size
497
495
obj_source_node .file_source = 'local file'
498
496
obj_source_node .model = model
499
- obj_source_node .total_pages = total_pages
500
497
obj_source_node .created_at = datetime .now ()
501
498
graphDb_data_Access = graphDBdataAccess (graph )
502
499
503
500
graphDb_data_Access .create_source_node (obj_source_node )
504
- return {'file_size' : file_size , 'total_pages' : total_pages , ' file_name' : originalname , 'file_extension' :file_extension , 'message' :f"Chunk { chunk_number } /{ total_chunks } saved" }
501
+ return {'file_size' : file_size , 'file_name' : originalname , 'file_extension' :file_extension , 'message' :f"Chunk { chunk_number } /{ total_chunks } saved" }
505
502
return f"Chunk { chunk_number } /{ total_chunks } saved"
506
503
507
504
def get_labels_and_relationtypes (graph ):
0 commit comments