Update app.py

app.py

@@ -307,32 +307,75 @@ async def process_file(file_path):
 import gradio as gr
 import base64, pathlib
 
-def embed_file(file, scale):
-    """根据后缀决定用 PDF 还是 <img>."""
+# def embed_file(file, scale):
+#     """根据后缀决定用 PDF 还是 <img>."""
+#     path = pathlib.Path(file.name)
+#     data = base64.b64encode(path.read_bytes()).decode()
+#     if path.suffix.lower() == ".pdf":
+#         b64 = base64.b64encode(pathlib.Path(file.name).read_bytes()).decode()
+#     # 注意：把 JS/HTML 写在一起返回给 gr.HTML
+#         return f"""
+#             <script>
+#             const pdf = atob("{b64}");
+#             const bytes = Uint8Array.from(pdf, c => c.charCodeAt(0));
+#             const blob = new Blob([bytes], {{type: "application/pdf"}});
+#             const url = URL.createObjectURL(blob);
+#             const iframe = document.getElementById("pdf_view") || document.createElement("iframe");
+#             iframe.id = "pdf_view";
+#             iframe.style = "width:100%;height:800px;border:none";
+#             iframe.src = url + "#zoom={int(scale*100)}";
+#             document.currentScript.parentElement.replaceChildren(iframe);
+#             </script>
+#         """
+#     else:
+#         # 图片：CSS transform 实现 zoom；支持拖拽滚轮缩放
+#         return (
+#             f'<div style="overflow:auto;border:1px solid #ccc;'
+#             f'width:100%;height:800px;text-align:center">'
+#             f'<img src="data:image/{path.suffix[1:]};base64,{data}" '
+#             f'style="transform:scale({scale});transform-origin:0 0;" />'
+#             f'</div>'
+#         )
+
+
+def embed_file(file, scale: float):
+    """
+    将上传的文件转换为可嵌入 HTML：
+    • PDF  →  生成 blob:URL，用 <iframe> + #zoom=xx% 展示
+    • 图片 →  <img style="transform:scale()"> 并允许拖拽缩放
+    """
+    if file is None:
+        return ""
+
     path = pathlib.Path(file.name)
+    ext  = path.suffix.lower()
     data = base64.b64encode(path.read_bytes()).decode()
-    if path.suffix.lower() == ".pdf":
-        b64 = base64.b64encode(pathlib.Path(file.name).read_bytes()).decode()
-    # 注意：把 JS/HTML 写在一起返回给 gr.HTML
+
+    # ---------- 1. PDF ----------
+    if ext == ".pdf":
+        # 输出 JS：把 Base64 转 Blob，然后注入/更新 iframe
         return f"""
-            <script>
-            const pdf = atob("{b64}");
-            const bytes = Uint8Array.from(pdf, c => c.charCodeAt(0));
-            const blob = new Blob([bytes], {{type: "application/pdf"}});
-            const url = URL.createObjectURL(blob);
-            const iframe = document.getElementById("pdf_view") || document.createElement("iframe");
-            iframe.id = "pdf_view";
-            iframe.style = "width:100%;height:800px;border:none";
-            iframe.src = url + "#zoom={int(scale*100)}";
-            document.currentScript.parentElement.replaceChildren(iframe);
-            </script>
-        """
+<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. 图片 ----------
     else:
-        # 图片：CSS transform 实现 zoom；支持拖拽滚轮缩放
         return (
             f'<div style="overflow:auto;border:1px solid #ccc;'
             f'width:100%;height:800px;text-align:center">'
-            f'<img src="data:image/{path.suffix[1:]};base64,{data}" '
+            f'<img src="data:image/{ext[1:]};base64,{data}" '
             f'style="transform:scale({scale});transform-origin:0 0;" />'
             f'</div>'
         )
@@ -360,7 +403,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()
+                out = gr.HTML(sanitize=False)
             
                 file.change(embed_file, [file, slider], out)
                 slider.change(embed_file, [file, slider], out)