From 0c04b18f366118c4a132261608aef21e7aafcb2f Mon Sep 17 00:00:00 2001
From: "Timothy J. Baek" <timothyjrbeck@gmail.com>
Date: Sun, 26 May 2024 13:02:40 -0700
Subject: [PATCH] refac

---
 backend/main.py                                         | 4 ++--
 src/lib/components/admin/Settings/Database.svelte       | 2 +-
 src/lib/components/chat/Messages/ResponseMessage.svelte | 2 +-
 src/lib/components/chat/ShareChatModal.svelte           | 2 +-
 src/lib/stores/index.ts                                 | 2 +-
 src/routes/auth/+page.svelte                            | 9 +++------
 6 files changed, 9 insertions(+), 12 deletions(-)

diff --git a/backend/main.py b/backend/main.py
index 586ad27e2..d1d267ce0 100644
--- a/backend/main.py
+++ b/backend/main.py
@@ -361,10 +361,10 @@ async def get_app_config():
         "default_locale": default_locale,
         "default_models": webui_app.state.config.DEFAULT_MODELS,
         "default_prompt_suggestions": webui_app.state.config.DEFAULT_PROMPT_SUGGESTIONS,
-        "feature_flags": {
+        "features": {
             "auth": WEBUI_AUTH,
-            "enable_signup": webui_app.state.config.ENABLE_SIGNUP,
             "auth_trusted_header": bool(webui_app.state.AUTH_TRUSTED_EMAIL_HEADER),
+            "enable_signup": webui_app.state.config.ENABLE_SIGNUP,
             "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/admin/Settings/Database.svelte b/src/lib/components/admin/Settings/Database.svelte
index 87f29f4f5..3ae01a668 100644
--- a/src/lib/components/admin/Settings/Database.svelte
+++ b/src/lib/components/admin/Settings/Database.svelte
@@ -34,7 +34,7 @@
 		<div>
 			<div class=" mb-2 text-sm font-medium">{$i18n.t('Database')}</div>
 
-			{#if $config?.feature_flags.enable_admin_export ?? true}
+			{#if $config?.features.enable_admin_export ?? true}
 				<div class="  flex w-full justify-between">
 					<!-- <div class=" self-center text-xs font-medium">{$i18n.t('Allow Chat Deletion')}</div> -->
 
diff --git a/src/lib/components/chat/Messages/ResponseMessage.svelte b/src/lib/components/chat/Messages/ResponseMessage.svelte
index 05d8f0cda..c5d753b51 100644
--- a/src/lib/components/chat/Messages/ResponseMessage.svelte
+++ b/src/lib/components/chat/Messages/ResponseMessage.svelte
@@ -693,7 +693,7 @@
 											</button>
 										</Tooltip>
 
-										{#if $config?.feature_flags.enable_image_generation && !readOnly}
+										{#if $config?.features.enable_image_generation && !readOnly}
 											<Tooltip content="Generate Image" placement="bottom">
 												<button
 													class="{isLastMessage
diff --git a/src/lib/components/chat/ShareChatModal.svelte b/src/lib/components/chat/ShareChatModal.svelte
index 0a1d02058..e6ccc1323 100644
--- a/src/lib/components/chat/ShareChatModal.svelte
+++ b/src/lib/components/chat/ShareChatModal.svelte
@@ -134,7 +134,7 @@
 				<div class="flex justify-end">
 					<div class="flex flex-col items-end space-x-1 mt-1.5">
 						<div class="flex gap-1">
-							{#if $config?.feature_flags.enable_community_sharing}
+							{#if $config?.features.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"
diff --git a/src/lib/stores/index.ts b/src/lib/stores/index.ts
index ebaa4d2ed..b65216315 100644
--- a/src/lib/stores/index.ts
+++ b/src/lib/stores/index.ts
@@ -135,7 +135,7 @@ type Config = {
 	default_locale: string;
 	default_models: string[];
 	default_prompt_suggestions: PromptSuggestion[];
-	feature_flags: {
+	features: {
 		auth: boolean;
 		enable_signup: boolean;
 		auth_trusted_header: boolean;
diff --git a/src/routes/auth/+page.svelte b/src/routes/auth/+page.svelte
index b8eda36e6..f836a5cc9 100644
--- a/src/routes/auth/+page.svelte
+++ b/src/routes/auth/+page.svelte
@@ -60,10 +60,7 @@
 			await goto('/');
 		}
 		loaded = true;
-		if (
-			($config?.feature_flags.auth_trusted_header ?? false) ||
-			$config?.feature_flags.auth === false
-		) {
+		if (($config?.features.auth_trusted_header ?? false) || $config?.features.auth === false) {
 			await signInHandler();
 		}
 	});
@@ -105,7 +102,7 @@
 		</div> -->
 
 		<div class="w-full sm:max-w-md px-10 min-h-screen flex flex-col text-center">
-			{#if ($config?.feature_flags.auth_trusted_header ?? false) || $config?.feature_flags.auth === false}
+			{#if ($config?.features.auth_trusted_header ?? false) || $config?.features.auth === false}
 				<div class=" my-auto pb-10 w-full">
 					<div
 						class="flex items-center justify-center gap-3 text-xl sm:text-2xl text-center font-bold dark:text-gray-200"
@@ -197,7 +194,7 @@
 								{mode === 'signin' ? $i18n.t('Sign in') : $i18n.t('Create Account')}
 							</button>
 
-							{#if $config?.feature_flags.enable_signup}
+							{#if $config?.features.enable_signup}
 								<div class=" mt-4 text-sm text-center">
 									{mode === 'signin'
 										? $i18n.t("Don't have an account?")