Rename app_5.py to app_6.py

app_5.py → app_6.py

@@ -137,12 +137,21 @@ def extract_arguments(text, tokenizer, model, beam_search=True):
         """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:
 
+            st.write(f"Start: {start_token}, End: {end_token}")
+            
+
+            if start_token > end_token:
+                return None
+                
+            if 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:
+                    st.write(f"Start_idx: {start_idx}, End_idx: {end_idx}")
+                    
                     return (
                         original_text[:start_idx]
                         + f"<mark style='background-color:{color}; padding:2px; border-radius:4px;'>"
@@ -151,7 +160,7 @@ def extract_arguments(text, tokenizer, model, beam_search=True):
                         + original_text[end_idx:]
                     )
     
-        return None       
+        return original_text