Add system

app.py

@@ -175,7 +175,10 @@ def run_model():
             "events": events
         }
         default_log_path = f"{FEEDBACK_DIR}/feedback_{timestamp}_{user_id}.jsonl"
-
+        step_id = outputs["step_id"]
+        if update_status: 
+            update_status(step_id, "All done!")
+            step_id += 1
         with open(default_log_path, "w") as f:
             f.write(json.dumps(log_entry, indent=1))
 
@@ -188,7 +191,7 @@ def run_model():
                 repo_type="dataset",
                 token=HF_TOKEN
             )
-            # update_status(7, "✅ done")
+            
 
         except Exception as e:
             traceback.print_exc()

system/pledge_tracking.py

@@ -165,7 +165,7 @@ def run_pipeline(claim, pledge_date, pledge_author, start_date, timestamp, user_
     events_num = count_total_events(extracted_event_path)
 
     if update_fn: 
-        update_fn(step_id, f"We have extracted {events_num}")
+        update_fn(step_id, f"We have extracted {events_num} events from the documents.")
         step_id+=1
 
 
@@ -185,9 +185,9 @@ def run_pipeline(claim, pledge_date, pledge_author, start_date, timestamp, user_
     sorted_event_path = f"{pipeline_base_dir}/sorted_events.xlsx"
     df.to_excel(sorted_event_path, index=False)
 
-    if update_fn: 
-        update_fn(step_id, "All done!")
-        step_id += 1
+    # if update_fn: 
+    #     update_fn(step_id, "All done!")
+    #     step_id += 1
 
     return {
         "claim_json": claim_json_path,
@@ -199,6 +199,7 @@ def run_pipeline(claim, pledge_date, pledge_author, start_date, timestamp, user_
         "meta_data_dir": meta_data_dir,
         "unsorted_events": extracted_event_path,
         "sorted_events": sorted_event_path,
+        "step_id": step_id
     }