Update app.py

app.py

@@ -27,12 +27,23 @@ bert_legal_model = Summarizer(custom_model=custom_model, custom_tokenizer=custom
 
 
 def pdf(file):
-    temp = open(file, 'rb')
-    PDF_read = PDFFileReader(temp)
-    first_page = PDF_read.getPage
-     
+      
+    path = os.getcwd()
+    folder_name = './'
+    path = os.path.join(path, folder_name) 
+    
+    list_of_files = []
+    for root, dirs, files in os.walk(path):
+        for file in files:
+            if(file.endswith(".pdf")):
+                # print(os.path.join(root,file))
+                list_of_files.append(os.path.join(root,file))
+    
+    total_pages  = print("\nProcessing {} files...\n".format(len(list_of_files)))
+
+    
     
-    return first_page 
+    return total_pages  
     #pageObject.extractText() 
           
  
@@ -41,7 +52,7 @@ def pdf(file):
 
 iface = gr.Interface(
     pdf, 
-    "file", "file"
+    "file", "text"
    )
 
 if __name__ == "__main__":