Update app.py

app.py

@@ -280,7 +280,7 @@ def to_file(image_path):
 
 
 
-def files_to_b64(files, pdf_dpi: int = 200):
+def files_to_b64(file, pdf_dpi: int = 200):
     """
     把上传文件统一转成 base64 图片列表。
     • 普通图片：直接 base64
@@ -291,44 +291,42 @@ def files_to_b64(files, pdf_dpi: int = 200):
     List[str]  每个元素形如  "data:image/png;base64,xxxx..."
     """
     out: list[str] = []
-
-    for f in files:
-        # ---------- Gradio 4.x NamedString / FileData ----------
-        if hasattr(f, "data"):             # 新版 FileData / NamedString
-            raw_bytes = f.data
-            suffix    = pathlib.Path(f.name).suffix.lower()
-
-            # -- PDF --
-            if suffix == ".pdf":
-                pages = convert_from_bytes(raw_bytes, dpi=pdf_dpi)
-                for page in pages:
-                    buf = BytesIO()
-                    page.save(buf, format="PNG")
-                    b64 = base64.b64encode(buf.getvalue()).decode()
-                    out.append(f"data:image/png;base64,{b64}")
-                continue  # 进入下一文件
-
-            # -- 普通图片 --
-            b64 = base64.b64encode(raw_bytes).decode()
-            out.append(f"data:image/{suffix[1:]};base64,{b64}")
-
-        # ---------- 旧版：直接获得文件路径 ----------
-        else:
-            path   = pathlib.Path(f)
-            suffix = path.suffix.lower()
-
-            if suffix == ".pdf":
-                pages = convert_from_path(str(path), dpi=pdf_dpi)
-                for page in pages:
-                    buf = BytesIO()
-                    page.save(buf, format="PNG")
-                    b64 = base64.b64encode(buf.getvalue()).decode()
-                    out.append(f"data:image/png;base64,{b64}")
-                continue
-
-            raw_bytes = path.read_bytes()
-            b64 = base64.b64encode(raw_bytes).decode()
-            out.append(f"data:image/{suffix[1:]};base64,{b64}")
+    # ---------- Gradio 4.x NamedString / FileData ----------
+    if hasattr(file, "data"):             # 新版 FileData / NamedString
+        raw_bytes = file.data
+        suffix    = pathlib.Path(file.name).suffix.lower()
+
+        # -- PDF --
+        if suffix == ".pdf":
+            pages = convert_from_bytes(raw_bytes, dpi=pdf_dpi)
+            for page in pages:
+                buf = BytesIO()
+                page.save(buf, format="PNG")
+                b64 = base64.b64encode(buf.getvalue()).decode()
+                out.append(f"data:image/png;base64,{b64}")
+            continue  # 进入下一文件
+
+        # -- 普通图片 --
+        b64 = base64.b64encode(raw_bytes).decode()
+        out.append(f"data:image/{suffix[1:]};base64,{b64}")
+
+    # ---------- 旧版：直接获得文件路径 ----------
+    else:
+        path   = pathlib.Path(file)
+        suffix = path.suffix.lower()
+
+        if suffix == ".pdf":
+            pages = convert_from_path(str(path), dpi=pdf_dpi)
+            for page in pages:
+                buf = BytesIO()
+                page.save(buf, format="PNG")
+                b64 = base64.b64encode(buf.getvalue()).decode()
+                out.append(f"data:image/png;base64,{b64}")
+            continue
+
+        raw_bytes = path.read_bytes()
+        b64 = base64.b64encode(raw_bytes).decode()
+        out.append(f"data:image/{suffix[1:]};base64,{b64}")
 
     return out
 
@@ -368,61 +366,14 @@ if __name__ == '__main__':
                 with gr.Row():
                     change_bu = gr.Button('Parse')
                     clear_bu = gr.ClearButton(value='Clear')
-                pdf_show = PDF(label='Preview', interactive=False, visible=True, height=800)
-
-
 
                 with gr.Row():
-                    prev_btn = gr.Button("⬅️ 上一张")
-                    next_btn = gr.Button("下一张 ➡️")
-                    zoom = gr.Slider(0.5, 3, value=1, step=0.1, label="缩放倍数")
+                    prev_btn = gr.Button("⬅️ Pre")
+                    next_btn = gr.Button("Next ➡️")
+                    zoom = gr.Slider(0.5, 3, value=1, step=0.1, label="Scale")
             
-                # HTML 容器
                 viewer = gr.HTML()
 
-                # ---- 状态变量 ----
-                img_list_state = gr.State([])   # base64 列表
-                idx_state = gr.State(0)         # 当前索引
-            
-                # ---- 事件逻辑 ----
-                def upload_handler(files):
-                    b64s = files_to_b64(files)
-                    return b64s, 0, render_img(b64s, 0, 1)
-            
-                file.change(
-                    upload_handler,
-                    inputs=file,
-                    outputs=[img_list_state, idx_state, viewer],
-                )
-            
-                def show_prev(b64s, idx, scale):
-                    idx -= 1
-                    return idx, render_img(b64s, idx, scale)
-            
-                prev_btn.click(
-                    show_prev,
-                    inputs=[img_list_state, idx_state, zoom],
-                    outputs=[idx_state, viewer],
-                )
-            
-                def show_next(b64s, idx, scale):
-                    idx += 1
-                    return idx, render_img(b64s, idx, scale)
-            
-                next_btn.click(
-                    show_next,
-                    inputs=[img_list_state, idx_state, zoom],
-                    outputs=[idx_state, viewer],
-                )
-            
-                # 缩放滑块变化
-                zoom.change(
-                    lambda b64s, idx, scale: render_img(b64s, idx, scale),
-                    inputs=[img_list_state, idx_state, zoom],
-                    outputs=viewer,
-                )
-    
-         
                 example_root = os.path.join(os.path.dirname(__file__), 'examples')
                 images = [
                     os.path.join(example_root, f)
@@ -474,6 +425,47 @@ if __name__ == '__main__':
                
         # file.change(fn=process_file, inputs=file, outputs=pdf_show)
 
+        img_list_state = gr.State([])   # base64 列表
+        idx_state = gr.State(0)         # 当前索引
+     
+        def upload_handler(files):
+            b64s = files_to_b64(files)
+            return b64s, 0, render_img(b64s, 0, 1)
+    
+        file.change(
+            upload_handler,
+            inputs=file,
+            outputs=[img_list_state, idx_state, viewer],
+        )
+    
+        def show_prev(b64s, idx, scale):
+            idx -= 1
+            return idx, render_img(b64s, idx, scale)
+    
+        prev_btn.click(
+            show_prev,
+            inputs=[img_list_state, idx_state, zoom],
+            outputs=[idx_state, viewer],
+        )
+    
+        def show_next(b64s, idx, scale):
+            idx += 1
+            return idx, render_img(b64s, idx, scale)
+    
+        next_btn.click(
+            show_next,
+            inputs=[img_list_state, idx_state, zoom],
+            outputs=[idx_state, viewer],
+        )
+    
+        # 缩放滑块变化
+        zoom.change(
+            lambda b64s, idx, scale: render_img(b64s, idx, scale),
+            inputs=[img_list_state, idx_state, zoom],
+            outputs=viewer,
+        )
+    
+
 
         change_bu.click(
             fn=check_prompt,