Floke
da00d461e1
- Backend: Switched prompt to JSON output for structured data
- Backend: Added PUT /chunks/{id} endpoint for persistence
- Backend: Fixed app.py imports and initialization logic
- Frontend: Complete rewrite for Unified View (flattened chunks)
- Frontend: Added Inline Editing (Text/Speaker) and Row Deletion
- Docs: Updated TRANSCRIPTION_TOOL.md with v0.4 features
136 lines
4.1 KiB
Python
136 lines
4.1 KiB
Python
from fastapi import FastAPI, Depends, HTTPException, UploadFile, File, BackgroundTasks, Body
|
|
from fastapi.staticfiles import StaticFiles
|
|
from fastapi.middleware.cors import CORSMiddleware
|
|
from sqlalchemy.orm import Session, joinedload
|
|
from typing import List, Dict, Any
|
|
import os
|
|
import shutil
|
|
import uuid
|
|
from datetime import datetime
|
|
|
|
from .config import settings
|
|
from .database import init_db, get_db, Meeting, TranscriptChunk, AnalysisResult, SessionLocal
|
|
from .services.orchestrator import process_meeting_task
|
|
|
|
# Initialize FastAPI App
|
|
app = FastAPI(
|
|
title=settings.APP_NAME,
|
|
version=settings.VERSION,
|
|
root_path="/tr"
|
|
)
|
|
|
|
app.add_middleware(
|
|
CORSMiddleware,
|
|
allow_origins=["*"],
|
|
allow_credentials=True,
|
|
allow_methods=["*"],
|
|
allow_headers=["*"],
|
|
)
|
|
|
|
@app.on_event("startup")
|
|
def startup_event():
|
|
init_db()
|
|
|
|
@app.get("/api/health")
|
|
def health():
|
|
return {"status": "ok", "version": settings.VERSION}
|
|
|
|
@app.get("/api/meetings")
|
|
def list_meetings(db: Session = Depends(get_db)):
|
|
return db.query(Meeting).order_by(Meeting.created_at.desc()).all()
|
|
|
|
@app.get("/api/meetings/{meeting_id}")
|
|
def get_meeting(meeting_id: int, db: Session = Depends(get_db)):
|
|
meeting = db.query(Meeting).options(
|
|
joinedload(Meeting.chunks)
|
|
).filter(Meeting.id == meeting_id).first()
|
|
|
|
if not meeting:
|
|
raise HTTPException(404, detail="Meeting not found")
|
|
|
|
# Sort chunks by index
|
|
meeting.chunks.sort(key=lambda x: x.chunk_index)
|
|
|
|
return meeting
|
|
|
|
@app.put("/api/chunks/{chunk_id}")
|
|
def update_chunk(chunk_id: int, payload: Dict[str, Any] = Body(...), db: Session = Depends(get_db)):
|
|
chunk = db.query(TranscriptChunk).filter(TranscriptChunk.id == chunk_id).first()
|
|
if not chunk:
|
|
raise HTTPException(404, detail="Chunk not found")
|
|
|
|
# Update JSON content (e.g. after editing/deleting lines)
|
|
if "json_content" in payload:
|
|
chunk.json_content = payload["json_content"]
|
|
db.commit()
|
|
|
|
return {"status": "updated"}
|
|
|
|
@app.post("/api/upload")
|
|
async def upload_audio(
|
|
background_tasks: BackgroundTasks,
|
|
file: UploadFile = File(...),
|
|
db: Session = Depends(get_db)
|
|
):
|
|
# 1. Save File
|
|
file_id = str(uuid.uuid4())
|
|
ext = os.path.splitext(file.filename)[1]
|
|
filename = f"{file_id}{ext}"
|
|
file_path = os.path.join(settings.UPLOAD_DIR, filename)
|
|
|
|
with open(file_path, "wb") as buffer:
|
|
shutil.copyfileobj(file.file, buffer)
|
|
|
|
# 2. Create DB Entry
|
|
meeting = Meeting(
|
|
title=file.filename,
|
|
filename=filename,
|
|
file_path=file_path,
|
|
status="UPLOADED"
|
|
)
|
|
db.add(meeting)
|
|
db.commit()
|
|
db.refresh(meeting)
|
|
|
|
# 3. Trigger Processing in Background
|
|
background_tasks.add_task(process_meeting_task, meeting.id, SessionLocal)
|
|
|
|
return meeting
|
|
|
|
@app.delete("/api/meetings/{meeting_id}")
|
|
def delete_meeting(meeting_id: int, db: Session = Depends(get_db)):
|
|
meeting = db.query(Meeting).filter(Meeting.id == meeting_id).first()
|
|
if not meeting:
|
|
raise HTTPException(404, detail="Meeting not found")
|
|
|
|
# 1. Delete Files
|
|
try:
|
|
if os.path.exists(meeting.file_path):
|
|
os.remove(meeting.file_path)
|
|
|
|
# Delete chunks dir
|
|
chunk_dir = os.path.join(settings.UPLOAD_DIR, "chunks", str(meeting_id))
|
|
if os.path.exists(chunk_dir):
|
|
shutil.rmtree(chunk_dir)
|
|
except Exception as e:
|
|
print(f"Error deleting files: {e}")
|
|
|
|
# 2. Delete DB Entry (Cascade deletes chunks/analyses)
|
|
db.delete(meeting)
|
|
db.commit()
|
|
return {"status": "deleted"}
|
|
|
|
# Serve Frontend
|
|
# This must be the last route definition to avoid catching API routes
|
|
static_path = "/frontend_static"
|
|
if not os.path.exists(static_path):
|
|
# Fallback for local development if not in Docker
|
|
static_path = os.path.join(os.path.dirname(__file__), "../frontend/dist")
|
|
|
|
if os.path.exists(static_path):
|
|
app.mount("/", StaticFiles(directory=static_path, html=True), name="static")
|
|
|
|
if __name__ == "__main__":
|
|
import uvicorn
|
|
uvicorn.run("backend.app:app", host="0.0.0.0", port=8001, reload=True)
|