Spaces:
Runtime error
Runtime error
File size: 4,011 Bytes
9487f03 b13884f 24d2a22 b13884f 9487f03 bcc800f b13884f 8c83892 b13884f 9487f03 b13884f 9487f03 b13884f 8c83892 b13884f 8c83892 b13884f 8c83892 b13884f 9487f03 b13884f 9487f03 b13884f 9487f03 b13884f 8c83892 b13884f 9171aa1 b13884f |
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 |
from fastapi import FastAPI, File, UploadFile, Form
from fastapi.responses import JSONResponse
from enum import Enum
from transformers import pipeline, MarianMTModel, MarianTokenizer
import shutil
import os
import uuid
import uvicorn
from googletrans import Translator
os.environ["HF_HOME"] = "/app/.cache/huggingface"
app = FastAPI()
# 🎯 Hugging Face Pipelines
asr_pipeline = pipeline("automatic-speech-recognition", model="openai/whisper-medium")
generator_pipeline = pipeline("text-generation", model="distilgpt2")
# 🌍 Language Enum for dropdown in Swagger
class LanguageEnum(str, Enum):
ta = "ta" # Tamil
fr = "fr" # French
es = "es" # Spanish
de = "de" # German
it = "it" # Italian
hi = "hi" # Hindi
ru = "ru" # Russian
zh = "zh" # Chinese
ar = "ar" # Arabic
# 🌐 Map target language to translation model
model_map = {
"fr": "Helsinki-NLP/opus-mt-en-fr",
"es": "Helsinki-NLP/opus-mt-en-es",
"de": "Helsinki-NLP/opus-mt-en-de",
"it": "Helsinki-NLP/opus-mt-en-it",
"hi": "Helsinki-NLP/opus-mt-en-hi",
"ru": "Helsinki-NLP/opus-mt-en-ru",
"zh": "Helsinki-NLP/opus-mt-en-zh",
"ar": "Helsinki-NLP/opus-mt-en-ar",
"ta": "gsarti/opus-mt-en-ta"
}
def translate_text(text, target_lang):
if target_lang == "ta":
# Use Google Translate for Tamil
try:
translator = Translator()
result = translator.translate(text, dest="ta")
return result.text
except Exception as e:
return f"Google Translate failed: {str(e)}"
# Use MarianMT for other supported languages
if target_lang not in model_map:
return f"No model for language: {target_lang}"
model_name = model_map[target_lang]
tokenizer = MarianTokenizer.from_pretrained(model_name)
model = MarianMTModel.from_pretrained(model_name)
encoded = tokenizer([text], return_tensors="pt", padding=True)
translated = model.generate(**encoded)
return tokenizer.batch_decode(translated, skip_special_tokens=True)[0]
# 🧠 Generate a random English sentence
def generate_random_sentence(prompt="Daily conversation", max_length=30):
result = generator_pipeline(prompt, max_length=max_length, num_return_sequences=1)
return result[0]["generated_text"].strip()
# 🎤 Transcription endpoint
@app.post("/transcribe")
async def transcribe(audio: UploadFile = File(...)):
temp_filename = f"temp_{uuid.uuid4().hex}.wav"
with open(temp_filename, "wb") as f:
shutil.copyfileobj(audio.file, f)
try:
result = asr_pipeline(temp_filename)
return JSONResponse(content={"transcribed_text": result["text"]})
finally:
os.remove(temp_filename)
# 🌍 Translation endpoint
@app.post("/translate")
async def translate(text: str = Form(...), target_lang: LanguageEnum = Form(...)):
translated = translate_text(text, target_lang.value)
return JSONResponse(content={"translated_text": translated})
# 🔁 Combined endpoint (speech-to-translation)
@app.post("/process")
async def process(audio: UploadFile = File(...), target_lang: LanguageEnum = Form(...)):
temp_filename = f"temp_{uuid.uuid4().hex}.wav"
with open(temp_filename, "wb") as f:
shutil.copyfileobj(audio.file, f)
try:
result = asr_pipeline(temp_filename)
transcribed_text = result["text"]
translated_text = translate_text(transcribed_text, target_lang.value)
return JSONResponse(content={
"transcribed_text": transcribed_text,
"translated_text": translated_text
})
finally:
os.remove(temp_filename)
# ✨ Generate + Translate endpoint
@app.get("/generate")
def generate(prompt: str = "Daily conversation", target_lang: LanguageEnum = LanguageEnum.it):
english = generate_random_sentence(prompt)
translated = translate_text(english, target_lang.value)
return {
"prompt": prompt,
"english": english,
"translated": translated
}
|