Create app.py

app.py

@@ -0,0 +1,53 @@
+import gradio as gr
+import tempfile
+import re
+from PyPDF2 import PdfReader, PdfFileReader
+import os 
+import spacy
+import pytesseract
+import pdf2image
+import subprocess
+from pdf2image.exceptions import (
+    PDFInfoNotInstalledError,
+    PDFPageCountError,
+    PDFSyntaxError
+)
+
+def clean_text(text):
+    nlp = spacy.load("en_core_web_sm", disable=["tagger", "parser", "ner", "textcat"])
+    text = re.sub(r'\n+', '\n', text) 
+    text = re.sub(r'\s+', ' ', text)
+    return text.strip()
+
+def image_to_latex(image):
+    image_path = "/tmp/equation.png"  # Modify as needed
+    image.save(image_path)
+    result = subprocess.run(["pix2tex", image_path], capture_output=True, text=True)
+    return result.stdout
+
+def pdf_to_text(file):
+    with open(file.name, 'rb') as f:
+        reader = PdfReader(f)
+        full_text = ''
+        for i, page in enumerate(reader.pages):
+            page_text = page.extract_text()
+            if page_text is None:
+                images = pdf2image.convert_from_path(file.name, first_page=i+1, last_page=i+2)
+                for image in images:
+                    page_text = image_to_latex(image)
+            page_text = clean_text(page_text)
+            if len(page_text.split()) > 5: 
+                # Adding the page number as part of the page text
+                full_text += f"Page Number: {i+1}\n{page_text}\n\n"
+    base_name = os.path.splitext(os.path.basename(file.name))[0]
+    output_file_name = base_name + ".txt"
+    with open(output_file_name, 'w') as f:
+        f.write(full_text)
+    return output_file_name
+
+iface = gr.Interface(fn=pdf_to_text,
+                     inputs=gr.inputs.File(label="Your PDF"),
+                     outputs=gr.outputs.File(label="Download TXT"),
+                     title="PDF to TXT",
+                     description="Convert your PDF files to clean text")
+iface.launch()