Rename app_3.py to app_4.py

app_3.py → app_4.py

@@ -159,13 +159,18 @@ 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
+    
     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")
-    
+        
+    if beam_search:
+        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")
+    else:
+        cause_text2 = None
+        effect_text2 = None
     return cause_text1, effect_text1, signal_text, cause_text2, effect_text2
 
 st.title("Causal Relation Extraction")