Update pdf2zh/gui.py

pdf2zh/gui.py

@@ -49,6 +49,25 @@ if os.environ.get("PDF2ZH_DEMO"):
         "First 20 pages": list(range(0, 20)),
     }
 
+def download_with_limit(url, save_path, size_limit):
+    chunk_size = 1024
+    total_size = 0
+    with requests.get(url, stream=True, timeout=10) as response:
+        response.raise_for_status()
+        content = response.headers.get("Content-Disposition")
+        try:
+            _, params = cgi.parse_header(content)
+            filename = params["filename"]
+        except Exception:
+            filename = os.path.basename(url)
+        with open(save_path / filename, "wb") as file:
+            for chunk in response.iter_content(chunk_size=chunk_size):
+                total_size += len(chunk)
+                if size_limit and total_size > size_limit:
+                    raise gr.Error("Exceeds file size limit")
+                file.write(chunk)
+    return save_path / filename
+
 def pdf_preview(file):
     doc = pymupdf.open(file)
     page = doc[0]
@@ -70,25 +89,6 @@ def upload_file(file, service, progress=gr.Progress()):
         print(f"Error converting PDF: {e}")
         return None, None
 
-def download_with_limit(url, save_path, size_limit):
-    chunk_size = 1024
-    total_size = 0
-    with requests.get(url, stream=True, timeout=10) as response:
-        response.raise_for_status()
-        content = response.headers.get("Content-Disposition")
-        try:
-            _, params = cgi.parse_header(content)
-            filename = params["filename"]
-        except Exception:
-            filename = os.path.basename(url)
-        with open(save_path / filename, "wb") as file:
-            for chunk in response.iter_content(chunk_size=chunk_size):
-                total_size += len(chunk)
-                if size_limit and total_size > size_limit:
-                    raise gr.Error("Exceeds file size limit")
-                file.write(chunk)
-    return save_path / filename
-
 def translate(
     file_type,
     file_input,