Update app.py

app.py

@@ -39,7 +39,7 @@ except Exception as e:
 from gradio_pdf import PDF
 import gradio as gr
 
-from mineru.cli.common import prepare_env, read_fn, aio_do_parse
+from mineru.cli.common import prepare_env, read_fn, aio_do_parse, pdf_suffixes, image_suffixes
 from mineru.utils.hash_utils import str_sha256
 
 os.environ['MINERU_MODEL_SOURCE'] = 'local'
@@ -149,8 +149,8 @@ latex_delimiters = [
 ]
 
 
-with open("header.html", "r") as file:
-    header = file.read()
+with open("header.html", "r") as header_file:
+    header = header_file.read()
 
 
 latin_lang = [
@@ -227,7 +227,8 @@ def main():
         with gr.Row():
             with gr.Column(variant='panel', scale=5):
                 with gr.Row():
-                    file = gr.File(label='Please upload a PDF or image', file_types=['.pdf', '.png', '.jpeg', '.jpg'])
+                    suffixes = pdf_suffixes + image_suffixes
+                    input_file = gr.File(label='Please upload a PDF or image', file_types=suffixes)
                 with gr.Row():
                     max_pages = gr.Slider(1, 20, 10, step=1, label='Max convert pages')
                 with gr.Row():
@@ -251,7 +252,7 @@ def main():
                             gr.Examples(
                                 examples=[os.path.join(example_root, _) for _ in os.listdir(example_root) if
                                           _.endswith('pdf')],
-                                inputs=file
+                                inputs=input_file
                             )
 
             with gr.Column(variant='panel', scale=5):
@@ -285,10 +286,10 @@ def main():
             api_name=False
         )
 
-        file.change(fn=to_pdf, inputs=file, outputs=pdf_show, api_name=False)
-        change_bu.click(fn=to_markdown, inputs=[file, max_pages, is_ocr, formula_enable, table_enable, language, backend, url],
+        input_file.change(fn=to_pdf, inputs=input_file, outputs=pdf_show, api_name=False)
+        change_bu.click(fn=to_markdown, inputs=[input_file, max_pages, is_ocr, formula_enable, table_enable, language, backend, url],
                         outputs=[md, md_text, output_file, pdf_show], api_name=False)
-        clear_bu.add([file, md, pdf_show, md_text, output_file, is_ocr])
+        clear_bu.add([input_file, md, pdf_show, md_text, output_file, is_ocr])
 
     demo.launch(ssr_mode=True)