Spaces:
Sleeping
Sleeping
Update controllers/ws_controller.py
Browse files- controllers/ws_controller.py +21 -16
controllers/ws_controller.py
CHANGED
@@ -272,23 +272,28 @@ def extract_attachments_from_mail(access_token: str, message_data: dict) -> List
|
|
272 |
print(f"Error processing attachment {filename}: {str(e)}")
|
273 |
continue
|
274 |
struct_data = ut.strcuture_document_data(raw_text)
|
275 |
-
st_str =
|
276 |
-
|
277 |
-
"
|
278 |
-
"
|
279 |
-
"
|
280 |
-
"
|
281 |
-
"
|
282 |
-
"
|
283 |
-
"
|
284 |
-
"
|
285 |
-
"payment_method": "INSERT PAYMENT METHOD FROM THE RECEIPT OCR TEXT. LOOK FOR KEYWORDS LIKE 'CASH', 'CARD', 'CREDIT', 'DEBIT', 'VISA', 'MASTERCARD', 'AMEX', 'PAYPAL', ETC. IF NOT FOUND RETURN null."
|
286 |
}
|
287 |
-
|
288 |
-
|
289 |
-
|
290 |
-
|
291 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
292 |
|
293 |
|
294 |
|
|
|
272 |
print(f"Error processing attachment {filename}: {str(e)}")
|
273 |
continue
|
274 |
struct_data = ut.strcuture_document_data(raw_text)
|
275 |
+
st_str ={
|
276 |
+
"brand": "INSERT BRAND NAME",
|
277 |
+
"total_cost": "INSERT TOTAL COST",
|
278 |
+
"location": "INSERT LOCATION FROM",
|
279 |
+
"purchase_category": "INSERT PURCHASE CATEGORY",
|
280 |
+
"brand_category": "INSERT BRAND CATEGORY",
|
281 |
+
"Date": "INSERT RECEIPT DATE",
|
282 |
+
"currency": "INSERT CURRENCY",
|
283 |
+
"filename": "GENERATE A FILENAME",
|
284 |
+
"payment_method": "INSERT PAYMENT METHOD"
|
|
|
285 |
}
|
286 |
+
|
287 |
+
|
288 |
+
|
289 |
+
|
290 |
+
|
291 |
+
|
292 |
+
if struct_data is not None:
|
293 |
+
for key,value in st_str:
|
294 |
+
if value in struct_data[key]: struct_data[key]=None
|
295 |
+
|
296 |
+
structured_data.append(struct_data)
|
297 |
|
298 |
|
299 |
|