Skip to content

Commit 9b87543

Browse files
praveshkumar1988kartikpersistent
authored andcommitted
Remove TotalPages when save file on local (#684)
1 parent bf2c29c commit 9b87543

File tree

1 file changed

+4
-7
lines changed

1 file changed

+4
-7
lines changed

backend/src/main.py

+4-7
Original file line numberDiff line numberDiff line change
@@ -455,10 +455,9 @@ def merge_chunks_local(file_name, total_chunks, chunk_dir, merged_dir):
455455
shutil.copyfileobj(chunk_file, write_stream)
456456
os.unlink(chunk_file_path) # Delete the individual chunk file after merging
457457
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+
460459
file_size = os.path.getsize(merged_file_path)
461-
return pdf_total_pages,file_size
460+
return file_size
462461

463462

464463

@@ -484,9 +483,8 @@ def upload_file(graph, model, chunk, chunk_number:int, total_chunks:int, origina
484483
# If this is the last chunk, merge all chunks into a single file
485484
if gcs_file_cache == 'True':
486485
file_size = merge_file_gcs(BUCKET_UPLOAD, originalname, folder_name, int(total_chunks))
487-
total_pages = 1
488486
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)
490488

491489
logging.info("File merged successfully")
492490
file_extension = originalname.split('.')[-1]
@@ -496,12 +494,11 @@ def upload_file(graph, model, chunk, chunk_number:int, total_chunks:int, origina
496494
obj_source_node.file_size = file_size
497495
obj_source_node.file_source = 'local file'
498496
obj_source_node.model = model
499-
obj_source_node.total_pages = total_pages
500497
obj_source_node.created_at = datetime.now()
501498
graphDb_data_Access = graphDBdataAccess(graph)
502499

503500
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"}
505502
return f"Chunk {chunk_number}/{total_chunks} saved"
506503

507504
def get_labels_and_relationtypes(graph):

0 commit comments

Comments
 (0)