text

app.py

@@ -6,6 +6,7 @@ from pdf2image import convert_from_path
 import shutil
 import tempfile
 from transformers import pipeline
+import subprocess as sp
 
 out_files = gr.State([])
 FILE_TIMEOUT = 10 ** 3
@@ -31,6 +32,18 @@ def handle_files(cur_files):
         cur_files = cur_files[-MAX_FILES:]
     return cur_files
 
+
+def extract_text(pdf_file):
+    """
+    Generate a text rendering of a PDF file in the form of a list of lines.
+    """
+    args = ['pdftotext', '-layout', path, '-']
+    cp = sp.run(
+      args, stdout=sp.PIPE, stderr=sp.DEVNULL,
+      check=True, text=True
+    )
+    return cp.stdout
+
 # Function to process PDF and generate ZIP file
 def process_pdf(pdf_file, cur_files):
     
@@ -60,7 +73,14 @@ def process_image_and_text(doc, text_query):
         outputs += p(img, question)
     return sorted(outputs, key=lambda x: x["score"], reverse=True)[0]['answer']
 
-# Interface for the Gradio app
+text_interface = gr.Interface(
+    fn=extract_text,
+    inputs=[PDF(label="Upload PDF"), out_files],
+    outputs=[gr.Textbox(label="Extracted Text"), out_files],
+    title="PDF extractor",
+    description="Extracts text from the PDF container."
+)
+
 pdf_interface = gr.Interface(
     fn=process_pdf,
     inputs=[PDF(label="Upload PDF"), out_files],
@@ -82,9 +102,9 @@ image_interface = gr.Interface(
 
 # Create a tabbed interface
 tabbed_interface = gr.TabbedInterface(
-    [pdf_interface, image_interface],
+    [text_interface, pdf_interface, image_interface],
     title="PDF interaction",
-    tab_names=["Converter", "Interaction"],
+    tab_names=["Text extractor", "Converter", "Interaction"],
     # description="Choose a tab to perform the desired task."
 )