change UI

app.py

@@ -194,6 +194,7 @@ ui_design = UI_design()
 with gr.Blocks(css=custom_css) as demo:
     gr.Markdown("<h1 style='text-align: center;'> 😸 Meowthamatical AI Chatbot 😸</h1>")
     gr.Markdown(" Interact with the AI chatbot using customizable settings below.")
+    use_local_model = gr.Checkbox(label="Use Local Model", value=False)
 
     with gr.Row():
         with gr.Column():
@@ -204,31 +205,22 @@ with gr.Blocks(css=custom_css) as demo:
             input_text = gr.Textbox(label="input your question")
 
             with gr.Row():
-                # with gr.Column():
-                #     clear_btn = gr.ClearButton(
-                #         [input_sketchpad, input_text])
                 with gr.Column():
-                    submit_btn = gr.Button("Submit", variant="primary") 
-
-    with gr.Row():
-        system_message = gr.Textbox(value= "You are a cat and you love talking about and teaching math.", label="System message", interactive=True)
-        use_local_model = gr.Checkbox(label="Use Local Model", value=False)
-        button_1 = gr.Button("Submit", variant="primary")
-    with gr.Row():
-        max_tokens = gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens")
-        temperature = gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature")
-        top_p = gr.Slider(minimum=0.1, maximum=1.0, value=0.95, step=0.05, label="Top-p (nucleus sampling)")
-
-    chat_history = gr.Chatbot(label="Chat")
+                    clear_btn = gr.ClearButton(
+                        [input_sketchpad, input_text])
+                with gr.Column():
+                    submit_btn = gr.Button("Submit", variant="primary")
+        with gr.Column():
+            chat_history = gr.Chatbot(label="Chat")
 
-    user_input = gr.Textbox(show_label=False, placeholder="Type your message here...")
+            user_input = gr.Textbox(show_label=False, placeholder="Type your message here...")
 
-    cancel_button = gr.Button("Cancel Inference", variant="danger")
+            # cancel_button = gr.Button("Cancel Inference", variant="danger")
 
     # Adjusted to ensure history is maintained and passed correctly
-    user_input.submit(respond, [user_input, chat_history, system_message, max_tokens, temperature, top_p, use_local_model], chat_history)
+    user_input.submit(respond, [user_input, chat_history, use_local_model], chat_history)
 
-    cancel_button.click(cancel_inference)
+    # cancel_button.click(cancel_inference)
 
 if __name__ == "__main__":
     demo.launch(share=False)  # Remove share=True because it's not supported on HF Spaces