Refactor Gradio interface and add parsed test cases

app.py

@@ -36,18 +36,21 @@ def launch_gradio_widget(metric, test_cases):
         (feature_names, feature_types) = zip(*metric.features.items())
     gradio_input_types = infer_gradio_input_types(feature_types)
 
+    parsed_test_cases = parse_test_cases(test_cases, feature_names, gradio_input_types)
+
     def compute(data):
         return metric.compute(**parse_gradio_data(data, gradio_input_types))
 
-    iface = gr.Interface(
+    demo = gr.Interface(
         fn=compute,
-        inputs=gr.inputs.Dataframe(
+        inputs=gr.Dataframe(
+            value=parsed_test_cases[0],
             headers=feature_names,
             col_count=len(feature_names),
-            row_count=5,
             datatype=json_to_string_type(gradio_input_types),
+            # row_count=5,
         ),
-        outputs=gr.outputs.Textbox(label=metric.name),
+        outputs=gr.Textbox(label=metric.name),
         description=(
             metric.info.description
             + "\nISCO codes must be wrapped in double quotes."
@@ -56,11 +59,15 @@ def launch_gradio_widget(metric, test_cases):
         title=f"Metric: {metric.name}",
         article=parse_readme(local_path / "README.md"),
         # TODO: load test cases and use them to populate examples
-        examples=[parse_test_cases(test_cases, feature_names, gradio_input_types)],
+        examples=[parsed_test_cases],
     )
-    iface.launch()
+    if __name__ == "__main__":
+        demo.launch()
+    else:
+        return demo
 
 
 module = evaluate.load("danieldux/isco_hierarchical_accuracy")
 
-launch_gradio_widget(module, test_cases)
+if __name__ == "__main__":
+    launch_gradio_widget(module, test_cases)