fbng

app.py

@@ -1,8 +1,4 @@
 import nltk
-
-# Ensure the necessary tokenizer is downloaded
-nltk.download("punkt_tab")
-
 import streamlit as st
 import fitz  # PyMuPDF for PDF extraction
 import re
@@ -11,6 +7,8 @@ from sumy.nlp.tokenizers import Tokenizer
 from sumy.summarizers.lsa import LsaSummarizer
 from rouge_score import rouge_scorer  # For ROUGE score evaluation
 
+# Ensure the necessary tokenizer is downloaded
+nltk.download("punkt_tab")
 
 # Function to extract text from PDF
 def extract_text_from_pdf(uploaded_file):
@@ -84,11 +82,12 @@ if st.sidebar.button("Summarize"):
         st.sidebar.error("Please upload a file or enter text.")
         st.stop()
 
-    # Generate summary
-    summary = summarize_text(text, num_sentences=5)
-
-    # Calculate ROUGE score
-    rouge1, rouge2, rougeL = calculate_rouge(text, summary)
+    # Show loading animation
+    with st.spinner("Summarizing text... Please wait."):
+        # Generate summary
+        summary = summarize_text(text, num_sentences=5)
+        # Calculate ROUGE score
+        rouge1, rouge2, rougeL = calculate_rouge(text, summary)
 
     # Display summary in justified format
     st.subheader("📌 Summarized Text")