File size: 3,217 Bytes
1a252b6 fc2f017 1a252b6 fc2f017 1a252b6 fc2f017 1a252b6 | 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 92 93 94 95 96 | from fastapi import APIRouter, Request, HTTPException
from fastapi.responses import JSONResponse, FileResponse
import uuid
from app.core.config import settings
from app.db import crud
from app.services.worker import start_worker, is_worker_running
from custom_logger import logger_config as logger
router = APIRouter()
@router.get("/")
async def index():
return FileResponse('index.html')
@router.post("/api/tasks/upload")
async def submit_task(request: Request):
data = await request.json()
if not data or not data.get('text', '').strip():
raise HTTPException(status_code=400, detail="No input text provided")
task_id = str(uuid.uuid4())
input_text = data['text'].strip()
system_prompt = data.get('system_prompt', '').strip() or None
hide_from_ui = 1 if data.get('hide_from_ui') else 0
model = (data.get('model') or 'qwen').strip().lower()
await crud.insert_task(task_id, input_text, system_prompt, 'not_started', hide_from_ui, model)
await start_worker()
return JSONResponse(status_code=201, content={
'id': task_id,
'filename': input_text[:50] + ("..." if len(input_text) > 50 else ""),
'status': 'not_started',
'model': model,
'message': 'Task submitted successfully'
})
@router.get("/api/tasks")
async def get_tasks():
rows, queue_ids, processing_count, avg_time = await crud.get_all_tasks()
tasks = []
for row in rows:
queue_position = None
estimated_start_seconds = None
if row['status'] == 'not_started' and row['id'] in queue_ids:
queue_position = queue_ids.index(row['id']) + 1
tasks_ahead = queue_position - 1 + processing_count
estimated_start_seconds = round(tasks_ahead * avg_time)
tasks.append({
'id': row['id'],
'filename': row['input_text'][:200] + ('...' if len(row['input_text']) > 200 else ''),
'status': row['status'],
'result': "HIDDEN_IN_LIST_VIEW",
'created_at': row['created_at'],
'processed_at': row['processed_at'],
'progress': row['progress'] or 0,
'progress_text': row['progress_text'],
'queue_position': queue_position,
'estimated_start_seconds': estimated_start_seconds
})
return tasks
@router.get("/api/tasks/{task_id}")
async def get_task(task_id: str):
result = await crud.get_task_by_id(task_id)
if not result:
raise HTTPException(status_code=404, detail="Task not found")
row, queue_position, estimated_start_seconds = result
return {
'id': row['id'],
'filename': row['input_text'],
'status': row['status'],
'result': row['result'],
'created_at': row['created_at'],
'processed_at': row['processed_at'],
'progress': row['progress'] or 0,
'progress_text': row['progress_text'],
'queue_position': queue_position,
'estimated_start_seconds': estimated_start_seconds
}
@router.get("/health")
async def health():
return {
'status': 'healthy',
'service': 'ttt-runner',
'worker_running': is_worker_running()
}
|