From 7f526e2af3984434ce6bc9810d606dbc32b30476 Mon Sep 17 00:00:00 2001
From: chaoke <101492509+8baby8@users.noreply.github.com>
Date: Thu, 29 Feb 2024 11:16:45 +0800
Subject: [PATCH] Add files via upload

---
 scripts/ptf2txt.py | 69 ++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 69 insertions(+)
 create mode 100644 scripts/ptf2txt.py

diff --git a/scripts/ptf2txt.py b/scripts/ptf2txt.py
new file mode 100644
index 0000000..8569033
--- /dev/null
+++ b/scripts/ptf2txt.py
@@ -0,0 +1,69 @@
+import os 
+import sys  
+import glob  
+try :
+    import  cv2 
+except:
+    os.system('pip install opencv-python')
+try :
+    from  paddleocr  import  PaddleOCR ,  draw_ocr ,  download_with_progressbar 
+except:
+    os.system('pip install paddleocr')
+output_folder_path = 'res/'
+if not os.path.exists(output_folder_path):  
+    os.makedirs(output_folder_path) 
+
+def get_pdf_files_in_directory(directory_path):  
+    # 确保路径存在  
+    if os.path.exists(directory_path) and os.path.isdir(directory_path):  
+        # 使用glob模块搜索所有PDF文件  
+        pdf_files = glob.glob(os.path.join(directory_path, '**', '*.pdf'), recursive=True)  
+        return pdf_files  
+    else:  
+        return []  
+def ocr_pdf_folder(folder_path):
+    ocr  =  PaddleOCR ( use_angle_cls = True ,  lang = "ch" , page_num = 0 )   # 只需运行一次即可将模型下载并加载到内存中
+    print("ppocrv4 加载完毕！！！")
+    pdf_paths = get_pdf_files_in_directory(folder_path)
+    print(f"共检测到 {len(pdf_paths)} 个PDF文件")
+    # 打印所有PDF文件的路径  
+    for pdf_path in pdf_paths:  
+        print(f'正在处理文件：{pdf_path}')
+        
+        result  =  ocr . ocr ( pdf_path ,  cls = True ) 
+        for  idx  in  range ( len ( result )): 
+            res  =  result [ idx ] 
+            for  line  in  res : 
+                print ( line )  
+        print(f'{pdf_path} 处理完毕')
+        ocr_result = ""
+        for idx in range(len(result)):
+            res = result[idx]
+            for line in res:
+        #         print(line[1][0])
+                ocr_result = ocr_result + " " + str(line[1][0])
+        
+        filename = os.path.splitext(os.path.basename(pdf_path))[0]  
+      
+        # 构建TXT文件的完整路径  
+        txt_path = os.path.join('res/', filename + '.txt')  
+        
+        # 将提取的文本写入TXT文件  
+        with open(txt_path, 'w', encoding='utf-8') as txt_file:  
+            txt_file.write(ocr_result)  
+
+        print(f'生成的txt文档保存在{txt_path}')
+        #         break
+        # print(ocr_result)
+        # with open('my_file.txt', 'a') as f:  
+        #     # 写入字符串  
+        #     f.write(ocr_result)
+
+
+if __name__ == "__main__":  
+    if len(sys.argv) > 1:  
+        # sys.argv[0] 是脚本名，sys.argv[1:] 是传递给脚本的参数列表  
+        pdf_path = sys.argv[1]  
+        print(f'需要处理的文件夹是：{pdf_path}')
+        ocr_pdf_folder(pdf_path)
+    
\ No newline at end of file