Spaces:
Running
Running
| from fastapi import FastAPI, HTTPException | |
| from transformers import pipeline | |
| import langdetect | |
| import logging | |
| import os | |
| # Atur direktori cache | |
| os.environ["HF_HOME"] = "/app/cache" | |
| os.environ["TRANSFORMERS_CACHE"] = "/app/cache" | |
| app = FastAPI() | |
| logging.basicConfig(level=logging.INFO) | |
| logger = logging.getLogger(__name__) | |
| # Daftar model untuk bahasa yang didukung | |
| MODEL_MAP = { | |
| "id": "Helsinki-NLP/opus-mt-id-en", # Indonesia ke Inggris | |
| "th": "Helsinki-NLP/opus-mt-th-en", # Thailand ke Inggris | |
| "fr": "Helsinki-NLP/opus-mt-fr-en", # Prancis ke Inggris | |
| "es": "Helsinki-NLP/opus-mt-es-en", # Spanyol ke Inggris | |
| "ja": "Helsinki-NLP/opus-mt-ja-en", # Jepang ke Inggris | |
| "zh": "Helsinki-NLP/opus-mt-zh-en", # Mandarin ke Inggris | |
| } | |
| # Inisialisasi pipeline untuk setiap model | |
| translators = {} | |
| try: | |
| for lang, model_name in MODEL_MAP.items(): | |
| logger.info(f"Loading model for {lang}...") | |
| translators[lang] = pipeline("translation", model=model_name) | |
| logger.info(f"Model for {lang} loaded successfully") | |
| except Exception as e: | |
| logger.error(f"Model initialization failed: {str(e)}") | |
| raise Exception(f"Model initialization failed: {str(e)}") | |
| def detect_language(text: str) -> str: | |
| """Deteksi bahasa dari teks menggunakan langdetect.""" | |
| try: | |
| lang = langdetect.detect(text) | |
| return lang if lang in MODEL_MAP else "en" # Default ke Inggris jika tidak didukung | |
| except Exception as e: | |
| logger.warning(f"Language detection failed: {str(e)}, defaulting to English") | |
| return "en" | |
| async def translate(text: str): | |
| if not text: | |
| raise HTTPException(status_code=400, detail="Text input is required") | |
| try: | |
| # Deteksi bahasa | |
| source_lang = detect_language(text) | |
| logger.info(f"Detected source language: {source_lang}") | |
| # Jika sudah Inggris atau bahasa tidak didukung, kembalikan teks asli | |
| if source_lang == "en": | |
| return {"translated_text": text} | |
| # Terjemahkan ke Inggris | |
| translator = translators.get(source_lang) | |
| if not translator: | |
| raise HTTPException(status_code=400, detail=f"Translation not supported for language: {source_lang}") | |
| result = translator(text) | |
| return {"translated_text": result[0]["translation_text"]} | |
| except Exception as e: | |
| logger.error(f"Processing failed: {str(e)}") | |
| raise HTTPException(status_code=500, detail=f"Processing failed: {str(e)}") |