Rename app_6.py to app_7.py

app_6.py → app_7.py

@@ -163,10 +163,18 @@ def extract_arguments(text, tokenizer, model, beam_search=True):
     list2 = [start_cause2, end_cause2, start_effect2, end_effect2, start_signal, end_signal]
     return cause1, cause2, effect1, effect2, signal, list1, list2
 
+
 def mark_text(original_text, span, color):
-    """Replace extracted span with a colored background marker."""
+    """Replace extracted span with a colored background marker, starting at the exact first token."""
     if span:
-        return re.sub(re.escape(span), f"<mark style='background-color:{color}; padding:2px; border-radius:4px;'>{span}</mark>", original_text, flags=re.IGNORECASE)
+        # Get the starting index of the span in the original text
+        start_idx = original_text.find(span)
+        
+        # Ensure the span is found and replace it with <mark> tags directly at the first token's position
+        if start_idx != -1:
+            # Replace the span with the mark tag at the exact first token position
+            return re.sub(re.escape(span), f"<mark style='background-color:{color}; padding:2px; border-radius:4px;'>{span}</mark>", original_text, flags=re.IGNORECASE)
+        
     return original_text  # Return unchanged text if no span is found
 
 st.title("Causal Relation Extraction")