Update app_3.py

app_3.py

@@ -152,7 +152,10 @@ def extract_arguments(text, tokenizer, model, beam_search=True):
 
     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
-    signal_text = mark_text_by_position(input_text, start_signal, end_signal, "#FF6347")  # Tomato red for signal
+    if start_signal is not None and end_signal is not None:
+        signal_text = mark_text_by_position(input_text, start_signal, end_signal, "#FF6347")  # Tomato red for signal
+    else: 
+        signal_text = None
     cause_text2 = mark_text_by_position(input_text, start_cause2, end_cause2, "#FFD700")
     effect_text2 = mark_text_by_position(input_text, start_effect2, end_effect2, "#90EE90")