Update app.py

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 import re
 from transformers import pipeline, AutoTokenizer
 from PyPDF2 import PdfReader
+import tempfile
 
 # =========================
 # Model setup (CPU-safe)
@@ -15,20 +16,20 @@ summarizer = pipeline(
     device=-1  # CPU only
 )
 
-# Better AI advice generator - use flan-t5-base for improved quality (still CPU-friendly)
+# Better AI advice generator - flan-t5-base is still quite CPU friendly
 advice_generator = pipeline(
     "text2text-generation",
     model="google/flan-t5-base",
     device=-1  # CPU only
 )
 
-CHUNK_SIZE = 900  # safe margin for summarizer
+CHUNK_SIZE = 900  # safe margin under typical max input
 
 # =========================
 # Utilities
 # =========================
 def clean_text(text: str) -> str:
-    """Fix quotes, spacing, repetition, and broken punctuation."""
+    """Fix quotes, spacing, repetition, broken punctuation."""
     text = text.replace("‘", "'").replace("’", "'")
     text = text.replace("“", '"').replace("”", '"')
     text = re.sub(r"[.]{2,}", ".", text)
@@ -55,11 +56,9 @@ def chunk_text(text: str):
     return chunks
 
 def generate_ai_advice(summary: str) -> str:
-    """Use AI to generate personalized study advice based on the summary."""
-    # Truncate summary if too long
-    truncated_summary = summary[:1000]  # Slightly longer for base model
+    """Generate personalized study advice based on the paper summary."""
+    truncated_summary = summary[:1000]
     
-    # Improved prompt for better, more relevant tips
     prompt = (
         f"Read this summary of a technical paper: '{truncated_summary}'\n\n"
         "Generate exactly 5 practical study tips for a student to better understand and retain this content. "
@@ -72,26 +71,25 @@ def generate_ai_advice(summary: str) -> str:
         prompt,
         max_length=250,
         num_return_sequences=1,
-        do_sample=False,  # Greedy for consistency
-        temperature=0.7   # Slight creativity if needed
+        do_sample=False,
+        temperature=0.7
     )[0]["generated_text"]
     
-    # Post-process to ensure bullet format
+    # Try to clean into bullet points
     tips = [tip.strip() for tip in generated.split('\n') if tip.strip().startswith('-') or tip.strip()]
-    if not tips:
-        tips = generated.split('. ')  # Fallback split
+    if not tips or len(tips) < 3:
+        tips = [t.strip() for t in generated.split('.') if t.strip()]
     
     advice_md = "\n\n---\n\n### 📚 AI-Generated Study Tips\n\n"
     for i, tip in enumerate(tips[:5], 1):
         clean_tip = tip.lstrip('- ').strip()
         advice_md += f"- {clean_tip}\n"
     
-    advice_md += "\n**Pro tip**: Combine these with spaced repetition for long-term mastery!"
-    
+    advice_md += "\n**Pro tip**: Combine these with spaced repetition (Anki / Quizlet) for long-term retention!"
     return advice_md
 
 def summarize_long_text(text: str) -> str:
-    """Summarize arbitrarily long text safely + add AI study advice."""
+    """Summarize long text in chunks + add AI study advice."""
     if not text or len(text.strip()) == 0:
         return "No text provided."
     
@@ -99,20 +97,21 @@ def summarize_long_text(text: str) -> str:
     summaries = []
     
     for chunk in chunks:
-        summary = summarizer(
-            chunk,
-            max_length=150,
-            min_length=40,
-            do_sample=False
-        )[0]["summary_text"]
-        summaries.append(summary)
+        try:
+            summary = summarizer(
+                chunk,
+                max_length=150,
+                min_length=40,
+                do_sample=False
+            )[0]["summary_text"]
+            summaries.append(summary)
+        except Exception:
+            pass  # skip problematic chunks
     
     merged = " ".join(summaries)
     cleaned_summary = clean_text(merged)
     
-    # Generate AI advice based on the summary
     ai_advice = generate_ai_advice(cleaned_summary)
-    
     return cleaned_summary + ai_advice
 
 def read_pdf(file) -> str:
@@ -122,15 +121,34 @@ def read_pdf(file) -> str:
         pages = [page.extract_text() or "" for page in reader.pages]
         return " ".join(pages)
     except Exception as e:
-        return f"PDF read error: {e}"
+        return f"PDF read error: {str(e)}"
+
+# =========================
+# Download helper
+# =========================
+def create_download_file(content: str) -> str:
+    """Create temporary file for Gradio file download component"""
+    with tempfile.NamedTemporaryFile(delete=False, suffix=".txt", mode="w", encoding="utf-8") as tmp:
+        tmp.write(content)
+        return tmp.name
 
 # =========================
 # Main handler
 # =========================
-def process_input(text, file):
+def process_input(text: str, file):
+    input_text = ""
+    
     if file is not None:
-        text = read_pdf(file)
-    return summarize_long_text(text)
+        input_text = read_pdf(file)
+    elif text.strip():
+        input_text = text
+    else:
+        return "Please paste some text or upload a PDF.", None
+    
+    result = summarize_long_text(input_text)
+    download_path = create_download_file(result)
+    
+    return result, download_path
 
 # =========================
 # Gradio UI
@@ -138,35 +156,41 @@ def process_input(text, file):
 with gr.Blocks() as demo:
     gr.Markdown("# 📄 Long Text Summarizer + AI Study Assistant")
     gr.Markdown(
-        "• Handles **thousands of words**\n"
-        "• Supports **PDF upload**\n"
-        "• Optimized for **CPU / free tier**\n"
-        "• Includes **AI-generated study tips** based on the summary content"
+        "• Handles very long documents (thousands of words)\n"
+        "• Supports **PDF** upload or direct paste\n"
+        "• Runs on CPU – works on free hardware\n"
+        "• Gives you **5 AI-generated study tips** tailored to the content\n"
+        "• Download result as .txt file"
     )
     
-    text_input = gr.Textbox(
-        lines=15,
-        label="Paste text (optional)",
-        placeholder="Paste lecture notes, textbook chapter, article..."
-    )
+    with gr.Row():
+        text_input = gr.Textbox(
+            lines=10,
+            label="Paste your text here (optional)",
+            placeholder="Paste lecture notes, article, book chapter...",
+        )
+        file_input = gr.File(
+            label="Or upload a PDF",
+            file_types=[".pdf"]
+        )
     
-    file_input = gr.File(
-        label="Upload PDF (optional)",
-        file_types=[".pdf"]
-    )
+    summarize_btn = gr.Button("Summarize & Get Study Tips", variant="primary")
     
     output = gr.Textbox(
         lines=16,
-        label="Summary + AI Study Advice",
-        placeholder="Summary appears first, followed by AI-generated learning tips..."
+        label="Summary + AI-generated study advice",
+        interactive=False
     )
     
-    summarize_btn = gr.Button("Summarize & Get AI Study Tips", variant="primary")
+    download_output = gr.File(
+        label="Download full result (.txt)",
+        interactive=False
+    )
     
     summarize_btn.click(
         fn=process_input,
         inputs=[text_input, file_input],
-        outputs=output
+        outputs=[output, download_output]
     )
 
 demo.launch()