fix: dep

pdf2zh/gui.py

@@ -20,8 +20,6 @@
 
 import gradio as gr
 from gradio_pdf import PDF
-import numpy as np
-import pymupdf
 import tqdm
 import requests
 import cgi
@@ -79,19 +77,6 @@ def verify_recaptcha(response):
     return result.get("success")
 
 
-def pdf_preview(file):
-    doc = pymupdf.open(file)
-    page = doc[0]
-    pix = page.get_pixmap()
-    image = np.frombuffer(pix.samples, np.uint8).reshape(pix.height, pix.width, 3)
-    return image
-
-
-def upload_file(file, service, progress=gr.Progress()):
-    preview_image = pdf_preview(file)
-    return file, file
-
-
 def download_with_limit(url, save_path, size_limit):
     chunk_size = 1024
     total_size = 0
@@ -181,11 +166,6 @@ def progress_bar(t: tqdm.tqdm):
     if not file_mono.exists() or not file_dual.exists():
         raise gr.Error("No output")
 
-    try:
-        translated_preview = pdf_preview(str(file_mono))
-    except Exception:
-        raise gr.Error("No preview")
-
     progress(1.0, desc="Translation complete!")
 
     return (
@@ -335,7 +315,9 @@ def on_select_filetype(file_type):
                 )
 
             output_title = gr.Markdown("## Translated", visible=False)
-            output_file = gr.File(label="Download Translation", visible=False)
+            output_file_mono = gr.File(
+                label="Download Translation (Mono)", visible=False
+            )
             output_file_dual = gr.File(
                 label="Download Translation (Dual)", visible=False
             )
@@ -385,9 +367,9 @@ def on_select_filetype(file_type):
 
     # Event handlers
     file_input.upload(
-        upload_file,
-        inputs=[file_input, service],
-        outputs=[file_input, preview],
+        lambda x: x,
+        inputs=file_input,
+        outputs=preview,
         js=(
             f"""
             (a,b)=>{{
@@ -419,10 +401,10 @@ def on_select_filetype(file_type):
             *envs,
         ],
         outputs=[
-            output_file,
+            output_file_mono,
             preview,
             output_file_dual,
-            output_file,
+            output_file_mono,
             output_file_dual,
             output_title,
         ],

pyproject.toml

@@ -27,6 +27,7 @@ dependencies = [
     "opencv-python-headless",
     "tencentcloud-sdk-python",
     "pdfminer.six>=20240706",
+    "gradio_pdf",
 ]
 
 [project.optional-dependencies]