paper_version

remove chuncks
add judge selection
more description

app.py

@@ -11,74 +11,89 @@ for index, row in df[df.category == "per_curiam"].iterrows():
     if len(row["text"]) > 1000:
         choices.append((f"""{row["case_name"]}""", [row["text"], row["year_filed"]]))
 
-max_textboxes = 100
+unique_judges_by_year = (
+    df[df.author_name != "per_curiam"].groupby("year_filed")["author_name"].unique()
+)
+additional_judges = ["Justice Breyer", "Justice Kennedy"]
+unique_judges_by_year[1994] = list(unique_judges_by_year[1994]) + additional_judges
 
 
 # https://www.gradio.app/guides/controlling-layout
-def greet(opinion, year):
-    judges_l = (
-        df[(df["year_filed"] == year) & (df["category"] != "per_curiam")]
-        .author_name.unique()
-        .tolist()
-    )
-
-    if year == 1994:
-        judges_l.extend(["Justice Breyer", "Justice Kennedy"])
-
+def greet(opinion, judges_l):
     chunks = chunk_data(remove_citations(opinion))["text"].to_list()
     result = average_text(chunks, pipe, judges_l)
-    k = len(chunks)
-
-    wrt_boxes = []
-    for i in range(k):
-        wrt_boxes.append(gr.Textbox(chunks[i], visible=True))
-        wrt_boxes.append(gr.Label(value=result[1][i], visible=True))
-    return (
-        [result[0]]
-        + wrt_boxes
-        + [gr.Textbox(visible=False), gr.Label(visible=False)] * (max_textboxes - k)
-    )
+
+    return result[0]
 
 
 def set_input(drop):
-    return drop[0], drop[1], gr.Slider(visible=False)
+    return drop[0], drop[1], gr.Slider(visible=True)
+
+
+def update_year(year):
+    return gr.CheckboxGroup(
+        unique_judges_by_year[year].tolist(),
+        value=unique_judges_by_year[year].tolist(),
+        label="Select Judges",
+    )
 
 
 with gr.Blocks() as demo:
     with gr.Row():
         with gr.Column():
-            opinion = gr.Textbox(label="Opinion")
-            year = gr.Slider(1994, 2020, step=1, label="Year")
-            drop = gr.Dropdown(choices=sorted(choices))
+            drop = gr.Dropdown(
+                choices=sorted(choices),
+                label="Per Curiam Opinions",
+                info="Select a per curiam opinion to use as input",
+            )
+            year = gr.Slider(
+                1994,
+                2020,
+                step=1,
+                label="Year",
+                info="Select the year of the opinion if you manually pass the opinion below",
+            )
+            exc_judg = gr.CheckboxGroup(
+                unique_judges_by_year[year.value],
+                value=unique_judges_by_year[year.value],
+                label="Select Judges",
+                info="Select judges to consider in prediction",
+            )
+
+            opinion = gr.Textbox(
+                label="Opinion", info="Paste opinion text here or select from dropdown"
+            )
+        with gr.Column():
             with gr.Row():
                 clear_btn = gr.Button("Clear")
                 greet_btn = gr.Button("Predict")
-        op_level = gr.outputs.Label(num_top_classes=13, label="Overall")
-
-    textboxes = []
-    for i in range(max_textboxes):
-        with gr.Row():
-            t = gr.Textbox(f"Textbox {i}", visible=False, label=f"Paragraph {i+1} Text")
-            par_level = gr.Label(
-                num_top_classes=5, label=f"Paragraph {i+1} Prediction", visible=False
+            op_level = gr.outputs.Label(
+                num_top_classes=9, label="Predicted author of opinion"
             )
-        textboxes.append(t)
-        textboxes.append(par_level)
 
+    year.release(
+        update_year,
+        inputs=[year],
+        outputs=[exc_judg],
+    )
+    year.change(
+        update_year,
+        inputs=[year],
+        outputs=[exc_judg],
+    )
     drop.select(set_input, inputs=drop, outputs=[opinion, year, year])
 
     greet_btn.click(
         fn=greet,
-        inputs=[opinion, year],
-        outputs=[op_level] + textboxes,
+        inputs=[opinion, exc_judg],
+        outputs=[op_level],
     )
 
     clear_btn.click(
-        fn=lambda: [None, 1994, gr.Slider(visible=True), None, None]
-        + [gr.Textbox(visible=False), gr.Label(visible=False)] * max_textboxes,
-        outputs=[opinion, year, year, drop, op_level] + textboxes,
+        fn=lambda: [None, 1994, gr.Slider(visible=True), None, None],
+        outputs=[opinion, year, year, drop, op_level],
     )
 
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch(debug=True)