dify/api/controllers/console/explore/recommended_app.py

from flask import request
from flask_restx import Resource, fields, marshal_with
from pydantic import BaseModel, Field

from constants.languages import languages
from controllers.common.schema import get_or_create_model
from controllers.console import console_ns
from controllers.console.wraps import account_initialization_required
from libs.helper import AppIconUrlField
from libs.login import current_user, login_required
from services.recommended_app_service import RecommendedAppService

app_fields = {
    "id": fields.String,
    "name": fields.String,
    "mode": fields.String,
    "icon": fields.String,
    "icon_type": fields.String,
    "icon_url": AppIconUrlField,
    "icon_background": fields.String,
}

app_model = get_or_create_model("RecommendedAppInfo", app_fields)

recommended_app_fields = {
    "app": fields.Nested(app_model, attribute="app"),
    "app_id": fields.String,
    "description": fields.String(attribute="description"),
    "copyright": fields.String,
    "privacy_policy": fields.String,
    "custom_disclaimer": fields.String,
    "category": fields.String,
    "position": fields.Integer,
    "is_listed": fields.Boolean,
    "can_trial": fields.Boolean,
}

recommended_app_model = get_or_create_model("RecommendedApp", recommended_app_fields)

recommended_app_list_fields = {
    "recommended_apps": fields.List(fields.Nested(recommended_app_model)),
    "categories": fields.List(fields.String),
}

recommended_app_list_model = get_or_create_model("RecommendedAppList", recommended_app_list_fields)


class RecommendedAppsQuery(BaseModel):
    language: str | None = Field(default=None)


console_ns.schema_model(
    RecommendedAppsQuery.__name__,
    RecommendedAppsQuery.model_json_schema(ref_template="#/definitions/{model}"),
)


@console_ns.route("/explore/apps")
class RecommendedAppListApi(Resource):
    @console_ns.expect(console_ns.models[RecommendedAppsQuery.__name__])
    @login_required
    @account_initialization_required
    @marshal_with(recommended_app_list_model)
    def get(self):
        # language args
        args = RecommendedAppsQuery.model_validate(request.args.to_dict(flat=True))  # type: ignore
        language = args.language
        if language and language in languages:
            language_prefix = language
        elif current_user and current_user.interface_language:
            language_prefix = current_user.interface_language
        else:
            language_prefix = languages[0]

        return RecommendedAppService.get_recommended_apps_and_categories(language_prefix)


@console_ns.route("/explore/apps/<uuid:app_id>")
class RecommendedAppApi(Resource):
    @login_required
    @account_initialization_required
    def get(self, app_id):
        app_id = str(app_id)
        return RecommendedAppService.get_recommend_app_detail(app_id)