[FEATURE] Add stop generation button to cancel ongoing AI response (#364)

backend/routes/sessions.py

@@ -1,14 +1,20 @@
-"""Sessions routes — /api/sessions — full CRUD"""
+"""Sessions routes — /api/sessions — full CRUD + reorder"""
 
 import uuid
+from typing import List
 from fastapi import APIRouter, HTTPException
+from pydantic import BaseModel
 from models.schemas import SessionCreate, SessionUpdate, BulkSessionRenameRequest
 from services import db_service
-# from backend.models.schemas import BulkSessionRenameRequest  # Adjust if other items are imported from here
 
 router = APIRouter()
 
 
+# ─── Request models for reorder ─────────────────────────────
+class ReorderSessionsRequest(BaseModel):
+    session_ids: List[str]
+
+
 @router.get("/")
 async def list_sessions():
     return db_service.get_all_sessions()
@@ -20,6 +26,7 @@ async def create_session(body: SessionCreate):
     session = db_service.create_session(sid, title=body.title, model=body.model)
     return session
 
+
 @router.patch("/bulk-rename")
 async def bulk_rename_sessions(body: BulkSessionRenameRequest):
     try:
@@ -62,8 +69,21 @@ async def bulk_rename_sessions(body: BulkSessionRenameRequest):
             status_code=500, 
             detail=f"Bulk rename failed: {str(e)}"
         )
-
-
+
+
+@router.patch("/reorder")
+async def reorder_sessions(req: ReorderSessionsRequest):
+    """
+    Update the order of sessions for drag‑and‑drop.
+    Expects a list of session IDs in the desired order (top to bottom).
+    """
+    try:
+        db_service.update_sessions_order(req.session_ids)
+        return {"success": True, "message": "Session order updated"}
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=f"Reorder failed: {str(e)}")
+
+
 @router.get("/{session_id}")
 async def get_session(session_id: str):
     s = db_service.get_session(session_id)
@@ -83,7 +103,6 @@ async def delete_session(session_id: str):
     db_service.delete_session(session_id)
     try:
         from services import rag_service
-
         rag_service.delete_session_index(session_id)
     except Exception:
         pass
@@ -112,15 +131,7 @@ async def get_documents(session_id: str):
 async def rag_stats(session_id: str):
     try:
         from services import rag_service
-
         count = rag_service.get_indexed_count(session_id)
     except Exception:
         count = 0
     return {"session_id": session_id, "indexed_chunks": count}
-
-
-@router.delete("/")
-async def clear_all_sessions():
-    db_service.clear_all_sessions()
-    return {"message": "All sessions cleared"}
-

backend/services/db_service.py

@@ -56,6 +56,16 @@ def get_db():
         if conn:
             conn.close()
 
+
+def ensure_order_index_column():
+    """Add order_index column to sessions table if not exists (for drag‑and‑drop reordering)."""
+    with get_db() as conn:
+        try:
+            conn.execute("ALTER TABLE sessions ADD COLUMN order_index INTEGER DEFAULT 0")
+        except sqlite3.OperationalError:
+            pass  # column already exists
+
+
 def init_db():
     """Create all tables on startup."""
     with get_db() as conn:
@@ -122,12 +132,8 @@ def init_db():
                 prompt TEXT NOT NULL,
                 created_at TEXT DEFAULT (datetime('now'))
             );
-
         """)
-        try:
-            conn.execute("ALTER TABLE documents ADD COLUMN status TEXT DEFAULT 'completed'")
-        except sqlite3.OperationalError:
-            pass  # column already exists
+        ensure_order_index_column()   # <-- add order_index column
 
         cols = [row[1] for row in conn.execute("PRAGMA table_info(messages)").fetchall()]
         if "benchmarks" not in cols:
@@ -168,9 +174,10 @@ def clear_all_sessions():
 
 
 def get_all_sessions() -> list[dict]:
+    """Fetch sessions sorted by manual order (order_index) then by last updated."""
     with get_db() as conn:
         rows = conn.execute(
-            "SELECT * FROM sessions ORDER BY updated_at DESC"
+            "SELECT * FROM sessions ORDER BY order_index ASC, updated_at DESC"
         ).fetchall()
         return [dict(r) for r in rows]
 
@@ -349,4 +356,15 @@ def update_prompt_template(template_id: int, prompt_title: str = None, prompt: s
 def delete_prompt_template(template_id: int):
     """Delete a prompt template by ID."""
     with get_db() as conn:
-        conn.execute("DELETE FROM prompt_templates WHERE id = ?", (template_id,))
+        conn.execute("DELETE FROM prompt_templates WHERE id = ?", (template_id,))
+
+
+# ─── Drag‑and‑drop reordering───────────
+def update_sessions_order(session_ids: list):
+    """Update order_index for each session based on its position in the list (0,1,2...)."""
+    with get_db() as conn:
+        for idx, session_id in enumerate(session_ids):
+            conn.execute(
+                "UPDATE sessions SET order_index = ? WHERE id = ?",
+                (idx, session_id)
+            )

frontend/package.json

@@ -10,6 +10,9 @@
     "screenshot": "node scripts/capture-screenshots.js"
   },
   "dependencies": {
+    "@dnd-kit/core": "^6.3.1",
+    "@dnd-kit/sortable": "^10.0.0",
+    "@dnd-kit/utilities": "^3.2.2",
     "react": "^18.3.0",
     "react-dom": "^18.3.0",
     "uuid": "^10.0.0"