Update app.py

app.py

@@ -104,13 +104,9 @@ from nltk.tokenize import sent_tokenize
 import torch
 from fastapi import FastAPI
 from fastapi.responses import RedirectResponse
-import os
 
 # Download required NLTK data
-try:
-    nltk.data.find('tokenizers/punkt')
-except LookupError:
-    nltk.download('punkt')
+nltk.download('punkt', quiet=True)
 
 # Initialize components
 app = FastAPI()
@@ -226,22 +222,23 @@ def generate_summary(text: str, length: str = "medium") -> str:
 def summarize_document(file, summary_length: str):
     """Main processing function for Gradio interface"""
     if file is None:
-        return "Please upload a document first"
+        return "Please upload a document first", "Ready"
     
     file_path = file.name
     file_extension = file_path.split(".")[-1].lower()
     
     text, error = extract_text(file_path, file_extension)
     if error:
-        return error
+        return error, "Error"
     
     if not text or len(text.split()) < 30:
-        return "Document is too short or contains too little text to summarize"
+        return "Document is too short or contains too little text to summarize", "Ready"
     
     try:
-        return generate_summary(text, summary_length)
+        summary = generate_summary(text, summary_length)
+        return summary, "Summary complete"
     except Exception as e:
-        return f"Summarization error: {str(e)}"
+        return f"Summarization error: {str(e)}", "Error"
 
 # Gradio Interface
 with gr.Blocks(title="Document Summarizer", theme=gr.themes.Soft()) as demo: