File size: 1,660 Bytes
805e1e5
 
0117df3
805e1e5
 
 
 
 
 
 
 
eb5aac2
805e1e5
 
 
 
0117df3
90debd3
0117df3
 
 
 
90d3850
805e1e5
 
0117df3
 
 
805e1e5
 
0117df3
 
90d3850
805e1e5
 
88da32f
 
 
 
31fda96
 
805e1e5
31fda96
eb5aac2
90debd3
 
805e1e5
 
90debd3
 
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
from fastapi import APIRouter, Depends, HTTPException, UploadFile, File, Request
from fastapi.security import HTTPBearer
from pydantic import BaseModel
from slowapi.util import get_remote_address
from slowapi import Limiter
from slowapi.errors import RateLimitExceeded
from config import ACCESS_RATE
from .controller import (
    handle_text_analysis,
    handle_file_upload,
    handle_sentence_level_analysis,
    handle_file_sentence,
    verify_token
)

limiter = Limiter(key_func=get_remote_address)
router = APIRouter()
security = HTTPBearer()

class TextInput(BaseModel):
    text: str

@router.post("/analyse")
@limiter.limit(ACCESS_RATE)
async def analyze(request: Request, data: TextInput, token: str = Depends(verify_token)):
    return await handle_text_analysis(data.text)

@router.post("/upload")
@limiter.limit(ACCESS_RATE)
async def upload_file(request: Request, file: UploadFile = File(...), token: str = Depends(verify_token)):
    return await handle_file_upload(file)

@router.post("/analyse-sentences")
@limiter.limit(ACCESS_RATE)
async def analyze_sentences(request: Request, data: TextInput, token: str = Depends(verify_token)):
    if not data.text:
        raise HTTPException(status_code=400, detail="Missing 'text' in request body")
    return await handle_sentence_level_analysis(data.text)


@router.post("/analyse-sentence-file")
@limiter.limit(ACCESS_RATE)
async def analyze_sentence_file(request: Request, file: UploadFile = File(...), token: str = Depends(verify_token)):
    return await handle_file_sentence(file)

@router.get("/health")
@limiter.limit(ACCESS_RATE)
def health(request: Request):
    return {"status": "ok"}