40 lines
1.3 KiB
Python
40 lines
1.3 KiB
Python
from typing import List
|
|
from fastapi import APIRouter
|
|
from sqlmodel import select
|
|
|
|
from models.image_prompt import ImagePrompt
|
|
from models.sql.image_asset import ImageAsset
|
|
from services.database import get_sql_session
|
|
from services.image_generation_service import ImageGenerationService
|
|
from utils.asset_directory_utils import get_images_directory
|
|
|
|
IMAGES_ROUTER = APIRouter(prefix="/images", tags=["Images"])
|
|
|
|
|
|
@IMAGES_ROUTER.get("/generate")
|
|
async def generate_image(prompt: str):
|
|
images_directory = get_images_directory()
|
|
image_prompt = ImagePrompt(prompt=prompt)
|
|
image_generation_service = ImageGenerationService(images_directory)
|
|
|
|
image = await image_generation_service.generate_image(image_prompt)
|
|
if not isinstance(image, ImageAsset):
|
|
return image
|
|
|
|
with get_sql_session() as sql_session:
|
|
sql_session.add(image)
|
|
sql_session.commit()
|
|
|
|
return image.path
|
|
|
|
|
|
@IMAGES_ROUTER.get("/generated", response_model=List[ImageAsset])
|
|
async def get_generated_images():
|
|
try:
|
|
with get_sql_session() as sql_session:
|
|
images = sql_session.exec(
|
|
select(ImageAsset).order_by(ImageAsset.created_at.desc())
|
|
).all()
|
|
return images
|
|
except Exception as e:
|
|
return {"error": f"Failed to retrieve generated images: {str(e)}"}
|