Update utils.py

utils.py

@@ -1,23 +1,22 @@
-# -*- coding:utf-8 -*-
-from __future__ import annotations
-from typing import TYPE_CHECKING, Any, Callable, Dict, List, Tuple, Type
 import logging
 import json
 import os
-
+import pathlib
 import csv
 import requests
 import re
-
 import gradio as gr
 from pypinyin import lazy_pinyin
 import tiktoken
 import mdtex2html
+from typing import TYPE_CHECKING, Any, Callable, Dict, List, Tuple, Type
 from markdown import markdown
 from pygments import highlight
 from pygments.lexers import get_lexer_by_name
 from pygments.formatters import HtmlFormatter
+from gradio.themes.utils import ThemeAsset
 
+    
 use_websearch_checkbox=False
 use_streaming_checkbox=True
 model_select_dropdown="gpt-3.5-turbo"
@@ -61,7 +60,6 @@ The default model role of the app is the original assistant of ChatGPT, but you
         
 MODELS = ["gpt-3.5-turbo", "gpt-3.5-turbo-0301",]  
 
-
 if TYPE_CHECKING:
     from typing import TypedDict
 
@@ -309,5 +307,56 @@ def submit_key(key):
     logging.info(msg)
     return key, msg
 
+def create_theme_dropdown():
+    import gradio as gr
+
+    asset_path = pathlib.Path(__file__).parent / "themes"
+    themes = []
+    for theme_asset in os.listdir(str(asset_path)):
+        themes.append(
+            (ThemeAsset(theme_asset), gr.Theme.load(str(asset_path / theme_asset)))
+        )
+
+    def make_else_if(theme_asset):
+        return f"""
+        else if (theme == '{str(theme_asset[0].version)}') {{
+            var theme_css = `{theme_asset[1]._get_theme_css()}`
+        }}"""
+
+    head, tail = themes[0], themes[1:]
+    if_statement = f"""
+        if (theme == "{str(head[0].version)}") {{
+            var theme_css = `{head[1]._get_theme_css()}`
+        }} {" ".join(make_else_if(t) for t in tail)}
+    """
+
+    latest_to_oldest = sorted([t[0] for t in themes], key=lambda asset: asset.version)[
+        ::-1
+    ]
+    latest_to_oldest = [str(t.version) for t in latest_to_oldest]
+
+    component = gr.Dropdown(
+        choices=latest_to_oldest,
+        value=latest_to_oldest[0],
+        render=False,
+        label="Select Version",
+    ).style(container=False)
+
+    return (
+        component,
+        f"""
+        (theme) => {{
+            if (!document.querySelector('.theme-css')) {{
+                var theme_elem = document.createElement('style');
+                theme_elem.classList.add('theme-css');
+                document.head.appendChild(theme_elem);
+            }} else {{
+                var theme_elem = document.querySelector('.theme-css');
+            }}
+            {if_statement}
+            theme_elem.innerHTML = theme_css;
+        }}
+    """,
+    )