File size: 3,524 Bytes
e84d389 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 | # app/routes/sessions.py
from fastapi import APIRouter, Depends, HTTPException
from typing import List, Dict, Any
import os
from ..dependencies import get_current_user
from ..db.mongodb import mongodb
from ..db.chat_manager import chat_manager
from ..config import settings
router = APIRouter()
@router.get("/sessions", response_model=List[Dict[str, Any]])
async def list_sessions(current_user = Depends(get_current_user)):
"""
List all video sessions for the current user.
"""
videos = list(mongodb.videos.find({"user_id": current_user.username}))
sessions_list = []
for v in videos:
sessions_list.append({
"session_id": v["video_id"],
"title": v["title"],
"source_type": v["source_type"],
"created_at": v["created_at"],
"transcription_preview": (v["transcription"][:200] + "...") if len(v["transcription"]) > 200 else v["transcription"]
})
return sessions_list
@router.get("/sessions/{session_id}", response_model=Dict[str, Any])
async def get_session(session_id: str, current_user = Depends(get_current_user)):
"""
Retrieve details and chat history for a specific session.
"""
video = mongodb.videos.find_one({"video_id": session_id})
if not video:
raise HTTPException(status_code=404, detail="Session not found")
if video.get("user_id") != current_user.username:
raise HTTPException(status_code=403, detail="Not authorized to access this session")
# Fetch chat history
history = chat_manager.get_chat_history(session_id)
chat_messages = []
if history:
msgs = history.messages
for i in range(0, len(msgs) - 1, 2):
chat_messages.append({
"question": msgs[i].content,
"answer": msgs[i+1].content
})
return {
"session_id": session_id,
"title": video["title"],
"source_type": video["source_type"],
"source_url": video.get("source_url"),
"created_at": video["created_at"],
"transcription_preview": (video["transcription"][:200] + "...") if len(video["transcription"]) > 200 else video["transcription"],
"full_transcription": video["transcription"],
"chat_history": chat_messages
}
@router.delete("/sessions/{session_id}")
async def delete_session(session_id: str, current_user = Depends(get_current_user)):
"""
Delete a session, its chunks, chat history, and associated video file.
"""
video = mongodb.videos.find_one({"video_id": session_id})
if not video:
raise HTTPException(status_code=404, detail="Session not found")
if video.get("user_id") != current_user.username:
raise HTTPException(status_code=403, detail="Not authorized to delete this session")
# Delete video metadata
mongodb.videos.delete_one({"video_id": session_id})
# Delete chunks
mongodb.db.get_collection("chunks").delete_many({"session_id": session_id})
# Delete chat history
history = chat_manager.get_chat_history(session_id)
if history:
mongodb.db.get_collection(settings.COLLECTION_NAME).delete_many({"session_id": session_id})
# Delete video file(s)
video_files = [f for f in os.listdir(settings.VIDEOS_DIR) if f.startswith(session_id)]
for file in video_files:
try:
os.remove(os.path.join(settings.VIDEOS_DIR, file))
except OSError:
pass
return {"message": f"Session {session_id} deleted successfully"}
|