Update app.py
Browse files
app.py
CHANGED
@@ -144,16 +144,16 @@ class PDFTextExtractor:
|
|
144 |
max_tokens=4096
|
145 |
)
|
146 |
|
147 |
-
|
148 |
-
|
149 |
-
|
150 |
})
|
151 |
|
152 |
-
|
153 |
|
154 |
-
|
155 |
-
|
156 |
-
|
157 |
|
158 |
def extract_text(pdf_file):
|
159 |
if OPENAI_API_KEY is None:
|
|
|
144 |
max_tokens=4096
|
145 |
)
|
146 |
|
147 |
+
extracted_texts.append({
|
148 |
+
'page': i + 1,
|
149 |
+
'text': response.choices[0].message.content
|
150 |
})
|
151 |
|
152 |
+
return extracted_texts
|
153 |
|
154 |
+
except Exception as e:
|
155 |
+
print(f"Error in text extraction: {str(e)}")
|
156 |
+
return None
|
157 |
|
158 |
def extract_text(pdf_file):
|
159 |
if OPENAI_API_KEY is None:
|