from django.urls import path
from rest_framework.routers import DefaultRouter

from .views import AITaskViewSet, GenerateImageView, ModelConfigViewSet

router = DefaultRouter()
router.register("tasks", AITaskViewSet, basename="ai-task")
router.register("models", ModelConfigViewSet, basename="model-config")

urlpatterns = [
    path("generate-image/", GenerateImageView.as_view(), name="ai-generate-image"),
] + router.urls
