Update utils/summarizer.py

utils/summarizer.py

@@ -6,7 +6,13 @@ from typing import List
 # ========== Load Summarization Pipeline ==========
 summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
 
-# ========== Helper Functions ==========
+# ========== Text Helpers ==========
+
+def clean_text(text: str) -> str:
+    """
+    🧹 Remove excessive whitespace and line breaks.
+    """
+    return text.replace("\n", " ").replace("  ", " ").strip()
 
 def split_text(text: str, max_chunk_len: int = 800) -> List[str]:
     """
@@ -28,33 +34,55 @@ def split_text(text: str, max_chunk_len: int = 800) -> List[str]:
 
     return chunks
 
-def clean_text(text: str) -> str:
-    """
-    🧹 Remove excessive whitespace and line breaks.
-    """
-    return text.replace("\n", " ").replace("  ", " ").strip()
-
-# ========== Summarization Function ==========
+# ========== Summarization Functions ==========
 
-def summarize_text(text: str) -> str:
+def summarize_text(text: str, as_paragraph: bool = False, fallback: bool = True) -> str:
     """
-    📄 Generate a readable executive summary using bullet points.
+    📄 Generate an executive summary.
+    
+    Params:
+    - as_paragraph: True → returns as 2–3 paragraph summary; False → bullet points
+    - fallback: True → if model fails, returns manual fallback
     """
     if not text.strip():
         return "No input provided."
 
-    cleaned = clean_text(text)
-    chunks = split_text(cleaned)
+    try:
+        cleaned = clean_text(text)
+        chunks = split_text(cleaned)
+        summaries = []
+
+        for chunk in chunks:
+            result = summarizer(chunk, max_length=130, min_length=30, do_sample=False)
+            summary = result[0]["summary_text"].strip()
+            summaries.append(summary)
+
+        if as_paragraph:
+            return "📄 Executive Summary:\n\n" + "\n\n".join(summaries)
 
-    bullet_points = []
+        # Otherwise → return as bullet points
+        bullet_points = []
+        for summary in summaries:
+            lines = summary.split('. ')
+            for line in lines:
+                cleaned_line = line.strip().rstrip('.')
+                if cleaned_line:
+                    bullet_points.append(f"• {cleaned_line}.")
 
-    for chunk in chunks:
-        result = summarizer(chunk, max_length=130, min_length=30, do_sample=False)
-        summary = result[0]["summary_text"].strip()
-        lines = summary.split('. ')
-        for line in lines:
-            cleaned_line = line.strip().rstrip('.')
-            if cleaned_line:
-                bullet_points.append(f"• {cleaned_line}.")
+        return "📄 Executive Summary:\n" + "\n".join(bullet_points)
 
-    return "📄 Executive Summary:\n" + "\n".join(bullet_points)
+    except Exception as e:
+        if fallback:
+            return fallback_summary(text)
+        return f"An error occurred: {str(e)}"
+
+# ========== Fallback Summary (manual) ==========
+
+def fallback_summary(text: str, max_lines: int = 5) -> str:
+    """
+    🧭 Fallback: Return first few sentences as pseudo-summary.
+    """
+    lines = text.split(". ")
+    selected = lines[:max_lines]
+    points = [f"• {line.strip().rstrip('.')}" for line in selected if line.strip()]
+    return "📄 (Fallback Summary)\n" + "\n".join(points)