Update app.py

app.py

@@ -60,7 +60,19 @@ def findRealProb(text):
     realProb = ans/len(text)
     return {"Real": realProb, "Fake": 1-realProb, "results": results, "text": text}
 
-def upload_file():
+def upload_file(file):
+    if file:
+        pdf_file = file
+        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)
     if 'pdfFile' in request.files:
         pdf_file = request.files['pdfFile']
         text = ""