Update app.py

app.py

@@ -117,8 +117,8 @@ def main():
         if 'summary' in st.session_state:
             st.write(st.session_state.summary)
 
-            # Check if a reference summary is available
-            if tos_selection_index is not None and 'summary' in dataset['train'][tos_selection_index]:
+            # Check if no PDF or text input is provided and a ToS document is selected
+            if not uploaded_file and not user_input and tos_selection_index is not None and 'summary' in dataset['train'][tos_selection_index]:
                 # Fetch the reference summary
                 reference_summary = dataset['train'][tos_selection_index]['summary']
     
@@ -126,12 +126,12 @@ def main():
                 rouge = Rouge()
                 scores = rouge.get_scores(st.session_state.summary, reference_summary)
     
-                 # Display ROUGE scores in a box
+                # Display ROUGE scores in a box
                 with st.container():
                     st.write("ROUGE Scores:")
-                    st.write(f"ROUGE-1: {scores[0]['rouge-1']['f']:.4f}", unsafe_allow_html=True)
-                    st.write(f"ROUGE-2: {scores[0]['rouge-2']['f']:.4f}", unsafe_allow_html=True)
-                    st.write(f"ROUGE-L: {scores[0]['rouge-l']['f']:.4f}", unsafe_allow_html=True)
+                    st.write(f"ROUGE-1: {scores[0]['rouge-1']['f']:.4f}")
+                    st.write(f"ROUGE-2: {scores[0]['rouge-2']['f']:.4f}")
+                    st.write(f"ROUGE-L: {scores[0]['rouge-l']['f']:.4f}")
 
 if __name__ == "__main__":
     main()