Brushed up the backend, added writing task 1 academic prompt gen and grading ENCOA-274

ielts_be/utils/__init__.py

@@ -0,0 +1,7 @@
+from .handle_exception import handle_exception
+from .logger import suppress_loggers
+
+__all__ = [
+    "handle_exception",
+    "suppress_loggers"
+]

ielts_be/utils/handle_exception.py

@@ -0,0 +1,15 @@
+import functools
+from typing import Callable, Any
+from fastapi import Response
+
+
+def handle_exception(status_code: int = 500):
+    def decorator(func: Callable) -> Callable:
+        @functools.wraps(func)
+        async def wrapper(*args: Any, **kwargs: Any) -> Any:
+            try:
+                return await func(*args, **kwargs)
+            except Exception as e:
+                return Response(content=str(e), status_code=status_code)
+        return wrapper
+    return decorator

ielts_be/utils/image_to_b64.py

@@ -0,0 +1,10 @@
+def _b64_vision_images(path: str):
+    for filename in os.listdir(path):
+        if filename.startswith("cv-") and filename.endswith(".png"):
+            cv_dict.append({
+              "type": "image_url",
+              "image_url": {
+                "url": f"data:image/png;base64,{_encode_image(os.path.join(path, filename))}"
+              }
+            })
+    return cv_dict

ielts_be/utils/logger.py

@@ -0,0 +1,20 @@
+import logging
+from functools import wraps
+
+
+def suppress_loggers():
+    def decorator(f):
+        @wraps(f)
+        def wrapped(*args, **kwargs):
+            root_logger = logging.getLogger()
+            original_level = root_logger.level
+
+            root_logger.setLevel(logging.ERROR)
+
+            try:
+                return f(*args, **kwargs)
+            finally:
+                root_logger.setLevel(original_level)
+
+        return wrapped
+    return decorator