Update app.py

app.py

@@ -5,6 +5,7 @@ import gradio as gr
 import os
 import re
 import PyPDF2
+import pdfplumber
 
 app = Flask(__name__)
 
@@ -66,29 +67,30 @@ def findRealProb(text):
 
 def upload_file(file):
    
-    # if 'pdfFile' in request.files:
-    #     pdf_file = request.files['pdfFile']
-    #     text = ""
-    #     with pdfplumber.open(pdf_file) as pdf:
-    #         cnt = 0
-    #         for page in pdf.pages:
-    #             cnt+=1
-    #             text+=(page.extract_text(x_tolerance = 1))
-    #             print(text)
-    #             if cnt>5: 
-    #                 break
-    #         return findRealProb(text)
-    #         # return jsonify({'text': text})
-    if file:
-
-        with open(file.name, 'rb') as pdf_file:
-            pdf_reader = PyPDF2.PdfReader(pdf_file)
-            text = ''
-            for page_num in range(len(pdf_reader.pages)):
-                page = pdf_reader.pages[page_num]
-                text += page.extract_text()
-            text = text.replace('\n', ' ')    
+    if 'pdfFile' in request.files:
+        pdf_file = request.files['pdfFile']
+        text = ""
+        with pdfplumber.open(pdf_file) as pdf:
+            cnt = 0
+            for page in pdf.pages:
+                cnt+=1
+                text+=(page.extract_text(x_tolerance = 1))
+                print(text)
+                if cnt>5: 
+                    break
+            text = text.replace('\n', ' ') 
             return findRealProb(text)
+            # return jsonify({'text': text})
+    # if file:
+
+    #     with open(file.name, 'rb') as pdf_file:
+    #         pdf_reader = PyPDF2.PdfReader(pdf_file)
+    #         text = ''
+    #         for page_num in range(len(pdf_reader.pages)):
+    #             page = pdf_reader.pages[page_num]
+    #             text += page.extract_text()
+    #         text = text.replace('\n', ' ')    
+    #         return findRealProb(text)
         # pdf_file = file.name
         # print(file, pdf_file)
         # text = ""