Update app.py

app.py

@@ -32,19 +32,13 @@ def nougat_ocr(file_name):
     subprocess.run(cli_command, stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True)
 
 def predict(pdf_file, pdf_link):
+    if pdf_file is None and pdf_link == '':
+        return "No data provided. Upload a pdf file or provide a pdf link and try again!", "", ""
+
     if pdf_file is None:
-        if pdf_link == '':
-            print("No file is uploaded and No link is provided")
-            return "No data provided. Upload a pdf file or provide a pdf link and try again!", "", ""
-        else:
-            print(f'pdf_link is - {pdf_link}')
-            file_name = get_pdf(pdf_link)
-            print(f'file_name is - {file_name}')
+        file_name = get_pdf(pdf_link)
     else:
         file_name = pdf_file.name
-        print(file_name)
-        pdf_name = pdf_file.name.split('/')[-1].split('.')[0]
-        print(pdf_name)
 
     # Call nougat
     nougat_ocr(file_name)
@@ -101,9 +95,12 @@ with gr.Blocks(css=css) as demo:
     markdown_hidden = gr.Textbox(elem_id='markdown-content', visible=False)
     download_link = gr.File(elem_id='download-link', visible=False)
 
-    btn.click(predict, [pdf_file, pdf_link], [parsed_output, markdown_hidden, download_link], _js=js_auto_copy)
+    btn.click(fn=predict, inputs=[pdf_file, pdf_link], outputs=[parsed_output, markdown_hidden, download_link], _js=js_auto_copy)
     clr.click(lambda: (gr.update(value=None), gr.update(value=None), gr.update(value=None)), [], [pdf_file, pdf_link, parsed_output, markdown_hidden, download_link])
     dwn.click(None, [download_link], None)
 
+# Enable queueing for request handling
 demo.queue()
+
+# Launch the interface in debug mode
 demo.launch(debug=True)