mirror of
https://github.com/JamesTheGiblet/BuddAI.git
synced 2026-01-08 21:58:40 +00:00
feat: add session title management and rename functionality in the BuddAI interface
This commit is contained in:
parent
6889087dd8
commit
a70b9ea2ce
3 changed files with 112 additions and 8 deletions
|
|
@ -275,10 +275,16 @@ class BuddAI:
|
||||||
CREATE TABLE IF NOT EXISTS sessions (
|
CREATE TABLE IF NOT EXISTS sessions (
|
||||||
session_id TEXT PRIMARY KEY,
|
session_id TEXT PRIMARY KEY,
|
||||||
started_at TIMESTAMP,
|
started_at TIMESTAMP,
|
||||||
ended_at TIMESTAMP
|
ended_at TIMESTAMP,
|
||||||
|
title TEXT
|
||||||
)
|
)
|
||||||
""")
|
""")
|
||||||
|
|
||||||
|
try:
|
||||||
|
cursor.execute("ALTER TABLE sessions ADD COLUMN title TEXT")
|
||||||
|
except sqlite3.OperationalError:
|
||||||
|
pass
|
||||||
|
|
||||||
cursor.execute("""
|
cursor.execute("""
|
||||||
CREATE TABLE IF NOT EXISTS messages (
|
CREATE TABLE IF NOT EXISTS messages (
|
||||||
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||||||
|
|
@ -799,10 +805,27 @@ float applyForge(float current, float target, float k) {{ return target + (curre
|
||||||
"""Retrieve recent sessions from DB"""
|
"""Retrieve recent sessions from DB"""
|
||||||
conn = sqlite3.connect(DB_PATH)
|
conn = sqlite3.connect(DB_PATH)
|
||||||
cursor = conn.cursor()
|
cursor = conn.cursor()
|
||||||
cursor.execute("SELECT session_id, started_at FROM sessions ORDER BY started_at DESC LIMIT ?", (limit,))
|
cursor.execute("SELECT session_id, started_at, title FROM sessions ORDER BY started_at DESC LIMIT ?", (limit,))
|
||||||
rows = cursor.fetchall()
|
rows = cursor.fetchall()
|
||||||
conn.close()
|
conn.close()
|
||||||
return [{"id": r[0], "date": r[1]} for r in rows]
|
return [{"id": r[0], "date": r[1], "title": r[2]} for r in rows]
|
||||||
|
|
||||||
|
def rename_session(self, session_id, new_title):
|
||||||
|
"""Rename a session"""
|
||||||
|
conn = sqlite3.connect(DB_PATH)
|
||||||
|
cursor = conn.cursor()
|
||||||
|
cursor.execute("UPDATE sessions SET title = ? WHERE session_id = ?", (new_title, session_id))
|
||||||
|
conn.commit()
|
||||||
|
conn.close()
|
||||||
|
|
||||||
|
def delete_session(self, session_id):
|
||||||
|
"""Delete a session and its messages"""
|
||||||
|
conn = sqlite3.connect(DB_PATH)
|
||||||
|
cursor = conn.cursor()
|
||||||
|
cursor.execute("DELETE FROM sessions WHERE session_id = ?", (session_id,))
|
||||||
|
cursor.execute("DELETE FROM messages WHERE session_id = ?", (session_id,))
|
||||||
|
conn.commit()
|
||||||
|
conn.close()
|
||||||
|
|
||||||
def load_session(self, session_id):
|
def load_session(self, session_id):
|
||||||
"""Load a specific session context"""
|
"""Load a specific session context"""
|
||||||
|
|
@ -900,6 +923,13 @@ if SERVER_AVAILABLE:
|
||||||
class SessionLoadRequest(BaseModel):
|
class SessionLoadRequest(BaseModel):
|
||||||
session_id: str
|
session_id: str
|
||||||
|
|
||||||
|
class SessionRenameRequest(BaseModel):
|
||||||
|
session_id: str
|
||||||
|
title: str
|
||||||
|
|
||||||
|
class SessionDeleteRequest(BaseModel):
|
||||||
|
session_id: str
|
||||||
|
|
||||||
# Initialize server instance
|
# Initialize server instance
|
||||||
server_buddai = BuddAI(server_mode=True)
|
server_buddai = BuddAI(server_mode=True)
|
||||||
|
|
||||||
|
|
@ -953,6 +983,16 @@ if SERVER_AVAILABLE:
|
||||||
history = server_buddai.load_session(req.session_id)
|
history = server_buddai.load_session(req.session_id)
|
||||||
return {"history": history, "session_id": req.session_id}
|
return {"history": history, "session_id": req.session_id}
|
||||||
|
|
||||||
|
@app.post("/api/session/rename")
|
||||||
|
async def rename_session_endpoint(req: SessionRenameRequest):
|
||||||
|
server_buddai.rename_session(req.session_id, req.title)
|
||||||
|
return {"status": "success"}
|
||||||
|
|
||||||
|
@app.post("/api/session/delete")
|
||||||
|
async def delete_session_endpoint(req: SessionDeleteRequest):
|
||||||
|
server_buddai.delete_session(req.session_id)
|
||||||
|
return {"status": "success"}
|
||||||
|
|
||||||
@app.post("/api/session/new")
|
@app.post("/api/session/new")
|
||||||
async def new_session_endpoint():
|
async def new_session_endpoint():
|
||||||
new_id = server_buddai.start_new_session()
|
new_id = server_buddai.start_new_session()
|
||||||
|
|
|
||||||
Binary file not shown.
|
|
@ -42,9 +42,16 @@
|
||||||
--code-border: #d1d5da;
|
--code-border: #d1d5da;
|
||||||
--code-text: #24292e;
|
--code-text: #24292e;
|
||||||
}
|
}
|
||||||
.sidebar-left { width: 260px; background: var(--header-bg); border-right: 1px solid var(--border-color); display: flex; flex-direction: column; flex-shrink: 0; }
|
.sidebar-left { width: 260px; background: var(--header-bg); border-right: 1px solid var(--border-color); display: flex; flex-direction: column; flex-shrink: 0; transition: width 0.3s, opacity 0.3s; overflow: hidden; }
|
||||||
|
.sidebar-left.collapsed { width: 0; opacity: 0; border: none; }
|
||||||
.session-list { flex: 1; overflow-y: auto; }
|
.session-list { flex: 1; overflow-y: auto; }
|
||||||
.session-item { padding: 12px 15px; cursor: pointer; border-bottom: 1px solid var(--border-color); font-size: 0.85em; color: var(--text-color); transition: background 0.2s; }
|
.session-item { padding: 12px 15px; cursor: pointer; border-bottom: 1px solid var(--border-color); font-size: 0.85em; color: var(--text-color); transition: background 0.2s; display: flex; justify-content: space-between; align-items: center; }
|
||||||
|
.session-info { flex: 1; overflow: hidden; }
|
||||||
|
.rename-input { width: 100%; background: var(--input-bg); color: var(--text-color); border: 1px solid var(--btn-bg); padding: 4px; border-radius: 3px; font-family: inherit; }
|
||||||
|
.edit-icon, .delete-icon { opacity: 0; cursor: pointer; padding: 4px; font-size: 1.1em; margin-left: 2px; }
|
||||||
|
.session-item:hover .edit-icon, .session-item:hover .delete-icon { opacity: 0.5; }
|
||||||
|
.edit-icon:hover, .delete-icon:hover { opacity: 1; }
|
||||||
|
.delete-icon:hover { color: #ff4444; }
|
||||||
.session-item:hover { background: var(--bg-color); }
|
.session-item:hover { background: var(--bg-color); }
|
||||||
.session-item.active { background: var(--btn-bg); color: white; border-color: var(--btn-bg); }
|
.session-item.active { background: var(--btn-bg); color: white; border-color: var(--btn-bg); }
|
||||||
.new-chat-btn { margin: 15px; padding: 10px; background: var(--btn-bg); color: white; border: none; border-radius: 4px; cursor: pointer; font-weight: bold; }
|
.new-chat-btn { margin: 15px; padding: 10px; background: var(--btn-bg); color: white; border: none; border-radius: 4px; cursor: pointer; font-weight: bold; }
|
||||||
|
|
@ -187,6 +194,9 @@
|
||||||
const [history, setHistory] = useState([]);
|
const [history, setHistory] = useState([]);
|
||||||
const [sessions, setSessions] = useState([]);
|
const [sessions, setSessions] = useState([]);
|
||||||
const [currentSessionId, setCurrentSessionId] = useState(null);
|
const [currentSessionId, setCurrentSessionId] = useState(null);
|
||||||
|
const [showSidebar, setShowSidebar] = useState(true);
|
||||||
|
const [editingSession, setEditingSession] = useState(null);
|
||||||
|
const [renameText, setRenameText] = useState("");
|
||||||
const [input, setInput] = useState("");
|
const [input, setInput] = useState("");
|
||||||
const [loading, setLoading] = useState(false);
|
const [loading, setLoading] = useState(false);
|
||||||
const [status, setStatus] = useState("connecting");
|
const [status, setStatus] = useState("connecting");
|
||||||
|
|
@ -248,6 +258,35 @@
|
||||||
return () => clearInterval(timer);
|
return () => clearInterval(timer);
|
||||||
}, []);
|
}, []);
|
||||||
|
|
||||||
|
const handleRename = async (e) => {
|
||||||
|
if (e.key === 'Enter') {
|
||||||
|
await fetch("/api/session/rename", {
|
||||||
|
method: "POST",
|
||||||
|
headers: { "Content-Type": "application/json" },
|
||||||
|
body: JSON.stringify({ session_id: editingSession, title: renameText })
|
||||||
|
});
|
||||||
|
setEditingSession(null);
|
||||||
|
fetchSessions();
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
const handleDeleteSession = async (e, sessionId) => {
|
||||||
|
e.stopPropagation();
|
||||||
|
if (!window.confirm("Delete this chat?")) return;
|
||||||
|
|
||||||
|
await fetch("/api/session/delete", {
|
||||||
|
method: "POST",
|
||||||
|
headers: { "Content-Type": "application/json" },
|
||||||
|
body: JSON.stringify({ session_id: sessionId })
|
||||||
|
});
|
||||||
|
|
||||||
|
if (sessionId === currentSessionId) {
|
||||||
|
setHistory([]);
|
||||||
|
setCurrentSessionId(null);
|
||||||
|
}
|
||||||
|
fetchSessions();
|
||||||
|
};
|
||||||
|
|
||||||
const loadSession = async (sessionId) => {
|
const loadSession = async (sessionId) => {
|
||||||
setLoading(true);
|
setLoading(true);
|
||||||
try {
|
try {
|
||||||
|
|
@ -358,6 +397,7 @@
|
||||||
onChange={handleFileUpload}
|
onChange={handleFileUpload}
|
||||||
/>
|
/>
|
||||||
<button className="clear-btn" onClick={() => document.getElementById('upload-input').click()}>📂 Upload</button>
|
<button className="clear-btn" onClick={() => document.getElementById('upload-input').click()}>📂 Upload</button>
|
||||||
|
<button className="clear-btn" onClick={() => setShowSidebar(!showSidebar)}>{showSidebar ? 'Hide History' : 'Show History'}</button>
|
||||||
<button className="clear-btn" onClick={() => setIsSidebarOpen(!isSidebarOpen)}>{isSidebarOpen ? 'Hide Code' : 'Show Code'}</button>
|
<button className="clear-btn" onClick={() => setIsSidebarOpen(!isSidebarOpen)}>{isSidebarOpen ? 'Hide Code' : 'Show Code'}</button>
|
||||||
<button className="clear-btn" onClick={() => setTheme(t => t === 'dark' ? 'light' : 'dark')}>{theme === 'dark' ? '☀️' : '🌙'}</button>
|
<button className="clear-btn" onClick={() => setTheme(t => t === 'dark' ? 'light' : 'dark')}>{theme === 'dark' ? '☀️' : '🌙'}</button>
|
||||||
<select
|
<select
|
||||||
|
|
@ -372,13 +412,37 @@
|
||||||
</div>
|
</div>
|
||||||
</div>
|
</div>
|
||||||
<div className="main-layout">
|
<div className="main-layout">
|
||||||
<div className="sidebar-left">
|
<div className={`sidebar-left ${!showSidebar ? 'collapsed' : ''}`}>
|
||||||
<button className="new-chat-btn" onClick={startNewSession}>+ New Chat</button>
|
<button className="new-chat-btn" onClick={startNewSession}>+ New Chat</button>
|
||||||
<div className="session-list">
|
<div className="session-list">
|
||||||
{sessions.map(s => (
|
{sessions.map(s => (
|
||||||
<div key={s.id} className={`session-item ${s.id === currentSessionId ? 'active' : ''}`} onClick={() => loadSession(s.id)}>
|
<div key={s.id} className={`session-item ${s.id === currentSessionId ? 'active' : ''}`} onClick={() => loadSession(s.id)}>
|
||||||
<span className="session-date">{new Date(s.date).toLocaleDateString()} {new Date(s.date).toLocaleTimeString([], {hour:'2-digit', minute:'2-digit'})}</span>
|
{editingSession === s.id ? (
|
||||||
<span className="session-id">{s.id}</span>
|
<input
|
||||||
|
className="rename-input"
|
||||||
|
value={renameText}
|
||||||
|
onChange={e => setRenameText(e.target.value)}
|
||||||
|
onKeyDown={handleRename}
|
||||||
|
onBlur={() => setEditingSession(null)}
|
||||||
|
autoFocus
|
||||||
|
onClick={e => e.stopPropagation()}
|
||||||
|
/>
|
||||||
|
) : (
|
||||||
|
<>
|
||||||
|
<div className="session-info">
|
||||||
|
<span className="session-date">{s.title || (new Date(s.date).toLocaleDateString() + ' ' + new Date(s.date).toLocaleTimeString([], {hour:'2-digit', minute:'2-digit'}))}</span>
|
||||||
|
{!s.title && <span className="session-id">{s.id}</span>}
|
||||||
|
</div>
|
||||||
|
<div style={{display:'flex'}}>
|
||||||
|
<span className="edit-icon" onClick={(e) => {
|
||||||
|
e.stopPropagation();
|
||||||
|
setEditingSession(s.id);
|
||||||
|
setRenameText(s.title || "");
|
||||||
|
}}>✏️</span>
|
||||||
|
<span className="delete-icon" onClick={(e) => handleDeleteSession(e, s.id)}>🗑️</span>
|
||||||
|
</div>
|
||||||
|
</>
|
||||||
|
)}
|
||||||
</div>
|
</div>
|
||||||
))}
|
))}
|
||||||
</div>
|
</div>
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue