Update app.py

app.py

@@ -4,13 +4,13 @@ import sys
 import argparse
 from utils import *
 from chat_func import *
-
+   
 my_api_key = os.environ.get('my_api_key')
 
 if my_api_key == "empty":
     print("Please give a api key!")
     sys.exit(1)
-
+    
 gr.Chatbot.postprocess = postprocess
 
 
@@ -18,7 +18,7 @@ with open("css_new.css", "r", encoding="utf-8") as f:
     css = f.read()
 
 with gr.Blocks(css=css) as demo:
-    
+    # ,theme='gradio/soft'
     history = gr.State([])
     token_count = gr.State([])
     promptTemplates = gr.State(load_template('myprompts.json', mode=2))
@@ -26,17 +26,22 @@ with gr.Blocks(css=css) as demo:
     TRUECOMSTANT = gr.State(True)
     FALSECONSTANT = gr.State(False)
     gr.Markdown(title)
-
-    with gr.Accordion("Build by [45度科研人](WeChat Public Accounts)", open=False):
-        gr.Markdown(description)
-        
+    
+    with gr.Row().style(equal_height=True):
+        with gr.Column(scale=12):
+            with gr.Accordion("Build by [45度科研人](WeChat Public Accounts)", open=False):
+                gr.Markdown(description)
+        with gr.Column(scale=1):
+            with gr.Box():
+                toggle_dark = gr.Button(value="Toggle Dark").style(full_width=True)
+   
     with gr.Row(scale=1).style(equal_height=True):
         with gr.Column(scale=5):
             with gr.Column():
                 chatbot = gr.Chatbot()
                 user_input = gr.Textbox(show_label=False, placeholder="Enter text and press submit", visible=True).style(container=False)
-                submitBtn = gr.Button("Submit", variant="primary" ).style(container=False)
-                emptyBtn = gr.Button("Restart Conversation",variant="secondary").style(container=False)
+                submitBtn = gr.Button("Submit",variant="primary").style(container=False)
+                emptyBtn = gr.Button("Restart Conversation",variant="secondary")
                 status_display = gr.Markdown("")
 
         with gr.Column():
@@ -69,6 +74,13 @@ with gr.Blocks(css=css) as demo:
         <img src="https://dunazo.oss-cn-beijing.aliyuncs.com/blog/shoukuanma222.png" style="margin-left:25px;width:170px;height:190px;">
     </div>
     """)
+    
+    toggle_dark.click(None,_js="""
+        () => {
+            document.body.classList.toggle('dark');
+            document.querySelector('gradio-app').style.backgroundColor = 'var(--color-background-primary)'
+        }""",)
+    
     keyTxt.submit(submit_key, keyTxt, [user_api_key, status_display])
     keyTxt.change(submit_key, keyTxt, [user_api_key, status_display])
     # Chatbot
@@ -86,4 +98,4 @@ with gr.Blocks(css=css) as demo:
 
 
 if __name__ == "__main__":
-    demo.queue().launch(debug=False,show_api=False) 
+    demo.queue().launch(debug=False,show_api=False)