Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -5,6 +5,8 @@ import gradio as gr
|
|
| 5 |
import os
|
| 6 |
import re
|
| 7 |
import pdfplumber
|
|
|
|
|
|
|
| 8 |
app = Flask(__name__)
|
| 9 |
|
| 10 |
ACCESS_TOKEN = os.environ["ACCESS_TOKEN"]
|
|
@@ -79,17 +81,26 @@ def upload_file(file):
|
|
| 79 |
# return findRealProb(text)
|
| 80 |
# # return jsonify({'text': text})
|
| 81 |
if file:
|
| 82 |
-
|
| 83 |
-
|
| 84 |
-
|
| 85 |
-
|
| 86 |
-
|
| 87 |
-
|
| 88 |
-
|
| 89 |
-
|
| 90 |
-
if cnt>5:
|
| 91 |
-
break
|
| 92 |
return findRealProb(text)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 93 |
else:
|
| 94 |
return {"error":'No PDF file found in request'}
|
| 95 |
|
|
|
|
| 5 |
import os
|
| 6 |
import re
|
| 7 |
import pdfplumber
|
| 8 |
+
import PyPDF2
|
| 9 |
+
|
| 10 |
app = Flask(__name__)
|
| 11 |
|
| 12 |
ACCESS_TOKEN = os.environ["ACCESS_TOKEN"]
|
|
|
|
| 81 |
# return findRealProb(text)
|
| 82 |
# # return jsonify({'text': text})
|
| 83 |
if file:
|
| 84 |
+
|
| 85 |
+
with open(file.name, 'rb') as pdf_file:
|
| 86 |
+
pdf_reader = PyPDF2.PdfFileReader(pdf_file)
|
| 87 |
+
text = ''
|
| 88 |
+
for page_num in range(pdf_reader.getNumPages()):
|
| 89 |
+
page = pdf_reader.getPage(page_num)
|
| 90 |
+
text += page.extractText()
|
| 91 |
+
|
|
|
|
|
|
|
| 92 |
return findRealProb(text)
|
| 93 |
+
# pdf_file = file.name
|
| 94 |
+
# print(file, pdf_file)
|
| 95 |
+
# text = ""
|
| 96 |
+
# with pdfplumber.open(pdf_file) as pdf:
|
| 97 |
+
# cnt = 0
|
| 98 |
+
# for page in pdf.pages:
|
| 99 |
+
# cnt+=1
|
| 100 |
+
# text+=(page.extract_text(x_tolerance = 1))
|
| 101 |
+
# if cnt>5:
|
| 102 |
+
# break
|
| 103 |
+
# return findRealProb(text)
|
| 104 |
else:
|
| 105 |
return {"error":'No PDF file found in request'}
|
| 106 |
|