Update app.py

app.py

@@ -19,6 +19,9 @@ os.system('cp magic-pdf.template.json ~/magic-pdf.json')
 os.system(f"sed -i 's|/tmp/models|{model_dir}/models|g' /home/user/magic-pdf.json")
 # os.system("sed -i 's|cpu|cuda|g' /home/user/magic-pdf.json")
 
+os.system("pip install gradio-pdf")
+from gradio_pdf import PDF
+
 import gradio as gr
 from loguru import logger
 
@@ -119,12 +122,16 @@ def to_markdown(file_path, end_pages):
     return md_content, txt_content, archive_zip_path, show_pdf(new_pdf_path)
 
 
-def show_pdf(file_path):
-    with open(file_path, "rb") as f:
-        base64_pdf = base64.b64encode(f.read()).decode('utf-8')
-    pdf_display = f'<embed src="data:application/pdf;base64,{base64_pdf}" ' \
-                  f'width="100%" height="1000" type="application/pdf">'
-    return pdf_display
+# def show_pdf(file_path):
+#     with open(file_path, "rb") as f:
+#         base64_pdf = base64.b64encode(f.read()).decode('utf-8')
+#     pdf_display = f'<embed src="data:application/pdf;base64,{base64_pdf}" ' \
+#                   f'width="100%" height="1000" type="application/pdf">'
+#     return pdf_display
+
+
+def show_pdf(file):
+    return file
 
 
 latex_delimiters = [{"left": "$$", "right": "$$", "display": True},
@@ -140,7 +147,8 @@ if __name__ == "__main__":
                     change_bu = gr.Button("Convert")
                     clear_bu = gr.ClearButton([file, max_pages], value="Clear")
                 gr.Markdown(value="### PDF preview")
-                pdf_show = gr.HTML(label="PDF preview")
+                # pdf_show = gr.HTML(label="PDF preview")
+                pdf_show = PDF(label="PDF preview", interactive=True, height=1000)
 
             with gr.Column(variant='panel', scale=5):
                 output_file = gr.File(label="convert result", interactive=False)