fix drop down

curated.py

@@ -20,8 +20,8 @@ data_sources = [
 ]
 
 
-def get_data(data_source: str = "Freelaw", data_ext_doc_id: int = 3, htmx=None):
-    data_ext_doc_id = max(0, min(int(data_ext_doc_id), 9))
+def get_data(data_source: str = "Freelaw", doc_id: int = 3):
+    doc_id = max(0, min(int(doc_id), 9))
 
     if data_source == "Freelaw":
         raw_sample_doc = json.load(open("data/curated_samples/freelaw_raw.json"))
@@ -57,8 +57,8 @@ def get_data(data_source: str = "Freelaw", data_ext_doc_id: int = 3, htmx=None):
     else:
         raw_sample_doc = extracted_sample_doc = [{} for _ in range(10)]
 
-    raw_json = raw_sample_doc[data_ext_doc_id]
-    extracted_json = extracted_sample_doc[data_ext_doc_id]
+    raw_json = raw_sample_doc[doc_id]
+    extracted_json = extracted_sample_doc[doc_id]
 
     drop_down = Select(
         *[Option(ds, value=ds, selected=(ds == data_source)) for ds in data_sources],
@@ -71,24 +71,24 @@ def get_data(data_source: str = "Freelaw", data_ext_doc_id: int = 3, htmx=None):
 
     slider = Input(
         type="range",
-        name="data_ext_doc_id",
+        name="doc_id",
         min="0",
         max="9",
-        value=str(data_ext_doc_id),
+        value=str(doc_id),
         hx_get="/curated",
         hx_target="#colcontent",
         hx_trigger="change",
         hx_swap="innerHTML",
-        oninput='document.getElementById("doc_id_value").innerText = "Selected document index: " + this.value',
+        hx_include="[name='data_source']",
     )
 
     form = Form(
         Div(
-            Label("Data source:", drop_down),
+            Label("Data source: ", drop_down),
             style="margin-bottom: 20px;",
         ),
         Div(
-            Label("Data sample:", slider, f"{data_ext_doc_id}"),
+            Label("Data sample: ", slider, f"{doc_id}"),
             style="margin-bottom: 20px;",
         ),
     )
@@ -111,5 +111,9 @@ def get_data(data_source: str = "Freelaw", data_ext_doc_id: int = 3, htmx=None):
         style="width: 48%; float: right; overflow-x: auto;",
     )
 
-    data_display = Div(col1, col2, style="overflow: auto; clear: both; height: 600px;")
-    return (Div(form, data_display, style="margin-top: 10px;", id="colcontent"),)
+    data_display = Div(
+        col1,
+        col2,
+        style="overflow: auto; clear: both; height: 600px; border: 1px solid #ccc; padding: 20px;",
+    )
+    return Div(form, data_display, style="margin-top: 10px;", id="colcontent")

main.py

@@ -311,12 +311,12 @@ def get_chart_28168342():
 def curated(request):
     from curated import get_data
 
-    # Partial Update
+    # Partial Updates
     params = request.query_params
     if data_source := params.get("data_source"):
-        return get_data(data_source=data_source)
-    if doc_id := params.get("data_ext_doc_id"):
-        return get_data(data_ext_doc_id=doc_id)
+        return get_data(data_source, params.get("doc_id", 3))
+    if doc_id := params.get("doc_id"):
+        return get_data(params.get("data_source"), doc_id)
 
     hr = HR()
     data_preparation_steps = pd.DataFrame(
@@ -345,9 +345,6 @@ def curated(request):
         }
     )
 
-    data_preparation_steps = data_preparation_steps.style.set_properties(
-        **{"padding": "10px"}
-    )
     table_html = data_preparation_steps.to_html(index=False, border=0)
     table_div = Div(NotStr(table_html), cls="l-body-outset")