Update app.py

app.py

@@ -38,10 +38,33 @@ class ChunkInput(BaseModel):
 def summarize_text(data: SummarizeInput):
     print("📥 Received summarize request.")
     start = time.time()
-    summary = summarizer(data.text, max_length=150, min_length=30, do_sample=False)
+
+    # Chunk the input text
+    chunks = chunk_by_token_limit(data.text, 1024, tokenizer)
+    total_chunks = len(chunks)
+    print(f"🧩 Starting summarization over {total_chunks} chunks...")
+
+    all_summaries = []
+    for idx, chunk in enumerate(chunks):
+        try:
+            summary = summarizer(chunk, max_length=150, min_length=30, do_sample=False)
+            summary_text = summary[0]["summary_text"]
+            all_summaries.append(summary_text)
+            print(f"✅ Summary chunk generated {idx + 1}/{total_chunks}")
+        except Exception as e:
+            print(f"❌ Error summarizing chunk {idx + 1}/{total_chunks}: {str(e)}")
+            all_summaries.append("[Error summarizing this chunk]")
+
     duration = time.time() - start
-    print(f"📝 Summary generated in {duration:.2f}s.")
-    return {"summary": summary[0]["summary_text"], "time_taken": f"{duration:.2f}s"}
+    print(f"🏁 All {total_chunks} summaries generated in {duration:.2f}s.")
+
+    return {
+        "summaries": all_summaries,
+        "summary_combined": " ".join(all_summaries),
+        "chunk_count": total_chunks,
+        "time_taken": f"{duration:.2f}s"
+    }
+
 
 @app.post("/chunk")
 def chunk_text(data: ChunkInput):