Update app.py

app.py

@@ -25,9 +25,20 @@ def run_ocr(pdf_file, langs: str, rotate_pages: bool, deskew: bool, clean: bool,
         st.error("系统未检测到 ocrmypdf，可检查 Docker/依赖安装。")
         return None, None, None
 
+    # 修复：重置文件指针到开头，然后读取内容
+    try:
+        pdf_file.seek(0)  # 重置文件指针
+        pdf_content = pdf_file.read()
+        if not pdf_content:
+            st.error("PDF 文件内容为空。")
+            return None, None, None
+    except Exception as e:
+        st.error(f"读取 PDF 文件失败：{e}")
+        return None, None, None
+
     # 保存上传文件到临时路径
     with tempfile.NamedTemporaryFile(delete=False, suffix=".pdf") as tmp:
-        tmp.write(pdf_file.read())
+        tmp.write(pdf_content)  # 使用读取到的内容
         in_path = Path(tmp.name)
 
     work = Path(tempfile.mkdtemp(prefix="ocr_"))
@@ -91,7 +102,7 @@ def run_extract(sidecar_text: str, provider: str, model_id: str, prompt: str):
     # 最小 few-shot 示例（演示用；生产请替换为你的法律 schema）
     examples = [
         lx.data.ExampleData(
-            text="On 12/03/2023, ABC Pty Ltd terminated Ms Wang’s employment.",
+            text="On 12/03/2023, ABC Pty Ltd terminated Ms Wang's employment.",
             extractions=[
                 lx.data.Extraction(
                     extraction_class="party",
@@ -143,7 +154,7 @@ def run_extract(sidecar_text: str, provider: str, model_id: str, prompt: str):
 with st.sidebar:
     st.header("参数")
 
-    # 用 form 把“上传 + 参数 + 提交”打包，避免按钮重跑导致 file_uploader 丢值
+    # 用 form 把"上传 + 参数 + 提交"打包，避免按钮重跑导致 file_uploader 丢值
     with st.form("run_form", clear_on_submit=False):
         pdf = st.file_uploader("上传扫描 PDF", type=["pdf"], accept_multiple_files=False, key="pdf_uploader")
 
@@ -194,6 +205,12 @@ if "pdf_uploader" in st.session_state and st.session_state["pdf_uploader"]:
                        f"({st.session_state['pdf_uploader'].size/1024:.1f} KB)")
 
 if submitted:
+    # 添加调试信息
+    if pdf is None:
+        st.error("PDF 为 None - 检查文件上传")
+    else:
+        st.info(f"PDF 文件信息：名称={pdf.name}, 大小={pdf.size} bytes")
+    
     out_pdf, sidecar_path, preview = run_ocr(
         pdf, langs, rotate_pages, deskew, clean, optimize_level,
         force_ocr, skip_text, export_sidecar
@@ -215,4 +232,4 @@ if submitted:
             st.components.v1.html(html_content, height=650, scrolling=True)
         if jsonl_path and Path(jsonl_path).exists():
             with open(jsonl_path, "rb") as f:
-                jsonl_slot.download_button("下载抽取结果 JSONL", f, file_name="extractions.jsonl")
+                jsonl_slot.download_button("下载抽取结果 JSONL", f, file_name="extractions.jsonl")