Penality commited on
Commit
e437d27
·
verified ·
1 Parent(s): ff980b9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -157,11 +157,13 @@ def retrieve_document(query):
157
  with open(recieved_file, "wb") as f:
158
  f.write(response.content)
159
 
 
 
160
  print(f"PDF received successfully: received_{filename}")
161
  else:
162
  print(f"Error: {response.status_code}, {response.json()}")
163
 
164
- return recieved_file
165
 
166
  def clean_text(text):
167
  """Cleans extracted text for better processing by the model."""
 
157
  with open(recieved_file, "wb") as f:
158
  f.write(response.content)
159
 
160
+ prompt_doc = extract_text_from_pdf(recieved_file)
161
+
162
  print(f"PDF received successfully: received_{filename}")
163
  else:
164
  print(f"Error: {response.status_code}, {response.json()}")
165
 
166
+ return prompt_doc
167
 
168
  def clean_text(text):
169
  """Cleans extracted text for better processing by the model."""