start reformat generate endpoint

src/endpoints/generate.py

@@ -1,4 +1,7 @@
 from fastapi import APIRouter
+from src.libs.model import CganCols
+from src.models.main import User, Method
+from src.models.generate import GenerateResult, ImageResult
 
 router = APIRouter(
     prefix="/generate",
@@ -6,6 +9,8 @@ router = APIRouter(
     responses={404: {"description": "Not found"}},
 )
 
+model = CganCols()
+
 
 @router.get("/")
 async def info():
@@ -15,3 +20,20 @@ async def info():
 @router.get("/status")
 async def status():
     return {"status": "OK"}
+
+
+@router.post("/")
+async def generate(user: User, index: int | None = None) -> GenerateResult:
+    if index:
+        result: GenerateResult = await generate_index(user, index)
+        return result
+
+    return await generate_all(user)
+
+
+async def generate_index(user: User, index: int) -> GenerateResult:
+    pass
+
+
+async def generate_all(user: User) -> GenerateResult:
+    pass

src/libs/model.py

@@ -0,0 +1,16 @@
+from huggingface_hub import from_pretrained_keras
+
+
+class CganCols:
+    def __init__(self):
+        self.model_0 = from_pretrained_keras('SupawitMarayat/pimthaigans-cgan-0010')
+        self.model_1 = from_pretrained_keras('SupawitMarayat/pimthaigans-cgan-1121')
+        self.model_2 = from_pretrained_keras('SupawitMarayat/pimthaigans-cgan-2232')
+        self.model_3 = from_pretrained_keras('SupawitMarayat/pimthaigans-cgan-3343')
+        self.model_4 = from_pretrained_keras('SupawitMarayat/pimthaigans-cgan-4454')
+        self.model_5 = from_pretrained_keras('SupawitMarayat/pimthaigans-cgan-5565')
+        self.model_6 = from_pretrained_keras('SupawitMarayat/pimthaigans-cgan-6676')
+        self.model_7 = from_pretrained_keras('SupawitMarayat/pimthaigans-cgan-7787')
+
+        self.model_cols = [self.model_0, self.model_1, self.model_2, self.model_3, self.model_4, self.model_5,
+                           self.model_6, self.model_7]

src/libs/s3.py

@@ -0,0 +1,26 @@
+import os
+from dotenv import load_dotenv
+import boto3
+
+load_dotenv(dotenv_path=".env")
+
+
+def s3session():
+    session = boto3.session.Session(
+        aws_access_key_id=os.environ.get("ACCESS_KEY_ID"),
+        aws_secret_access_key=os.environ.get("SECRET_ACCESS_KEY"),
+    )
+
+    return session
+
+
+def s3client():
+    session = s3session()
+    s3_client = session.client("s3")
+    return s3_client
+
+
+def s3resource():
+    session = s3session()
+    s3_resource = session.resource("s3")
+    return s3_resource

src/models/generate.py

@@ -0,0 +1,14 @@
+from typing import List
+from pydantic import BaseModel
+from src.models.main import User, Method
+
+
+class ImageResult(BaseModel):
+    index: int
+    image_url: str
+
+
+class GenerateResult(BaseModel):
+    user: User
+    method: Method
+    result: List[ImageResult]

src/models/main.py

@@ -0,0 +1,13 @@
+"""Main schema for the Api"""
+from pydantic import BaseModel
+from uuid import UUID
+from enum import Enum
+
+
+class User(BaseModel):
+    uuid: UUID
+
+
+class Method(str, Enum):
+    all = "all"
+    index = "index"