From a1429d1607aa7274f301478c9c8faf5127031243 Mon Sep 17 00:00:00 2001
From: oobabooga <112222186+oobabooga@users.noreply.github.com>
Date: Tue, 28 Feb 2023 02:20:11 -0300
Subject: [PATCH] Add default extensions to the settings

---
 modules/shared.py |  2 ++
 server.py         | 13 ++++++++++++-
 2 files changed, 14 insertions(+), 1 deletion(-)

diff --git a/modules/shared.py b/modules/shared.py
index d59cee99..f0408068 100644
--- a/modules/shared.py
+++ b/modules/shared.py
@@ -39,6 +39,8 @@ settings = {
     'name2_pygmalion': 'Kawaii',
     'context_pygmalion': "Kawaii's persona: Kawaii is a cheerful person who loves to make others smile. She is an optimist who loves to spread happiness and positivity wherever she goes.\n<START>",
     'stop_at_newline_pygmalion': False,
+    'default_extensions': [],
+    'chat_default_extensions': ["gallery"],
 }
 
 parser = argparse.ArgumentParser(formatter_class=lambda prog: argparse.HelpFormatter(prog,max_help_position=54))
diff --git a/server.py b/server.py
index f64b526d..564c8644 100644
--- a/server.py
+++ b/server.py
@@ -151,7 +151,18 @@ available_characters = get_available_characters()
 available_softprompts = get_available_softprompts()
 
 extensions_module.available_extensions = get_available_extensions()
-if shared.args.extensions is not None:
+# Activate the default extensions
+if shared.args.chat or shared.args.cai_chat:
+    for extension in shared.settings['chat_default_extensions']:
+        shared.args.extensions = shared.args.extensions or []
+        if extension not in shared.args.extensions:
+            shared.args.extensions.append(extension)
+else:
+    for extension in shared.settings['default_extensions']:
+        shared.args.extensions = shared.args.extensions or []
+        if extension not in shared.args.extensions:
+            shared.args.extensions.append(extension)
+if shared.args.extensions is not None and len(shared.args.extensions) > 0:
     extensions_module.load_extensions()
 
 # Choosing the default model