Update main.py

main.py

@@ -1,7 +1,7 @@
 import os
 os.environ['HF_HOME'] = '/tmp'
-# main.py (excerpt)
-from flask import Flask, request, jsonify
+
+from flask import Flask, request, jsonify, render_template
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, pipeline
 import math, textwrap
 
@@ -14,23 +14,21 @@ summarizer = pipeline("summarization", model=model, tokenizer=tokenizer, device=
 
 # Simple mapping of presets to generation lengths
 LENGTH_PRESETS = {
-  "short": {"min_length": 20, "max_length": 60},
-  "medium": {"min_length": 60, "max_length": 130},
-  "long": {"min_length": 130, "max_length": 300},
+    "short": {"min_length": 20, "max_length": 60},
+    "medium": {"min_length": 60, "max_length": 130},
+    "long": {"min_length": 130, "max_length": 300},
 }
 
 def chunk_text_by_chars(text, max_chars=1500, overlap=200):
-    if len(text) <= max_chars: 
+    if len(text) <= max_chars:
         return [text]
     parts = []
     start = 0
     while start < len(text):
         end = min(len(text), start + max_chars)
-        # try to break at newline or sentence boundary for nicer chunking
         chunk = text[start:end]
-        # extend to nearest newline if possible (avoid cutting sentences)
         nl = chunk.rfind('\n')
-        if nl > max_chars*0.6:
+        if nl > max_chars * 0.6:
             end = start + nl
             chunk = text[start:end]
         parts.append(chunk.strip())
@@ -49,10 +47,15 @@ def apply_tone_instruction(text, tone):
         instr = "Summarize:"
     return f"{instr}\n\n{text}"
 
+# NEW: Route to show summarizer.html (fixes 404)
+@app.route("/")
+def home():
+    return render_template("index.html")
+
 @app.route("/summarize", methods=["POST"])
 def summarize_route():
     data = request.get_json(force=True)
-    text = data.get("text", "")[:20000]  # safe cap
+    text = data.get("text", "")[:20000]
     length = data.get("length", "medium")
     tone = data.get("tone", "neutral")
 
@@ -65,25 +68,26 @@ def summarize_route():
 
     for chunk in chunks:
         prompted = apply_tone_instruction(chunk, tone)
-        # call summarizer with min/max lengths
-        out = summarizer(prompted,
-                         min_length=preset["min_length"],
-                         max_length=preset["max_length"],
-                         truncation=True)[0]["summary_text"]
+        out = summarizer(
+            prompted,
+            min_length=preset["min_length"],
+            max_length=preset["max_length"],
+            truncation=True
+        )[0]["summary_text"]
         summaries.append(out.strip())
 
-    # If multiple chunk summaries, join and compress once more
     if len(summaries) == 1:
         final = summaries[0]
     else:
         combined = "\n\n".join(summaries)
         prompted = apply_tone_instruction(combined, tone)
-        final = summarizer(prompted,
-                           min_length=preset["min_length"],
-                           max_length=preset["max_length"],
-                           truncation=True)[0]["summary_text"]
+        final = summarizer(
+            prompted,
+            min_length=preset["min_length"],
+            max_length=preset["max_length"],
+            truncation=True
+        )[0]["summary_text"]
 
-    # if bullet tone, post-process
     if tone == "bullet":
         lines = [l.strip() for s in final.splitlines() for l in s.split(". ") if l.strip()]
         final = "\n".join(f"- {l.rstrip('.')}" for l in lines[:20])
@@ -91,5 +95,4 @@ def summarize_route():
     return jsonify({"summary": final})
 
 if __name__ == "__main__":
-    app.run(debug=True,port=7860)
-
+    app.run(debug=True, port=7860)