Update app.py

app.py

@@ -12,17 +12,24 @@ print(f"Ready ✅ on {device}")
 
 def summarize_text(user_input, history, max_len, min_len, bullet_mode):
     user_input = user_input.strip()
+
     if not user_input:
-        history.append(("", "⚠️ Please enter some text."))
+        history.append({"role": "assistant", "content": "⚠️ Please enter some text."})
         return history, history
 
     if len(user_input.split()) < 30:
-        history.append((user_input, "👋 Please paste longer text (30+ words) to summarize."))
+        history.append({"role": "user",      "content": user_input})
+        history.append({"role": "assistant", "content": "👋 Please paste longer text (30+ words) to summarize."})
         return history, history
 
     try:
-        inputs = tokenizer(user_input, return_tensors="pt",
-                           max_length=1024, truncation=True).to(device)
+        inputs = tokenizer(
+            user_input,
+            return_tensors="pt",
+            max_length=1024,
+            truncation=True
+        ).to(device)
+
         ids = model.generate(
             inputs["input_ids"],
             max_new_tokens=int(max_len),
@@ -32,10 +39,11 @@ def summarize_text(user_input, history, max_len, min_len, bullet_mode):
             early_stopping=True,
             no_repeat_ngram_size=3
         )
+
         summary = tokenizer.decode(ids[0], skip_special_tokens=True)
 
         if bullet_mode:
-            sentences = summary.replace("?",".").replace("!",".").split(". ")
+            sentences = summary.replace("?", ".").replace("!", ".").split(". ")
             body = "\n".join(f"• {s.strip().capitalize()}" for s in sentences if s.strip())
             out  = f"📌 **Summary (Bullets)**:\n\n{body}"
         else:
@@ -43,36 +51,57 @@ def summarize_text(user_input, history, max_len, min_len, bullet_mode):
 
         orig = len(user_input.split())
         summ = len(summary.split())
-        out += f"\n\n---\n📊 *{orig} words → {summ} words | Reduced by {round((1-summ/orig)*100,1)}%*"
-        history.append((user_input[:80]+"..." if len(user_input)>80 else user_input, out))
+        out += f"\n\n---\n📊 *{orig} words → {summ} words | Reduced by {round((1 - summ/orig)*100, 1)}%*"
+
+        label = user_input[:80] + "..." if len(user_input) > 80 else user_input
+        history.append({"role": "user",      "content": label})
+        history.append({"role": "assistant", "content": out})
 
     except Exception as e:
-        history.append((user_input[:60]+"...", f"❌ Error: {e}"))
+        history.append({"role": "user",      "content": user_input[:60] + "..."})
+        history.append({"role": "assistant", "content": f"❌ Error: {e}"})
 
     return history, history
 
+
 def clear_chat():
     return [], []
 
+
 with gr.Blocks(title="Text Summarizer", theme=gr.themes.Soft()) as demo:
+
     gr.Markdown("# 📝 Text Summarization Chatbox\n### Powered by `facebook/bart-large-cnn`")
+
     with gr.Row():
         with gr.Column(scale=7):
-            chatbot = gr.Chatbot(height=450, bubble_full_width=False)
+            chatbot = gr.Chatbot(height=450, type="messages")  # ✅ Gradio 6.x format
             state   = gr.State([])
+
             with gr.Row():
-                txt   = gr.Textbox(placeholder="Paste text here...", lines=3, show_label=False, scale=8)
+                txt   = gr.Textbox(
+                    placeholder="Paste text here...",
+                    lines=3,
+                    show_label=False,
+                    scale=8
+                )
                 with gr.Column(scale=2):
                     btn   = gr.Button("✨ Summarize", variant="primary")
-                    clear = gr.Button("🗑️ Clear", variant="secondary")
+                    clear = gr.Button("🗑️ Clear",    variant="secondary")
+
         with gr.Column(scale=3):
             gr.Markdown("### ⚙️ Settings")
-            max_l  = gr.Slider(50, 500, value=150, step=10, label="Max Length")
-            min_l  = gr.Slider(10, 200, value=40,  step=5,  label="Min Length")
+            max_l  = gr.Slider(50,  500, value=150, step=10, label="Max Length")
+            min_l  = gr.Slider(10,  200, value=40,  step=5,  label="Min Length")
             bullet = gr.Checkbox(label="🔵 Bullet Point Mode", value=False)
+            gr.Markdown("""
+            ---
+            **💡 Tips:**
+            - Works best with 100–1000 word inputs
+            - Toggle Bullet Mode for point-wise output
+            """)
 
-    btn.click(summarize_text,   inputs=[txt, state, max_l, min_l, bullet], outputs=[chatbot, state])
-    txt.submit(summarize_text,  inputs=[txt, state, max_l, min_l, bullet], outputs=[chatbot, state])
+    btn.click(summarize_text,  inputs=[txt, state, max_l, min_l, bullet], outputs=[chatbot, state])
+    txt.submit(summarize_text, inputs=[txt, state, max_l, min_l, bullet], outputs=[chatbot, state])
     clear.click(clear_chat, outputs=[chatbot, state])
 
 demo.launch()