Rename app.py to app_1.py

app.py → app_1.py

@@ -208,16 +208,32 @@ if st.button("Extract"):
     if input_text:
         start_cause1, end_cause1, start_cause2, end_cause2, start_effect1, end_effect1, start_effect2, end_effect2, start_signal, end_signal = extract_arguments(input_text, tokenizer, model, beam_search=beam_search)
 
-        cause_text = mark_text_by_position(input_text, start_cause_id, end_cause_id, "#FFD700")  # Gold for cause
-        effect_text = mark_text_by_position(input_text, start_effect_id, end_effect_id, "#90EE90")  # Light green for effect
-        signal_text = mark_text_by_position(input_text, start_signal_id, end_signal_id, "#FF6347")  # Tomato red for signal
+        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
+
+        # Display first relation
+        st.markdown(f"<strong>Relation 1:</strong>", unsafe_allow_html=True)
+        st.markdown(f"**Cause:** {cause_text1}", unsafe_allow_html=True)
+        st.markdown(f"**Effect:** {effect_text1}", unsafe_allow_html=True)
+        st.markdown(f"**Signal:** {signal_text}", unsafe_allow_html=True)
+
+        # Display second relation if beam search is enabled
+        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")
+
+            st.markdown(f"<strong>Relation 2:</strong>", unsafe_allow_html=True)
+            st.markdown(f"**Cause:** {cause_text2}", unsafe_allow_html=True)
+            st.markdown(f"**Effect:** {effect_text2}", unsafe_allow_html=True)
+            st.markdown(f"**Signal:** {signal_text}", unsafe_allow_html=True)
 
-        st.markdown(f"**Cause:**<br>{cause_text}", unsafe_allow_html=True)
-        st.markdown(f"**Effect:**<br>{effect_text}", unsafe_allow_html=True)
-        st.markdown(f"**Signal:**<br>{signal_text}", unsafe_allow_html=True)
     else:
         st.warning("Please enter some text before extracting.")
 
+
+        
+
     
 if st.button("Extract1"):
     if input_text: