dify

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

@@ -0,0 +1,68 @@
+from flask_restx import Resource, fields, marshal_with, reqparse
+
+from constants.languages import languages
+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,
+}
+
+recommended_app_fields = {
+    "app": fields.Nested(app_fields, 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,
+}
+
+recommended_app_list_fields = {
+    "recommended_apps": fields.List(fields.Nested(recommended_app_fields)),
+    "categories": fields.List(fields.String),
+}
+
+
+parser_apps = reqparse.RequestParser().add_argument("language", type=str, location="args")
+
+
+@console_ns.route("/explore/apps")
+class RecommendedAppListApi(Resource):
+    @console_ns.expect(parser_apps)
+    @login_required
+    @account_initialization_required
+    @marshal_with(recommended_app_list_fields)
+    def get(self):
+        # language args
+        args = parser_apps.parse_args()
+
+        language = args.get("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)