[FEATURE] Add copy button to code blocks in assistant messages (#363)

backend/routes/export.py

@@ -2,28 +2,35 @@
 
 import json
 from datetime import datetime
+from typing import List
 from fastapi import APIRouter, HTTPException
 from fastapi.responses import Response
+from pydantic import BaseModel
 from models.schemas import ExportFormat
 from services import db_service
 
 router = APIRouter()
 
 
+class ExportMessagesRequest(BaseModel):
+    message_ids: List[str]
+    format: ExportFormat
+
+
 @router.get("/{session_id}/{fmt}")
 async def export_session(session_id: str, fmt: ExportFormat):
-    session  = db_service.get_session(session_id)
+    session = db_service.get_session(session_id)
     if not session:
         raise HTTPException(404, "Session not found")
 
     messages = db_service.get_messages_full(session_id)
-    title    = session.get("title", "LocalMind Chat")
-    ts       = datetime.now().strftime("%Y-%m-%d %H:%M")
+    title = session.get("title", "LocalMind Chat")
+    ts = datetime.now().strftime("%Y-%m-%d %H:%M")
 
     if fmt == ExportFormat.json:
-        content   = json.dumps({"session": session, "messages": messages, "exported_at": ts}, indent=2, ensure_ascii=False)
-        media     = "application/json"
-        filename  = f"localmind_{session_id[:8]}.json"
+        content = json.dumps({"session": session, "messages": messages, "exported_at": ts}, indent=2, ensure_ascii=False)
+        media = "application/json"
+        filename = f"localmind_{session_id[:8]}.json"
 
     elif fmt == ExportFormat.markdown:
         lines = [f"# {title}\n", f"*Exported: {ts} | Model: {session.get('model','?')}*\n\n---\n"]
@@ -33,21 +40,63 @@ async def export_session(session_id: str, fmt: ExportFormat):
             if m.get("sources"):
                 lines.append(f"*Sources: {', '.join(m['sources'])}*\n")
             lines.append("\n---\n")
-        content   = "\n".join(lines)
-        media     = "text/markdown"
-        filename  = f"localmind_{session_id[:8]}.md"
+        content = "\n".join(lines)
+        media = "text/markdown"
+        filename = f"localmind_{session_id[:8]}.md"
 
     else:  # txt
         lines = [f"LocalMind Export — {title}", f"Exported: {ts}", "=" * 50, ""]
         for m in messages:
             role = "YOU" if m["role"] == "user" else "LOCALMIND"
             lines += [f"[{role}]", m["content"], ""]
-        content   = "\n".join(lines)
-        media     = "text/plain"
-        filename  = f"localmind_{session_id[:8]}.txt"
+        content = "\n".join(lines)
+        media = "text/plain"
+        filename = f"localmind_{session_id[:8]}.txt"
 
     return Response(
         content=content.encode("utf-8"),
         media_type=media,
         headers={"Content-Disposition": f'attachment; filename="{filename}"'},
     )
+
+
+@router.post("/messages")
+async def export_messages(req: ExportMessagesRequest):
+    messages = db_service.get_messages_by_ids(req.message_ids)
+    if not messages:
+        raise HTTPException(404, "No messages found for the given IDs")
+
+    messages.sort(key=lambda m: m.get("timestamp", ""))
+    ts = datetime.now().strftime("%Y-%m-%d %H:%M")
+
+    if req.format == ExportFormat.json:
+        content = json.dumps({"messages": messages, "exported_at": ts}, indent=2, ensure_ascii=False)
+        media = "application/json"
+        filename = f"localmind_messages_{ts.replace(' ', '_')}.json"
+
+    elif req.format == ExportFormat.markdown:
+        lines = ["# LocalMind – Exported Messages\n", f"*Exported: {ts}*\n\n---\n"]
+        for m in messages:
+            role_label = "**You**" if m["role"] == "user" else "**LocalMind**"
+            lines.append(f"{role_label}\n\n{m['content']}\n")
+            if m.get("sources"):
+                lines.append(f"*Sources: {', '.join(m['sources'])}*\n")
+            lines.append("\n---\n")
+        content = "\n".join(lines)
+        media = "text/markdown"
+        filename = f"localmind_messages_{ts.replace(' ', '_')}.md"
+
+    else:
+        lines = ["LocalMind Export — Selected Messages", f"Exported: {ts}", "=" * 50, ""]
+        for m in messages:
+            role = "YOU" if m["role"] == "user" else "LOCALMIND"
+            lines += [f"[{role}]", m["content"], ""]
+        content = "\n".join(lines)
+        media = "text/plain"
+        filename = f"localmind_messages_{ts.replace(' ', '_')}.txt"
+
+    return Response(
+        content=content.encode("utf-8"),
+        media_type=media,
+        headers={"Content-Disposition": f'attachment; filename="{filename}"'},
+    )

backend/services/db_service.py

@@ -261,6 +261,21 @@ def log_plugin(session_id: str, plugin: str, inp: str, out: str, success: bool =
             (session_id, plugin, inp, out, int(success)),
         )
 
+
+def get_messages_by_ids(message_ids: list):
+    """Fetch messages by list of message IDs (used for batch export)."""
+    if not message_ids:
+        return []
+    placeholders = ','.join('?' for _ in message_ids)
+    with get_db() as conn:
+        rows = conn.execute(f"""
+            SELECT id, role, content, sources, created_at as timestamp
+            FROM messages
+            WHERE id IN ({placeholders})
+        """, message_ids).fetchall()
+        return [dict(row) for row in rows]
+
+
 # ─── Prompt Template Registry ───────────────────────────────────────────────
 def create_prompt_template(prompt_title: str, prompt: str) -> dict:
     """Create a new prompt template."""
@@ -311,4 +326,4 @@ 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,))