Rename app_4.py to app_5.py

app_4.py → app_5.py

@@ -136,27 +136,25 @@ def extract_arguments(text, tokenizer, model, beam_search=True):
     def mark_text_by_position(original_text, start_token, end_token, color):
         """Marks text in the original string based on character positions."""
         # Inserts tags into the original text based on token offsets.
+        if start_token is not None and end_token is not None:
+            if start_token <= end_token:
 
-        if start_token is not None and end_token is not None and start_token <= end_token:
         
-            start_idx, end_idx = offset_mapping[start_token][0], offset_mapping[end_token][1]
-             
-            if start_idx is not None and end_idx is not None and start_idx < end_idx:
-                return (
-                    original_text[:start_idx]
-                    + f"<mark style='background-color:{color}; padding:2px; border-radius:4px;'>"
-                    + original_text[start_idx:end_idx]
-                    + "</mark>"
-                    + original_text[end_idx:]
-                )
+                start_idx, end_idx = offset_mapping[start_token][0], offset_mapping[end_token][1]
+                 
+                if start_idx is not None and end_idx is not None and start_idx < end_idx:
+                    return (
+                        original_text[:start_idx]
+                        + f"<mark style='background-color:{color}; padding:2px; border-radius:4px;'>"
+                        + original_text[start_idx:end_idx]
+                        + "</mark>"
+                        + original_text[end_idx:]
+                    )
+    
+        return None       
     
-        if start_token > end_token:
-            st.write("The prediction is not correct: The position of the predicted end token comes before the position of the start token.")
-            
         
-        return original_text 
         
-
     cause_text1 = mark_text_by_position(input_text, start_cause1, end_cause1, "#FFD700")  # Gold for cause
     effect_text1 = mark_text_by_position(input_text, start_effect1, end_effect1, "#90EE90")  # Light green for effect
     
@@ -184,10 +182,15 @@ if st.button("Extract"):
 
         # Display first relation
         st.markdown(f"<strong>Relation 1:</strong>", unsafe_allow_html=True)
-        st.markdown(f"**Cause:** {cause_text1}", unsafe_allow_html=True)
-        st.markdown(f"**Effect:** {effect_text1}", unsafe_allow_html=True)
-        st.markdown(f"**Signal:** {signal_text}", unsafe_allow_html=True)
 
+        if cause_text1 is None or effect_text1 is None: 
+            st.write("The prediction is not correct for at least one span: The position of the predicted end token comes before the position of the start token.")
+        else:
+            st.markdown(f"**Cause:** {cause_text1}", unsafe_allow_html=True)
+            st.markdown(f"**Effect:** {effect_text1}", unsafe_allow_html=True)
+            st.markdown(f"**Signal:** {signal_text}", unsafe_allow_html=True)
+
+        
         # Display second relation if beam search is enabled
         if beam_search: