Update css

app.py

@@ -1,10 +1,12 @@
 from pathlib import Path
 
-from htmltools import HTMLDependency
+from htmltools import HTMLDependency, tags
 from shiny import App, reactive, ui
 
 from query import query_output_server, query_output_ui
 
+button_style = {"style": "margin: 15px"}
+
 www_dir = Path(__file__).parent / "www"
 app_ui = ui.page_fluid(
     HTMLDependency(
@@ -17,9 +19,14 @@ app_ui = ui.page_fluid(
     ui.row(
         ui.column(
             2,
-            ui.br(),
-            ui.br(),
-            ui.input_action_button("add_query", "Add Query"),
+            ui.row(
+                button_style,
+                ui.input_action_button("add_query", "Add Query"),
+            ),
+            ui.row(
+                button_style,
+                ui.input_action_button("remove_query", "Remove Query"),
+            ),
         ),
         ui.column(
             10,
@@ -45,5 +52,10 @@ def server(input, output, session):
         )
         query_output_server(id)
 
+    @reactive.Effect
+    @reactive.event(input.remove_query)
+    def _():
+        ui.remove_ui(selector=f"#module_container .row:first-child")
+
 
 app = App(app_ui, server)

query.py

@@ -9,6 +9,7 @@ from shiny import module, reactive, render, ui
 @module.ui
 def query_output_ui():
     out = ui.row(
+        {"style": "border: 1px solid gray; border-radius: 5px; margin:10px"},
         ui.column(
             4,
             ui.input_text("prompt", "Prompt", placeholder="Enter query"),