format

app.py

@@ -44,7 +44,12 @@ def get_table_info(hub_repo_id):
         gr.Error(f"Error getting column info: {e}")
 
 
-def get_table_name(config: str | None, split: str | None, config_choices: list[str], split_choices: list[str]):
+def get_table_name(
+    config: str | None,
+    split: str | None,
+    config_choices: list[str],
+    split_choices: list[str],
+):
     if len(config_choices) > 0 and config is None:
         config = config_choices[0]
     if len(split_choices) > 0 and split is None:
@@ -63,16 +68,19 @@ def get_table_name(config: str | None, split: str | None, config_choices: list[s
         if c in ["-", "_", "/"]:
             return "_"
         return ""
-    
-    table_name = "".join(
-        replace_char(c) for c in base_name
-    )
+
+    table_name = "".join(replace_char(c) for c in base_name)
     if table_name[0].isdigit():
         table_name = f"_{table_name}"
     return table_name.lower()
 
 
-def get_prompt_messages(card_data: dict[str, Any], natural_language_query: str, config: str | None, split: str | None):
+def get_prompt_messages(
+    card_data: dict[str, Any],
+    natural_language_query: str,
+    config: str | None,
+    split: str | None,
+):
     config_choices = get_config_choices(card_data)
     split_choices = get_split_choices(card_data)
 
@@ -136,6 +144,7 @@ def query_dataset(hub_repo_id, card_data, query, config, split):
     duck_query = _sanitize_duck_query(duck_query)
     return duck_query, get_iframe(hub_repo_id, duck_query)
 
+
 def _sanitize_duck_query(duck_query: str) -> str:
     # Sometimes the LLM wraps the query like this:
     # ```sql
@@ -190,14 +199,18 @@ with gr.Blocks() as demo:
         except Exception:
             config_choices = []
             split_choices = []
-        
+
         initial_config = config_choices[0] if len(config_choices) > 0 else None
         initial_split = split_choices[0] if len(split_choices) > 0 else None
         with gr.Row():
             with gr.Column():
-                config_selection = gr.Dropdown(label="Config Name", choices=config_choices, value=initial_config)
+                config_selection = gr.Dropdown(
+                    label="Config Name", choices=config_choices, value=initial_config
+                )
             with gr.Column():
-                split_selection = gr.Dropdown(label="Split Name", choices=split_choices, value=initial_split)
+                split_selection = gr.Dropdown(
+                    label="Split Name", choices=split_choices, value=initial_split
+                )
 
         with gr.Row():
             with gr.Column():
@@ -223,5 +236,6 @@ with gr.Blocks() as demo:
             outputs=[sql_out, search_out],
         )
 
+
 if __name__ == "__main__":
     demo.launch()