Rename app_19.py to app_20.py

app_19.py → app_20.py

@@ -167,13 +167,17 @@ def extract_arguments(text, tokenizer, model, beam_search=True):
         this_space_splitted_tokens = copy.deepcopy(space_splitted_tokens)
     
         def safe_insert(tag, position, start=True):
-            """Safely insert a tag, checking for None values."""
+            """Safely insert a tag, checking for None values and index validity."""
             if position is not None and word_ids[position] is not None:
-                if start:
-                    this_space_splitted_tokens[word_ids[position]] = tag + this_space_splitted_tokens[word_ids[position]]
-                else:
-                    this_space_splitted_tokens[word_ids[position]] += tag
-    
+                word_index = word_ids[position]
+                
+                # Ensure word_index is within range
+                if 0 <= word_index < len(this_space_splitted_tokens):
+                    if start:
+                        this_space_splitted_tokens[word_index] = tag + this_space_splitted_tokens[word_index]
+                    else:
+                        this_space_splitted_tokens[word_index] += tag
+            
         # Add argument tags safely
         safe_insert('<ARG0>', start_cause, start=True)
         safe_insert('</ARG0>', end_cause, start=False)