date improvements

app.py

@@ -50,17 +50,21 @@ def extract_features(tokens, labels):
                 # If there were any accumulated data so far
                 if current_date: 
                     merged_entities.append((current_date, 'date'))
-                else: 
-                    merged_entities.append((token, label))
-                current_date = ""
+                    print(f"Debug -- Date finalized: {current_date}")
+                    current_date = "" # Reset
+                
+                merged_entities.append((token, label))
         else: 
             # These are LABEL_0, supposedly trash but keep them for now
             if token: 
-                merged_entities.append((token, label))
-                current_date = ""
+                if current_date:  # If there's a leftover date fragment, add it first
+                    merged_entities.append((current_date, 'date'))
+                    print(f"Debug -- Finalizing leftover date added: {current_date}")
+                    current_date = ""  # Reset
 
     if current_date:
-        merged_entities.append(current_date, 'date')
+        print(f"Debug -- Dangling leftover date added: {current_date}")
+        merged_entities.append((current_date, 'date'))
 
     return merged_entities