fix: process pdf once

app.py

@@ -45,12 +45,10 @@ def ocr_process(file, got_mode, ocr_color="", ocr_box="", progress=gr.Progress()
         progress(0, desc="开始处理...")
 
         with tempfile.TemporaryDirectory() as temp_dir:
-            if file.name.lower().endswith(".pdf"):
-                pdf_path = os.path.join(temp_dir, "temp.pdf")
-                with open(pdf_path, "wb") as f:
-                    f.write(file.read())
+            file_path = file.name  # 使用文件的原始路径
 
-                images = pdf_to_images(pdf_path)
+            if file_path.lower().endswith(".pdf"):
+                images = pdf_to_images(file_path)
                 num_pages = len(images)
                 results = []
 
@@ -64,10 +62,7 @@ def ocr_process(file, got_mode, ocr_color="", ocr_box="", progress=gr.Progress()
 
                 final_result = "\n\n".join(results)
             else:
-                img_path = os.path.join(temp_dir, "temp_image.png")
-                with open(img_path, "wb") as f:
-                    f.write(file.read())
-                final_result = process_single_image(img_path, got_mode, ocr_color, ocr_box)
+                final_result = process_single_image(file_path, got_mode, ocr_color, ocr_box)
 
         progress(1, desc="处理完成")
         return final_result