VERY Important commit

.gitignore

@@ -8,6 +8,4 @@ venv/
 pip-wheel-metadata/
 .env
 .DS_Store
-
-
-
+static/

app/__init__.py

@@ -1,6 +1,7 @@
 from fastapi import FastAPI
 from fastapi.middleware.cors import CORSMiddleware
 from starlette.exceptions import HTTPException as StarletteHTTPException
+from starlette.staticfiles import StaticFiles
 
 from app.core.wrappers import HectoolResponseWrapper, ErrorHectoolResponse
 
@@ -27,6 +28,11 @@ def create_app() -> FastAPI:
         allow_headers=["*"],
     )
 
+    app.mount(
+        '/static',
+        StaticFiles(directory='static'),
+    )
+
     @app.exception_handler(StarletteHTTPException)
     async def http_exception_handler(_, exc):
         return HectoolResponseWrapper(

app/api/message/ai/utils.py

@@ -1,5 +1,13 @@
+import base64
+import os
+from uuid import uuid4
+
+from fastapi import HTTPException
+
 from app.api.message.ai.prompts import Prompts
 from app.api.message.model import MessageModel
+from app.api.message.schemas import UploadImageRequest
+from app.core.config import settings
 
 
 def transform_messages_to_openai(messages: list[MessageModel]) -> list[dict]:
@@ -30,3 +38,17 @@ def transform_messages_to_openai(messages: list[MessageModel]) -> list[dict]:
     return openai_messages
 
 
+def save_image(file: UploadImageRequest) -> str:
+    try:
+        image_data = base64.b64decode(file.base64String)
+    except Exception:
+        raise HTTPException(status_code=400, detail="Invalid base64 string")
+
+    file_extension = os.path.splitext(file.name)[1] or ".jpg"
+    unique_filename = f"{uuid4()}{file_extension}"
+    file_path = os.path.join(settings.STATIC_DIR, unique_filename)
+
+    with open(file_path, "wb") as f:
+        f.write(image_data)
+    file_url = f"{settings.Issuer}/static/{unique_filename}"
+    return file_url

app/api/message/dto.py

@@ -10,4 +10,4 @@ class Author(Enum):
 
 class File(BaseModel):
     name: str
-    base64String: str
+    url: str

app/api/message/schemas.py

@@ -1,7 +1,7 @@
 from pydantic import BaseModel
 
 from app.api.common.dto import Paging
-from app.api.message.dto import Author, File
+from app.api.message.dto import File
 from app.api.message.model import MessageModel
 from app.core.wrappers import HectoolResponseWrapper
 
@@ -22,3 +22,16 @@ class AllMessageResponse(BaseModel):
 
 class AllMessageWrapper(HectoolResponseWrapper[AllMessageResponse]):
     pass
+
+
+class UploadImageRequest(BaseModel):
+    name: str
+    base64String: str
+
+
+class UploadImageResponse(BaseModel):
+    url: str
+
+
+class UploadImageWrapper(HectoolResponseWrapper[UploadImageResponse]):
+    pass

app/api/message/views.py

@@ -5,11 +5,20 @@ from app.api.account.model import AccountModel
 from app.api.common.dto import Paging
 from app.api.message import message_router
 from app.api.message.ai.openai_request import response_generator
+from app.api.message.ai.utils import save_image
 from app.api.message.db_requests import get_all_chat_messages_obj, create_message_obj, update_chat_with_thread_id
-from app.api.message.schemas import AllMessageWrapper, AllMessageResponse, CreateMessageRequest
+from app.api.message.schemas import AllMessageWrapper, AllMessageResponse, CreateMessageRequest, UploadImageRequest, \
+    UploadImageWrapper, UploadImageResponse
 from app.core.security import PermissionDependency
 
 
+@message_router.post('/image')
+async def create_image_message(
+        file: UploadImageRequest,
+):
+    url = save_image(file)
+    return UploadImageWrapper(data=UploadImageResponse(url=url))
+
 @message_router.get('/{chatId}/all')
 async def get_all_chat_messages(
         chatId: str, account: AccountModel = Depends(PermissionDependency(is_public=True))

app/core/config.py

@@ -9,7 +9,8 @@ from openai import AsyncClient
 load_dotenv()
 
 class BaseConfig:
-    BASE_DIR: pathlib.Path = pathlib.Path(__file__).parent.parent
+    BASE_DIR: pathlib.Path = pathlib.Path(__file__).parent.parent.parent
+    STATIC_DIR = "static"
     SECRET_KEY = os.getenv('SECRET')
     DB_CLIENT = motor.motor_asyncio.AsyncIOMotorClient(os.getenv("MONGO_DB_URL")).hectool
     OPENAI_CLIENT = AsyncClient(api_key=os.getenv('OPENAI_API_KEY'))
@@ -22,8 +23,8 @@ class DevelopmentConfig(BaseConfig):
 
 
 class ProductionConfig(BaseConfig):
-    Issuer = "https://api.marscapita.com"
-    Audience = "https://app.marscapita.com"
+    Issuer = "https://brestok-hector-demo-backend.hf.space"
+    Audience = "https://brestok-1.github.io/hectool-demo-frontend/"
 
 
 @lru_cache()