1

app.py

@@ -1,8 +1,7 @@
 # === Загрузка библиотек ===
-
 from pypdf import PdfReader, PdfWriter
 import gradio as gr
-import fitz
+import fitz  # PyMuPDF
 from PIL import Image
 import pandas as pd
 import cv2
@@ -11,7 +10,6 @@ import os
 from transformers import TrOCRProcessor, VisionEncoderDecoderModel
 import torch
 import difflib
-from pdf2image import convert_from_path
 
 # Загрузка TrOCR
 processor = TrOCRProcessor.from_pretrained('kazars24/trocr-base-handwritten-ru')
@@ -143,7 +141,15 @@ def crop_and_recognize_cells(image, cells):
 
 # === 5. Полный процесс обработки изображения таблицы ===
 def process_pdf_table(pdf_path, output_excel='results.xlsx'):
-    images = convert_from_path(pdf_path, dpi=300)  # Загружаем первую страницу
+    # Извлечение изображений из PDF с помощью PyMuPDF
+    doc = fitz.open(pdf_path)
+    images = []
+    for page_num in range(len(doc)):
+        page = doc.load_page(page_num)
+        pix = page.get_pixmap(dpi=300)
+        img = Image.frombytes("RGB", [pix.width, pix.height], pix.samples)
+        images.append(img)
+
     if not images:
         print("Ошибка: PDF пустой или не удалось сконвертировать.")
         return
@@ -196,4 +202,4 @@ app = gr.Interface(
     allow_flagging="never"
 )
 
-app.launch(share=True)
+app.launch()