Spaces:
Sleeping
Sleeping
File size: 5,133 Bytes
d14e502 | 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 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 | """
FastAPI server for transliteration API
Provides RESTful endpoints for programmatic access
"""
from fastapi import FastAPI, HTTPException
from pydantic import BaseModel
from typing import List, Optional
from inference import Transliterator
import os
# Initialize FastAPI app
app = FastAPI(
title="Multilingual Transliteration API",
description="API for transliterating English text to Hindi, Bengali, and Tamil",
version="1.0.0",
)
# Initialize transliterator
MODEL_PATH = "ct2_model" if os.path.exists("ct2_model") else "models/translit"
USE_CT2 = os.path.exists("ct2_model")
trans = Transliterator(model_path=MODEL_PATH, use_ct2=USE_CT2)
# Pydantic models for request/response
class TransliterateRequest(BaseModel):
text: str
lang: str = "hi"
max_length: Optional[int] = 32
num_beams: Optional[int] = 4
class TransliterateBatchRequest(BaseModel):
texts: List[str]
lang: str = "hi"
max_length: Optional[int] = 32
num_beams: Optional[int] = 4
batch_size: Optional[int] = 32
class TransliterateResponse(BaseModel):
input: str
output: str
language: str
success: bool
class TransliterateBatchResponse(BaseModel):
inputs: List[str]
outputs: List[str]
language: str
count: int
success: bool
class TransliterateAllResponse(BaseModel):
input: str
outputs: dict
success: bool
@app.get("/")
def read_root():
"""Root endpoint with API information."""
return {
"message": "Multilingual Transliteration API",
"version": "1.0.0",
"languages": ["hi", "bn", "ta"],
"endpoints": {
"GET /": "This information",
"POST /transliterate": "Transliterate single text",
"POST /transliterate/batch": "Transliterate batch of texts",
"POST /transliterate/all": "Transliterate to all languages",
"GET /languages": "List supported languages",
},
}
@app.get("/languages")
def get_languages():
"""Get list of supported languages."""
return {
"languages": [
{"code": "hi", "name": "Hindi", "native": "हिन्दी"},
{"code": "bn", "name": "Bengali", "native": "বাংলা"},
{"code": "ta", "name": "Tamil", "native": "தமிழ்"},
]
}
@app.post("/transliterate", response_model=TransliterateResponse)
def transliterate_single(request: TransliterateRequest):
"""
Transliterate a single text to the specified language.
- **text**: English text to transliterate
- **lang**: Target language code (hi, bn, ta)
- **max_length**: Maximum output length (default: 32)
- **num_beams**: Number of beams for beam search (default: 4)
"""
try:
result = trans.transliterate(
text=request.text,
lang=request.lang,
max_length=request.max_length,
num_beams=request.num_beams,
)
return TransliterateResponse(
input=request.text, output=result, language=request.lang, success=True
)
except Exception as e:
raise HTTPException(status_code=500, detail=str(e))
@app.post("/transliterate/batch", response_model=TransliterateBatchResponse)
def transliterate_batch(request: TransliterateBatchRequest):
"""
Transliterate a batch of texts to the specified language.
- **texts**: List of English texts to transliterate
- **lang**: Target language code (hi, bn, ta)
- **max_length**: Maximum output length (default: 32)
- **num_beams**: Number of beams for beam search (default: 4)
- **batch_size**: Batch size for processing (default: 32)
"""
try:
results = trans.transliterate_batch(
texts=request.texts,
lang=request.lang,
max_length=request.max_length,
num_beams=request.num_beams,
batch_size=request.batch_size,
)
return TransliterateBatchResponse(
inputs=request.texts,
outputs=results,
language=request.lang,
count=len(results),
success=True,
)
except Exception as e:
raise HTTPException(status_code=500, detail=str(e))
@app.post("/transliterate/all", response_model=TransliterateAllResponse)
def transliterate_all(request: TransliterateRequest):
"""
Transliterate text to all supported languages.
- **text**: English text to transliterate
- **max_length**: Maximum output length (default: 32)
- **num_beams**: Number of beams for beam search (default: 4)
"""
try:
results = trans.transliterate_all(
text=request.text, languages=["hi", "bn", "ta"]
)
return TransliterateAllResponse(
input=request.text, outputs=results, success=True
)
except Exception as e:
raise HTTPException(status_code=500, detail=str(e))
@app.get("/health")
def health_check():
"""Health check endpoint."""
return {"status": "healthy", "model_loaded": True}
if __name__ == "__main__":
import uvicorn
uvicorn.run(app, host="0.0.0.0", port=8000)
|