fix feature mapping selection bug

app.py

@@ -1,14 +1,10 @@
 import atexit
-import threading
-
 import gradio as gr
 
 from app_leaderboard import get_demo as get_demo_leaderboard
 from app_text_classification import get_demo as get_demo_text_classification
 from run_jobs import start_process_run_job, stop_thread
 
-if threading.current_thread() is not threading.main_thread():
-    t = threading.current_thread()
 try:
     with gr.Blocks(theme=gr.themes.Soft(primary_hue="green")) as demo:
         with gr.Tab("Text Classification"):
@@ -22,6 +18,6 @@ try:
     demo.launch(share=False)
     atexit.register(stop_thread)
 
-except Exception:
-    print("stop background thread")
+except Exception as e:
+    print("stop background thread: ", e)
     stop_thread()

app_text_classification.py

@@ -119,6 +119,19 @@ def get_demo(demo):
         ],
     )
 
+    # label.change sometimes does not pass the changed value
+    gr.on(
+        triggers=[label.input for label in column_mappings],
+        fn=write_column_mapping_to_config,
+        inputs=[
+            dataset_id_input,
+            dataset_config_input,
+            dataset_split_input,
+            uid_label,
+            *column_mappings,
+        ],
+    )
+
     gr.on(
         triggers=[
             model_id_input.change,

io_utils.py

@@ -120,6 +120,8 @@ def pop_job_from_pipe():
     job_info = pipe.jobs.pop()
     write_log_to_user_file(job_info[0], f"Running job id {job_info[0]}\n")
     command = job_info[1]
+    print(f"Running job id {job_info[0]}")
+    print(f"Running command {command}")
 
     log_file = open(f"./tmp/{job_info[0]}_log", "a")
     subprocess.Popen(

run_jobs.py

@@ -1,6 +1,6 @@
 import threading
 import time
-
+import logging
 import pipe
 from io_utils import pop_job_from_pipe
 
@@ -9,7 +9,7 @@ is_running = False
 
 def start_process_run_job():
     try:
-        print("Running jobs in thread")
+        logging.debug("Running jobs in thread")
         global thread, is_running
         thread = threading.Thread(target=run_job)
         thread.daemon = True
@@ -22,7 +22,7 @@ def start_process_run_job():
 
 
 def stop_thread():
-    print("Stop thread")
+    logging.debug("Stop thread")
     global is_running
     is_running = False
 
@@ -34,6 +34,6 @@ def run_job():
             pop_job_from_pipe()
             time.sleep(10)
         except KeyboardInterrupt:
-            print("KeyboardInterrupt stop background thread")
+            logging.debug("KeyboardInterrupt stop background thread")
             is_running = False
             break

text_classification_ui_helpers.py

@@ -24,7 +24,7 @@ HF_WRITE_TOKEN = "HF_WRITE_TOKEN"
 
 def check_dataset_and_get_config(dataset_id, uid):
     try:
-        write_column_mapping(None, uid)  # reset column mapping
+        # write_column_mapping(None, uid)  # reset column mapping
         configs = datasets.get_dataset_config_names(dataset_id)
         return gr.Dropdown(configs, value=configs[0], visible=True)
     except Exception:
@@ -52,20 +52,17 @@ def write_column_mapping_to_config(
     )
     if labels is None:
         return
-    labels = [*labels]
-    all_mappings = read_column_mapping(uid)
 
-    if all_mappings is None:
-        all_mappings = dict()
+    all_mappings = dict()
 
     if "labels" not in all_mappings.keys():
         all_mappings["labels"] = dict()
     for i, label in enumerate(labels[:MAX_LABELS]):
         if label:
-            all_mappings["labels"][label] = ds_labels[i]
+            all_mappings["labels"][label] = ds_labels[i%len(ds_labels)]
     if "features" not in all_mappings.keys():
         all_mappings["features"] = dict()
-    for i, feat in enumerate(labels[MAX_LABELS : (MAX_LABELS + MAX_FEATURES)]):
+    for _, feat in enumerate(labels[MAX_LABELS : (MAX_LABELS + MAX_FEATURES)]):
         if feat:
             # TODO: Substitute 'text' with more features for zero-shot
             all_mappings["features"]["text"] = feat