Added additional error handling to job related functionalities. (Formatted file in db_utils, which might cause a merge conflict, just override my file later)

swagger_server/controllers/extract_controller.py

@@ -12,11 +12,11 @@ from swagger_server.util import get_random_filename, create_random_file_in_tmp_f
 import requests
 from werkzeug.utils import secure_filename
 
-# ATEapi_endpoint = "http://localhost:5000/predict"
-
-
 ATEapi_endpoint = "http://ate-api:5000/predict"
 
+# endpoint below to be used only for development purposes (don't need to run docker)
+# ATEapi_endpoint = "http://localhost:5000/predict"
+
 
 def do_izlusci(conllus, prepovedane_besede):
     tmp_file_path = ""

swagger_server/controllers/jobs_controller.py

@@ -1,6 +1,7 @@
 import datetime
 import json
 import os.path
+import traceback
 
 import peewee
 import asyncio
@@ -92,6 +93,7 @@ def try_do_jobs_ateapi():
                     [ex.submit(execute_ateapi_job, job) for job in unfinished_jobs]
         except Exception as e:
             print(f"Exception in try_do_jobs_ateapi")
+            traceback.print_exc()
         finally:
             time.sleep(3)
 
@@ -120,6 +122,7 @@ def try_do_jobs_classla():
 
         except Exception as e:
             print(f"Exception in try_do_jobs_classla")
+            traceback.print_exc()
         finally:
             time.sleep(3)
 
@@ -136,6 +139,7 @@ def try_do_jobs_doc2text():
                     [ex.submit(execute_doc2text_job, job) for job in unfinished_jobs]
         except Exception as e:
             print(f"Exception in try_do_jobs_doc2text")
+            traceback.print_exc()
         finally:
             time.sleep(3)
 
@@ -163,9 +167,9 @@ def execute_doc2text_job(job: Job):
             jtype = job.job_type
             text = ""
             if jtype in [1, 12]:
-                text = txt_utils.extract_text_prepResp(file)
+                text, _ = txt_utils.extract_text_prepResp(file)
             elif jtype in [3, 32]:
-                text = txt_utils.ocr_text_prepResp(file)
+                text, _ = txt_utils.ocr_text_prepResp(file)
 
             if jtype in [1, 3]:
                 job.job_output = text

swagger_server/controllers/oss_controller.py

@@ -1,4 +1,4 @@
-from swagger_server import db_utils
+from swagger_server.utils import db_utils
 from swagger_server import util
 from flask import send_file