modify interface

app.py

@@ -67,9 +67,15 @@ def generate_response(message, history, system_message, max_tokens, temperature,
 
     return assistant_response
 
+def user(user_message, history):
+    return "", history + [[user_message, None]]
 
-def update_advanced_settings(show_advanced):
-    return {"visible": show_advanced}
+def bot(history, system_prompt, max_length, temperature, top_p, advanced_checkbox, use_pipeline, tool_definition):
+    user_message = history[-1][0]
+    do_sample = advanced_checkbox 
+    bot_message = generate_response(user_message, history, system_prompt, max_length, temperature, top_p, do_sample, use_pipeline, tool_definition)
+    history[-1][1] = bot_message
+    return history
 
 with gr.Blocks() as demo:
     with gr.Row(): 
@@ -88,15 +94,13 @@ with gr.Blocks() as demo:
             system_prompt = gr.TextArea(label="📑Context", placeholder="add context here...", lines=5)
             user_input = gr.TextArea(label="🤷🏻‍♂️User Input", placeholder="Hi there my name is Tonic!", lines=2)
             advanced_checkbox = gr.Checkbox(label="🧪 Advanced Settings", value=False)
-            advanced_settings = gr.Column(visible=False)
-            with advanced_settings:
+            with gr.Column(visible=False) as advanced_settings:
                 max_length = gr.Slider(label="📏Max Length", minimum=12, maximum=1700, value=650, step=1)
                 temperature = gr.Slider(label="🌡️Temperature", minimum=0.01, maximum=1.0, value=0.7, step=0.01)
                 top_p = gr.Slider(label="⚛️Top-p (Nucleus Sampling)", minimum=0.1, maximum=1.0, value=0.9, step=0.01)
                 use_pipeline = gr.Checkbox(label="Use Pipeline", value=False)
                 use_tool = gr.Checkbox(label="Use Function Calling", value=False)
-                tool_options = gr.Column(visible=False)
-                with tool_options:
+                with gr.Column(visible=False) as tool_options:
                     tool_definition = gr.Code(
                         label="Tool Definition (JSON)",
                         value=customtool,
@@ -109,16 +113,6 @@ with gr.Blocks() as demo:
         with gr.Column(scale=2):
             chatbot = gr.Chatbot(label="🤖Mistral-NeMo-Minitron")
     
-    def user(user_message, history):
-        return "", history + [[user_message, None]]
-
-    def bot(history, system_prompt, max_length, temperature, top_p, advanced_checkbox, use_pipeline, tool_definition):
-        user_message = history[-1][0]
-        do_sample = advanced_checkbox 
-        bot_message = generate_response(user_message, history, system_prompt, max_length, temperature, top_p, do_sample, use_pipeline, tool_definition)
-        history[-1][1] = bot_message
-        return history
-
     generate_button.click(
         user,
         [user_input, chatbot],
@@ -131,9 +125,9 @@ with gr.Blocks() as demo:
     )
 
     advanced_checkbox.change(
-        fn=update_advanced_settings,
+        fn=lambda x: gr.update(visible=x),
         inputs=[advanced_checkbox],
-        outputs=advanced_settings
+        outputs=[advanced_settings]
     )
 
     use_tool.change(
@@ -144,4 +138,4 @@ with gr.Blocks() as demo:
 
 if __name__ == "__main__":
     demo.queue()
-    demo.launch()
+    demo.launch()