From 12357149147aff4d5f48585310ffab4534fcea6a Mon Sep 17 00:00:00 2001
From: Jun Siang Cheah <git@jscheah.me>
Date: Sun, 26 May 2024 17:10:25 +0100
Subject: [PATCH] feat: add ENABLE_COMMUNITY_SHARING

---
 backend/apps/webui/main.py                    |  2 ++
 backend/config.py                             |  6 +++++
 backend/main.py                               |  1 +
 src/lib/components/chat/ShareChatModal.svelte | 24 ++++++++++---------
 src/lib/stores/index.ts                       | 11 +++++----
 5 files changed, 28 insertions(+), 16 deletions(-)

diff --git a/backend/apps/webui/main.py b/backend/apps/webui/main.py
index d736cef9a..e5ef35058 100644
--- a/backend/apps/webui/main.py
+++ b/backend/apps/webui/main.py
@@ -24,6 +24,7 @@ from config import (
     WEBUI_AUTH_TRUSTED_EMAIL_HEADER,
     JWT_EXPIRES_IN,
     AppConfig,
+    ENABLE_COMMUNITY_SHARING,
 )
 
 app = FastAPI()
@@ -41,6 +42,7 @@ app.state.config.DEFAULT_USER_ROLE = DEFAULT_USER_ROLE
 app.state.config.USER_PERMISSIONS = USER_PERMISSIONS
 app.state.config.WEBHOOK_URL = WEBHOOK_URL
 
+app.state.config.ENABLE_COMMUNITY_SHARING = ENABLE_COMMUNITY_SHARING
 
 app.state.MODELS = {}
 app.state.AUTH_TRUSTED_EMAIL_HEADER = WEBUI_AUTH_TRUSTED_EMAIL_HEADER
diff --git a/backend/config.py b/backend/config.py
index daa89de57..83eb1b37a 100644
--- a/backend/config.py
+++ b/backend/config.py
@@ -566,6 +566,12 @@ WEBHOOK_URL = PersistentConfig(
 
 ENABLE_ADMIN_EXPORT = os.environ.get("ENABLE_ADMIN_EXPORT", "True").lower() == "true"
 
+ENABLE_COMMUNITY_SHARING = PersistentConfig(
+    "ENABLE_COMMUNITY_SHARING",
+    "ui.enable_community_sharing",
+    os.environ.get("ENABLE_COMMUNITY_SHARING", "True").lower() == "true",
+)
+
 ####################################
 # WEBUI_SECRET_KEY
 ####################################
diff --git a/backend/main.py b/backend/main.py
index ab43da853..7a5bd8126 100644
--- a/backend/main.py
+++ b/backend/main.py
@@ -365,6 +365,7 @@ async def get_app_config():
             "auth_trusted_header": bool(webui_app.state.AUTH_TRUSTED_EMAIL_HEADER),
             "enable_image_generation": images_app.state.config.ENABLED,
             "enable_admin_export": ENABLE_ADMIN_EXPORT,
+            "enable_community_sharing": webui_app.state.config.ENABLE_COMMUNITY_SHARING,
         },
     }
 
diff --git a/src/lib/components/chat/ShareChatModal.svelte b/src/lib/components/chat/ShareChatModal.svelte
index 2c0042fb5..0a1d02058 100644
--- a/src/lib/components/chat/ShareChatModal.svelte
+++ b/src/lib/components/chat/ShareChatModal.svelte
@@ -1,6 +1,6 @@
 <script lang="ts">
 	import { getContext, onMount } from 'svelte';
-	import { models } from '$lib/stores';
+	import { models, config } from '$lib/stores';
 
 	import { toast } from 'svelte-sonner';
 	import { deleteSharedChatById, getChatById, shareChatById } from '$lib/apis/chats';
@@ -134,16 +134,18 @@
 				<div class="flex justify-end">
 					<div class="flex flex-col items-end space-x-1 mt-1.5">
 						<div class="flex gap-1">
-							<button
-								class=" self-center px-3.5 py-2 rounded-xl text-sm font-medium bg-gray-100 hover:bg-gray-200 text-gray-800 dark:bg-gray-850 dark:hover:bg-gray-800 dark:text-white"
-								type="button"
-								on:click={() => {
-									shareChat();
-									show = false;
-								}}
-							>
-								{$i18n.t('Share to OpenWebUI Community')}
-							</button>
+							{#if $config?.feature_flags.enable_community_sharing}
+								<button
+									class=" self-center px-3.5 py-2 rounded-xl text-sm font-medium bg-gray-100 hover:bg-gray-200 text-gray-800 dark:bg-gray-850 dark:hover:bg-gray-800 dark:text-white"
+									type="button"
+									on:click={() => {
+										shareChat();
+										show = false;
+									}}
+								>
+									{$i18n.t('Share to OpenWebUI Community')}
+								</button>
+							{/if}
 
 							<button
 								class=" self-center flex items-center gap-1 px-3.5 py-2 rounded-xl text-sm font-medium bg-emerald-600 hover:bg-emerald-500 text-white"
diff --git a/src/lib/stores/index.ts b/src/lib/stores/index.ts
index 25b50681f..b5e97fdeb 100644
--- a/src/lib/stores/index.ts
+++ b/src/lib/stores/index.ts
@@ -133,11 +133,12 @@ type Config = {
 	default_models: string[];
 	default_prompt_suggestions: PromptSuggestion[];
 	feature_flags: {
-		auth?: boolean;
-		enable_signup?: boolean;
-		auth_trusted_header?: boolean;
-		enable_image_generation?: boolean;
-		enable_admin_export?: boolean;
+		auth: boolean;
+		enable_signup: boolean;
+		auth_trusted_header: boolean;
+		enable_image_generation: boolean;
+		enable_admin_export: boolean;
+		enable_community_sharing: boolean;
 	};
 };