update ui

app.py

@@ -9,9 +9,10 @@ import time
 
 DESCRIPTION = '''# <a href="https://github.com/THUDM/VisualGLM">VisualGLM</a>'''
 
-MAINTENANCE_NOTICE='Hint 1: If the app report "Something went wrong, connection error out", please turn off your proxy and retry.\nHint 2: If you upload a large size of image like 10MB, it may take some time to upload and process. Please be patient and wait.'
+MAINTENANCE_NOTICE1 = 'Hint 1: If the app report "Something went wrong, connection error out", please turn off your proxy and retry.\nHint 2: If you upload a large size of image like 10MB, it may take some time to upload and process. Please be patient and wait.'
+MAINTENANCE_NOTICE2 = '提示1: 如果应用报了“Something went wrong, connection error out”的错误，请关闭代理并重试。\n提示2: 如果你上传了很大的图片，比如10MB大小，那将需要一些时间来上传和处理，请耐心等待。'
 
-NOTES = 'This app is adapted from <a href="https://github.com/THUDM/VisualGLM">https://github.com/THUDM/VisualGLM</a>. It would be recommended to check out the repo if you want to see the detail of our model and training process.'
+NOTES = 'This app is adapted from <a href="https://github.com/THUDM/VisualGLM">https://github.com/THUDM/VisualGLM</a>. It would be recommended to check out the repo if you want to see the detail of our model.'
 
 import json
 import requests
@@ -109,6 +110,13 @@ def clear_fn2(value):
     return [("", "Hi, What do you want to know about this image?")]
 
 
+def change_language(value):
+    if value == "Change hint to English":
+        return "提示变为中文", MAINTENANCE_NOTICE1
+    else:
+        return "Change hint to English", MAINTENANCE_NOTICE2
+
+
 def main():
     gr.close_all()
     examples = []
@@ -119,13 +127,9 @@ def main():
 
 
     with gr.Blocks(css='style.css') as demo:
-        
-        gr.Markdown(DESCRIPTION)
-        gr.Markdown(MAINTENANCE_NOTICE)
-
 
         with gr.Row():
-            with gr.Column():
+            with gr.Column(scale=4.5):
                 with gr.Group():
                     input_text = gr.Textbox(label='Input Text', placeholder='Please enter text prompt below and press ENTER.')
                     with gr.Row():
@@ -136,8 +140,15 @@ def main():
                 with gr.Row():
                     temperature = gr.Slider(maximum=1, value=0.95, minimum=0, label='Temperature')
                     top_p = gr.Slider(maximum=1, value=0.7, minimum=0, label='Top P')
+                with gr.Group():
+                    with gr.Row():
+                        with gr.Column(scale=7):    
+                            maintenance_notice = gr.Markdown(MAINTENANCE_NOTICE1)
+                        with gr.Column(scale=2):    
+                            change_button = gr.Button('Change hint to English', visible=False)
+            with gr.Column(scale=5.5):
+                result_text = gr.components.Chatbot(label='Multi-round conversation History', value=[("", "Hi, What do you want to know about this image?")]).style(height=550)
 
-            result_text = gr.components.Chatbot(label='Multi-round conversation History', value=[("", "Hi, What do you want to know about this image?")])
 
 
         gr_examples = gr.Examples(examples=[[example["text"], example["image"]] for example in examples], 
@@ -155,6 +166,8 @@ def main():
         clear_button.click(fn=clear_fn, inputs=clear_button, outputs=[input_text, result_text, image_prompt])
         image_prompt.change(fn=clear_fn2, inputs=clear_button, outputs=[result_text])
 
+        change_button.click(fn=change_language, inputs=change_button, outputs=[change_button, maintenance_notice])
+
         print(gr.__version__)