feat: support upload image by multipart-form

app.py

@@ -1,7 +1,8 @@
+from http import HTTPStatus
 import os, io, base64
 import uvicorn
 import logging
-from fastapi import FastAPI, Header
+from fastapi import FastAPI, File, HTTPException, Header, UploadFile
 from apis.s3 import S3
 from dotenv import load_dotenv
 from models.response import Response
@@ -91,10 +92,45 @@ async def upload(
             except Exception as e:
                 res.error = str(e)
                 logging.warning(f"Error during upload: {res.error}")
-                return {"error": "Failed to upload content"}
     
     res.data = o.ref
+    return res
+
+
+@app.post("/image2")
+async def upload(
+    o: ImageObject,
+    x_request_user: str = Header(...),
+    x_api_key: str = Header(...),
+    file: UploadFile = File(...)
+):
+    res = Response()
+    logging.info("--------------------------------")
+    logging.info("Received request to upload image")
+    
+    # Validate headers
+    if not is_valid(x_request_user, x_api_key):
+        res.status = HTTPStatus.FORBIDDEN
+        res.error = "Invalid credentials"
+    else:
+        key = f'{o.key}/{o.job_no}/{o.name}'
+        logging.info(f'Key for S3 upload: {key}')
+
+        try:
+            # Read the file content directly from UploadFile
+            file_content = await file.read()
+            logging.info(f'File content length: {len(file_content)} bytes')
+
+            # Upload file object to S3
+            # logging.info(f"Uploading file to S3 bucket '{AWS_S3_BUCKET_NAME}' with key '{key}'")
+            res = s3client.upload_file(AWS_S3_BUCKET_NAME, file_content, key)
+            logging.info("File uploaded successfully")
+
+        except Exception as e:
+            res.error = str(e)
+            logging.warning(f"Error during upload: {res.error}")
     
+    res.data = o.ref
     return res
 
 def is_valid(u, p):