Update app.py

app.py

@@ -30,7 +30,7 @@ def main():
     st.title("Terms of Service Summarizer")
 
     # Layout: 3 columns
-    col1, col2, col3 = st.columns([1, 3, 2], gap="large")
+    col1, col2, col3 = st.columns([1, 2, 3], gap="large")
 
     # Left column: Radio buttons for summarizer choice
     with col1:
@@ -45,7 +45,7 @@ def main():
                 "Keyphrase Extraction: Keyphrase extraction is similar to keyword extraction but focuses on identifying multi-word phrases or expressions that are significant in the text using the Rake algorithm."
         
         radio_selection = st.radio("Choose type of summarizer:", radio_options, help=help_text)
-                                   
+
     # Middle column: Text input and File uploader
     with col2:
         user_input = st.text_area("Enter your text here:")
@@ -109,12 +109,27 @@ def main():
             if radio_selection == "Keyphrase Extraction (RAKE)":
                 summary = extract_sentences_with_obligations(file_content)
                 st.session_state.summary = summary
-
+    
     # Right column: Displaying text after pressing 'Summarize'
     with col3:
         st.write("Summary:")
         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]:
+                # Fetch the reference summary
+                reference_summary = data['train'][tos_selection_index]['summary']
+    
+                # Calculate ROUGE scores
+                rouge = Rouge()
+                scores = rouge.get_scores(st.session_state.summary, reference_summary)
+    
+                # Display ROUGE scores
+                st.write("ROUGE Scores:")
+                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()