Update app.py

app.py

@@ -1,37 +1,48 @@
 import gradio as gr
 import subprocess
 
-def fun(filepath):
-    print(f"filepath is - {filepath}")
-    # Command to run
-    bash_command = f"nougat {filepath}"
-    
-    # Run the command and capture its output
-    completed_process = subprocess.run(bash_command, shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True)
-    
-    # Get the output and error messages
-    output = completed_process.stdout
-    errors = completed_process.stderr
-    
-    # Print the output and errors
-    print("Output:")
-    print(len(output))
+
+def nougat_ocr(file_name):
+  # CLI Command to run
+  cli_command = [
+      'nougat',
+      '--out', '/output',
+      'pdf', f'{file_name}',
+      '--checkpoint', '/nougat'
+  ]
+
+  # Run the command and get .mmd file in an output folder
+  subprocess.run(cli_command, stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True)
+  return
+
+
+def predict(pdf_file):
+  print(f"temporary file - {pdf_file.name}")
+  pdf_name = pdf_file.name.split('/')[-1].split('.')[0]
+  print(f"pdf file name - {pdf_name}")
     
-    print("Errors:")
-    print(len(errors))
-    return output
+  #! Get prediction for a PDF using nougat
+  nougat_ocr(pdf_file.name)
+  
+  # Open the multimarkdown (.mmd) file for reading
+  with open(f'/content/output/{pdf_name}.mmd', 'r') as file:
+      content = file.read() 
+      
+  return content
 
-with gr.Blocks() as demo:
 
+with gr.Blocks() as demo:
+    
   with gr.Row():
     pdf_file = gr.File(label='Upload a PDF', scale=1)
-    mkd = gr.HTML('  <i>OR</i>  ', scale=1)
+    mkd = gr.Markdown('<h2><center><i>OR</i></center></h2>',scale=1)
     pdf_link = gr.Textbox(placeholder='Enter an arxiv link here', label='Provide a link', scale=1)
 
-  with gr.Row():
-    btn = gr.Button()
-    parsed_output = gr.Textbox(lines=5)
+  btn = gr.Button()
+  parsed_output = gr.Markdown()
+  
+  btn.click(predict, pdf_file, parsed_output )
 
-  btn.click(fun, pdf_file, parsed_output)
-    
+demo.queue()
 demo.launch(debug=True)
+