Rename app_18.py to app_19.py

app_18.py → app_19.py

@@ -163,25 +163,32 @@ def extract_arguments(text, tokenizer, model, beam_search=True):
 
     # Add the argument tags in the sentence directly
     def add_tags(original_text, word_ids, start_cause, end_cause, start_effect, end_effect, start_signal, end_signal):
-        
         space_splitted_tokens = original_text.split(" ")
-        
-        
-        if beam_search:
-            if start_cause is not None and end_cause is not None and start_effect is not None and end_effect is not None:
-                this_space_splitted_tokens = copy.deepcopy(space_splitted_tokens)
-                this_space_splitted_tokens[word_ids[start_cause]] = '<ARG0>' + this_space_splitted_tokens[word_ids[start_cause]]
-                this_space_splitted_tokens[word_ids[end_cause]] = this_space_splitted_tokens[word_ids[end_cause]] + '</ARG0>'
-                this_space_splitted_tokens[word_ids[start_effect]] = '<ARG1>' + this_space_splitted_tokens[word_ids[start_effect]]
-                this_space_splitted_tokens[word_ids[end_effect]] = this_space_splitted_tokens[word_ids[end_effect]] + '</ARG1>'
-
-            if has_signal:
-                this_space_splitted_tokens[word_ids[start_signal]] = '<SIG0>' + this_space_splitted_tokens[word_ids[start_signal]]
-                this_space_splitted_tokens[word_ids[end_signal]] = this_space_splitted_tokens[word_ids[end_signal]] + '</SIG0>'
+        this_space_splitted_tokens = copy.deepcopy(space_splitted_tokens)
     
-        # Join the tokens back into a single string
+        def safe_insert(tag, position, start=True):
+            """Safely insert a tag, checking for None values."""
+            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
+    
+        # Add argument tags safely
+        safe_insert('<ARG0>', start_cause, start=True)
+        safe_insert('</ARG0>', end_cause, start=False)
+        safe_insert('<ARG1>', start_effect, start=True)
+        safe_insert('</ARG1>', end_effect, start=False)
+    
+        # Add signal tags safely (if signal exists)
+        if start_signal is not None and end_signal is not None:
+            safe_insert('<SIG0>', start_signal, start=True)
+            safe_insert('</SIG0>', end_signal, start=False)
+    
+        # Join tokens back into a string
         return ' '.join(this_space_splitted_tokens)
 
+
     # Apply the tags to the sentence tokens
     tagged_sentence1 = add_tags(input_text, word_ids, start_cause1, end_cause1, start_effect1, end_effect1, start_signal, end_signal)
     tagged_sentence2 = add_tags(input_text, word_ids, start_cause2, end_cause2, start_effect2, end_effect2, start_signal, end_signal)