updated app.py

app.py

@@ -76,9 +76,12 @@ def init_files():
     sync_pull()
     for f in [LABEL_FILE, VERIFY_FILE, SKIP_FILE]:
         if not os.path.exists(f):
-            if f == LABEL_FILE: cols = ["timestamp", "user", "group_id", "url", "score", "label"]
-            elif f == VERIFY_FILE: cols = ["timestamp", "user", "group_id", "url", "is_correct", "corrected_label"]
-            else: cols = ["timestamp", "user", "group_id"]
+            if f == LABEL_FILE: 
+                cols = ["timestamp", "user", "group_id", "url", "score", "label"]
+            elif f == VERIFY_FILE: 
+                cols = ["timestamp", "user", "group_id", "url", "is_correct", "corrected_label", "corrected_score"]
+            else: 
+                cols = ["timestamp", "user", "group_id"]
             pd.DataFrame(columns=cols).to_csv(f, index=False)
             
     if not os.path.exists(URL_FILE):
@@ -151,7 +154,8 @@ def get_saved_values(gid, mode):
                 if mode == "label":
                     saved_data[row['url']] = {"score": row['score'], "label": row['label']}
                 else:
-                    saved_data[row['url']] = {"is_correct": row['is_correct'], "label": row['corrected_label']}
+                    sc = row['corrected_score'] if 'corrected_score' in row else 5
+                    saved_data[row['url']] = {"is_correct": row['is_correct'], "label": row['corrected_label'], "score": sc}
     except: pass
     return saved_data
 
@@ -194,12 +198,9 @@ def render_workspace(mode, history, specific_index=None, move_back=False):
         candidates = []
         for i, gid in enumerate(all_groups):
             if gid in s_done: continue
-            
             is_ready = False
-            if mode == "label" and gid not in l_done: 
-                is_ready = True
-            elif mode == "verify" and gid in l_done and gid not in v_done: 
-                is_ready = True
+            if mode == "label" and gid not in l_done: is_ready = True
+            elif mode == "verify" and gid in l_done and gid not in v_done: is_ready = True
             
             if is_ready:
                 candidates.append((i, gid))
@@ -254,9 +255,13 @@ def render_workspace(mode, history, specific_index=None, move_back=False):
                 updates[c_chk] = gr.update(visible=False)
                 updates[c_lbl] = gr.update(visible=False)
             else:
-                prev = r1_vals.get(u, {}).get('label', "Unknown")
-                updates[c_sld] = gr.update(visible=False)
-                updates[c_lbl] = gr.update(visible=True, value=f"**Labeled:** {prev}")
+                prev_lbl = r1_vals.get(u, {}).get('label', "Unknown")
+                prev_sc = r1_vals.get(u, {}).get('score', 0)
+                
+                display_score = v_sc if u in saved_vals else prev_sc
+                
+                updates[c_sld] = gr.update(visible=True, value=display_score) 
+                updates[c_lbl] = gr.update(visible=True, value=f"**Labeled:** {prev_lbl} | **Score:** {prev_sc}")
                 updates[c_drp] = gr.update(visible=True, value=v_lbl)
                 updates[c_chk] = gr.update(visible=True, value=v_chk)
         else:
@@ -275,8 +280,10 @@ def save_data(mode, history, urls, *args):
     for i, u in enumerate(urls):
         base = i * 4
         sc, lbl, chk = args[base], args[base+1], args[base+2]
-        if mode == "label": rows.append([ts, "user", gid, u, sc, lbl])
-        else: rows.append([ts, "user", gid, u, chk, lbl])
+        if mode == "label": 
+            rows.append([ts, "user", gid, u, sc, lbl])
+        else: 
+            rows.append([ts, "user", gid, u, chk, lbl, sc])
         
     fname = LABEL_FILE if mode == "label" else VERIFY_FILE
     with FileLock(LOCK_FILE):