Bulk runner UX improvements

Only show required fields and list them at the top
Turn all output rows into check boxes
List "Output Text" and "Run URL" at the top
Default check "Output Text" and "Run URL"
Add button "Show All Columns" > tapping expands the fields

recipes/BulkRunner.py

@@ -1,6 +1,8 @@
+import datetime
 import io
 import typing
 
+from django.utils.timesince import timesince
 from fastapi import HTTPException
 from furl import furl
 from pydantic import BaseModel, Field
@@ -41,8 +43,18 @@ class RequestModel(BaseModel):
             """,
         )
 
-        input_columns: dict[str, str]
-        output_columns: dict[str, str]
+        input_columns: dict[str, str] = Field(
+            title="Input Column Mappings",
+            description="""
+For each input field in the Gooey.AI workflow, specify the column in your input data that corresponds to it.
+            """,
+        )
+        output_columns: dict[str, str] = Field(
+            title="Output Column Mappings",
+            description="""
+For each output field in the Gooey.AI workflow, specify the column name that you'd like to use for it in the output data.
+            """,
+        )
 
     class ResponseModel(BaseModel):
         output_documents: list[str]
@@ -144,45 +156,70 @@ def render_form_v2(self):
             """
         )
 
-        col1, col2 = st.columns(2)
+        visible_col1, visible_col2 = st.columns(2)
+        with st.expander("🤲 Show All Columns"):
+            hidden_col1, hidden_col2 = st.columns(2)
 
-        with col1:
+        with visible_col1:
             st.write("##### Inputs")
 
-            input_columns_old = st.session_state.pop("input_columns", {})
-            input_columns_new = st.session_state.setdefault("input_columns", {})
+        input_columns_old = st.session_state.pop("input_columns", {})
+        input_columns_new = st.session_state.setdefault("input_columns", {})
 
-            column_options = [None, *get_columns(files)]
-            for fields in (required_input_fields, optional_input_fields):
-                for field, title in fields.items():
+        column_options = [None, *get_columns(files)]
+        for fields, div in (
+            (required_input_fields, visible_col1),
+            (optional_input_fields, hidden_col1),
+        ):
+            for field, title in fields.items():
+                with div:
                     col = st.selectbox(
                         label="`" + title + "`",
                         options=column_options,
                         key="--input-mapping:" + field,
                         default_value=input_columns_old.get(field),
                     )
-                    if col:
-                        input_columns_new[field] = col
-                st.write("---")
+                if col:
+                    input_columns_new[field] = col
 
-        with col2:
+        with visible_col2:
             st.write("##### Outputs")
 
-            output_columns_old = st.session_state.pop("output_columns", {})
-            output_columns_new = st.session_state.setdefault("output_columns", {})
+        # only show the first output field by default, and hide others
+        try:
+            first_out_field = next(
+                field for field in output_fields if "output" in field
+            )
+        except StopIteration:
+            first_out_field = next(iter(output_fields))
+
+        visible_out_fields = {
+            first_out_field: output_fields[first_out_field],
+            "run_url": "Run URL",
+        }
+        hidden_out_fields = {
+            k: v for k, v in output_fields.items() if k not in visible_out_fields
+        } | {
+            "run_time": "Run Time",
+            "error_msg": "Error Msg",
+        }
 
-            prev_fields = st.session_state.get("--prev-output-fields")
-            fields = {**output_fields, "error_msg": "Error Msg", "run_url": "Run URL"}
-            did_change = prev_fields is not None and prev_fields != fields
-            st.session_state["--prev-output-fields"] = fields
+        output_columns_old = st.session_state.pop("output_columns", {})
+        output_columns_new = st.session_state.setdefault("output_columns", {})
+
+        for fields, div, checked in (
+            (visible_out_fields, visible_col2, True),
+            (hidden_out_fields, hidden_col2, False),
+        ):
             for field, title in fields.items():
-                col = st.text_input(
-                    label="`" + title + "`",
-                    key="--output-mapping:" + field,
-                    value=output_columns_old.get(field, title if did_change else None),
-                )
+                with div:
+                    col = st.checkbox(
+                        label="`" + title + "`",
+                        key="--output-mapping:" + field,
+                        value=bool(output_columns_old.get(field, checked)),
+                    )
                 if col:
-                    output_columns_new[field] = col
+                    output_columns_new[field] = title
 
     def render_example(self, state: dict):
         render_documents(state)
@@ -238,6 +275,7 @@ def run_v2(
                     result.get(disable_sync_subtasks=False)
                     sr.refresh_from_db()
                     state = sr.to_dict()
+                    state["run_time"] = timesince(datetime.datetime.now() - sr.run_time)
                     state["run_url"] = sr.get_app_url()
                     state["error_msg"] = sr.error_msg
 

recipes/VideoBots.py

@@ -251,9 +251,7 @@ class RequestModel(BaseModel):
 
     class ResponseModel(BaseModel):
         final_prompt: str
-        raw_input_text: str | None
-        raw_output_text: list[str] | None
-        raw_tts_text: list[str] | None
+
         output_text: list[str]
 
         # tts
@@ -262,6 +260,11 @@ class ResponseModel(BaseModel):
         # lipsync
         output_video: list[str]
 
+        # intermediate text
+        raw_input_text: str | None
+        raw_tts_text: list[str] | None
+        raw_output_text: list[str] | None
+
         # doc search
         references: list[SearchReference] | None
         final_search_query: str | None