Added PDF upload UI.

app.py

@@ -13,7 +13,7 @@ from scipy.special import softmax
 from evaluate import load
 from datetime import date
 import nltk
-
+import fitz 
 from transformers import GPT2LMHeadModel, GPT2TokenizerFast
 import nltk, spacy, subprocess, torch
 import plotly.graph_objects as go
@@ -264,6 +264,14 @@ def len_validator(text):
     else :
         return f"Input length satisified." 
 
+def extract_text_from_pdf(pdf_path):
+    doc = fitz.open(pdf_path)
+    text = ""
+    for page in doc:
+        text += page.get_text()
+    return text
+
+
 # DEPTH ANALYSIS
 print("loading depth analysis")
 nltk.download('stopwords')
@@ -378,11 +386,14 @@ with gr.Blocks() as demo:
     # Copyright Checker 
     """
     )
-    input_text = gr.Textbox(label="Input text", lines=6, placeholder="")
+    with gr.Row():
+        input_text = gr.Textbox(label="Input text", lines=6, placeholder="")
+        file_input = gr.File(label="Upload PDF")
+        file_input.change(fn=extract_text_from_pdf, inputs=file_input, outputs=input_text)
+
     char_count = gr.Textbox(label="Minumum Character Limit Check") 
     input_text.change(fn=len_validator, inputs=input_text, outputs=char_count)
 
-
     with gr.Row():
         with gr.Column():
             only_ai_btn = gr.Button("AI Check")

requirements.txt

@@ -19,4 +19,5 @@ keras
 spacy
 textstat
 plotly
-tqdm
+tqdm
+pymupdf