Update app.py

app.py

@@ -11,14 +11,14 @@ logging.basicConfig(level=logging.ERROR)
 device = -1  # CPU
 print("⚠️ CPU-only mode. Expect ~20–30s for large documents.")
 
-# === Load the Summarization Model ===
+# === Load Summarization Model ===
 try:
     summarizer = pipeline("summarization", model="t5-small", device=device, torch_dtype=torch.float32)
 except Exception as e:
     print(f"❌ Model loading failed: {e}")
     exit(1)
 
-# === Sentence-Smart Chunking ===
+# === Sentence-based Chunking ===
 def smart_chunk(text, max_chunk_len=2000):
     sentences = re.split(r'(?<=[.!?]) +', text)
     chunks, current_chunk = [], ""
@@ -32,7 +32,7 @@ def smart_chunk(text, max_chunk_len=2000):
         chunks.append(current_chunk.strip())
     return chunks
 
-# === Summarization for a Single File ===
+# === Summarization for One File ===
 def summarize_file_bytes(file_bytes, filename):
     start_time = time.time()
     try:
@@ -47,7 +47,7 @@ def summarize_file_bytes(file_bytes, filename):
     if not text:
         return f"{filename}: ❌ No text found.", ""
 
-    text = text[:300000]  # Trim to model-safe size
+    text = text[:300000]
     chunks = smart_chunk(text)
     summaries, line_count = [], 0
 
@@ -68,14 +68,13 @@ def summarize_file_bytes(file_bytes, filename):
     summary_text = f"📄 **{filename}**\n**Characters**: {len(text)} | **Time**: {total_time:.2f}s\n\n" + "\n\n".join(summaries)
     return summary_text, summary_text
 
-# === Function for Multiple Files ===
+# === Multiple Files Handler ===
 def summarize_multiple_files(file_objs):
     all_summaries = []
     combined_text = ""
 
-    for file_obj in file_objs:
-        file_bytes = file_obj.read()
-        filename = file_obj.name.split("/")[-1]
+    for file_bytes, file_info in file_objs:
+        filename = file_info['name'].split("/")[-1]
         summary, raw_text = summarize_file_bytes(file_bytes, filename)
         all_summaries.append(summary)
         combined_text += f"\n\n{raw_text}\n" + "="*60 + "\n"
@@ -99,7 +98,7 @@ demo = gr.Interface(
     description="Summarizes multiple PDFs or TXTs into at least 15 lines each. Download final output as .txt. CPU-optimized."
 )
 
-# === Run the App ===
+# === Launch App ===
 if __name__ == "__main__":
     try:
         demo.launch(share=False, server_port=7860)