Update app.py

app.py

@@ -54,21 +54,23 @@ def summarize_file(file):
     chunks = [text[i:i+1000] for i in range(0, len(text), 1000)]
     print(f"Chunks created: {len(chunks)}")
     if not chunks: return "❌ No chunks to summarize"
+    # Select 12 chunks evenly spaced
+    selected_indices = [int(i * len(chunks) / 12) for i in range(12)] if len(chunks) >= 12 else list(range(len(chunks)))
     summaries = []
-    for i in range(0, len(chunks), 4):
-        if time.time() - start > 15:
-            summaries.append("⚠️ Stopped early")
-            break
-        batch = chunks[i:i+4]
-        if any(sum(1 for c in chunk if not c.isalnum()) / len(chunk) > 0.7 for chunk in batch):
-            summaries.append(f"**Chunk {i+1}–{i+len(batch)}**: Skipped (equation-heavy)")
+    for i in selected_indices:
+        chunk = chunks[i]
+        if sum(1 for c in chunk if not c.isalnum()) / len(chunk) > 0.7:
+            summaries.append(f"**Chunk {i+1}**: Skipped (equation-heavy)")
             continue
         try:
-            results = summarizer(batch, max_length=50, min_length=10, do_sample=False)
-            summaries.extend(f"**Chunk {i+j+1}**:\n{r['summary_text']}" for j, r in enumerate(results))
+            summary = summarizer(chunk, max_length=40, min_length=10, do_sample=False)[0]['summary_text']
+            summaries.append(f"**Chunk {i+1}**:\n{summary}")
         except Exception as e:
-            summaries.append(f"**Chunk {i+1}–{i+len(batch)}**: ❌ Error: {str(e)}")
-    return f"**Chars**: {len(text)}\n**Time**: {time.time()-start:.2f}s\n\n" + "\n\n".join(summaries)
+            summaries.append(f"**Chunk {i+1}**: ❌ Error: {str(e)}")
+    # Pad if <12 summaries
+    while len(summaries) < 12:
+        summaries.append(f"**Chunk {len(summaries)+1}**: Insufficient content for full summary")
+    return f"**Chars**: {len(text)}\n**Time**: {time.time()-start:.2f}s\n\n" + "\n\n".join(summaries[:12])
 
 demo = gr.Interface(
     fn=summarize_file, inputs=gr.File(label="📄 Any File", type="binary"),