Update app.py

app.py

@@ -354,23 +354,27 @@ def embed_file(file, scale: float):
     # ---------- 1. PDF ----------
     if ext == ".pdf":
         # 输出 JS：把 Base64 转 Blob，然后注入/更新 iframe
+        pdf_b64 = base64.b64encode(pathlib.Path(file.name).read_bytes()).decode()
         return f"""
-<script>
-const bytes = Uint8Array.from(atob("{data}"), c => c.charCodeAt(0));
-const blob  = new Blob([bytes], {{ type: "application/pdf" }});
-const url   = URL.createObjectURL(blob);
-
-let iframe = document.getElementById("pdf_view");
-if (!iframe) {{
-  iframe = document.createElement("iframe");
-  iframe.id    = "pdf_view";
-  iframe.style = "width:100%;height:800px;border:none;";
-  document.currentScript.parentElement.appendChild(iframe);
-}}
-iframe.src = url + "#zoom={int(scale*100)}";
-</script>
-"""
-    # ---------- 2. 图片 ----------
+            <script>
+            // -------- 把 base64 转为二进制 Blob ----------
+            const bytes = Uint8Array.from(
+                atob("{pdf_b64}"), ch => ch.charCodeAt(0)
+            );
+            const blobUrl = URL.createObjectURL(
+                new Blob([bytes], {{type: "application/pdf"}})
+            );
+            // -------- 创建或更新 iframe ----------
+            let frame = document.getElementById("pdf_view");
+            if (!frame) {{
+              frame = document.createElement("iframe");
+              frame.id = "pdf_view";
+              frame.style = "width:100%;height:800px;border:none;";
+              document.currentScript.parentElement.appendChild(frame);
+            }}
+            frame.src = blobUrl + "#zoom={int(scale*100)}";
+            </script>
+            """
     else:
         return (
             f'<div style="overflow:auto;border:1px solid #ccc;'
@@ -403,7 +407,7 @@ if __name__ == '__main__':
                 pdf_show = PDF(label='Preview', interactive=False, visible=True, height=800)
 
                 slider = gr.Slider(0.5, 3.0, value=1.0, step=0.1, label="缩放倍数")
-                out = gr.HTML(sanitize=False)
+                out = gr.HTML()
             
                 file.change(embed_file, [file, slider], out)
                 slider.change(embed_file, [file, slider], out)