Update app.py

app.py

@@ -97,20 +97,20 @@ body {
 
 # Interface using Blocks context
 with gr.Blocks() as demo:
-    # Add your components including buttons
-    gr.Textbox(value="You are a virtual health assistant...", label="System message", visible=False)
-    
+    # Add all your components here, including buttons
+    system_message = gr.Textbox(value="You are a virtual health assistant...", label="System message", visible=False)
+    message_input = gr.Textbox(label="User message")
+    history = gr.State([])  # Keep the history of interactions
+    max_tokens_slider = gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens")
+    temperature_slider = gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature")
+    top_p_slider = gr.Slider(minimum=0.1, maximum=1.0, value=0.95, step=0.05, label="Top-p (nucleus sampling)")
+
     # Buttons to select the model
-    gr.Button("Chatgpt").click(on_button_click, inputs=["Chatgpt", "textbox", "history", "system_message", "max_tokens", "temperature", "top_p"], outputs="text")
-    gr.Button("Llama").click(on_button_click, inputs=["Llama", "textbox", "history", "system_message", "max_tokens", "temperature", "top_p"], outputs="text")
-    gr.Button("Claude").click(on_button_click, inputs=["Claude", "textbox", "history", "system_message", "max_tokens", "temperature", "top_p"], outputs="text")
-    
-    # Sliders for max tokens, temperature, and top-p
-    gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens", visible=False)
-    gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature", visible=False)
-    gr.Slider(minimum=0.1, maximum=1.0, value=0.95, step=0.05, label="Top-p (nucleus sampling)", visible=False)
+    gr.Button("Chatgpt").click(on_button_click, inputs=[message_input, history, system_message, max_tokens_slider, temperature_slider, top_p_slider], outputs="text")
+    gr.Button("Llama").click(on_button_click, inputs=[message_input, history, system_message, max_tokens_slider, temperature_slider, top_p_slider], outputs="text")
+    gr.Button("Claude").click(on_button_click, inputs=[message_input, history, system_message, max_tokens_slider, temperature_slider, top_p_slider], outputs="text")
     
-    # Apply the custom CSS for styling
+    # Optional: customize your layout with CSS if needed
     demo.css = css
 
 # Launch the app