Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -3,6 +3,7 @@ from transformers import pipeline
|
|
3 |
import langdetect
|
4 |
import logging
|
5 |
import os
|
|
|
6 |
|
7 |
# Set environment variables for Hugging Face cache
|
8 |
# Ini penting agar model di-cache di lokasi yang benar di dalam container Hugging Face Space
|
@@ -44,57 +45,67 @@ except Exception as e:
|
|
44 |
|
45 |
---
|
46 |
|
47 |
-
## Fungsi Deteksi Bahasa yang Ditingkatkan
|
48 |
|
49 |
-
Ini adalah bagian kunci perbaikannya
|
50 |
|
51 |
```python
|
52 |
def detect_language(text: str) -> str:
|
53 |
try:
|
54 |
detected_lang = langdetect.detect(text)
|
55 |
-
|
|
|
56 |
|
57 |
# Normalisasi untuk bahasa Mandarin:
|
58 |
# Jika langdetect mengembalikan 'zh-cn', 'zh-tw', 'zh-hk', dll.,
|
59 |
# kita paksa menjadi 'zh' agar sesuai dengan kunci di MODEL_MAP.
|
60 |
if detected_lang.startswith('zh'):
|
|
|
61 |
return 'zh'
|
62 |
|
63 |
# Jika bahasa terdeteksi ada di MODEL_MAP, gunakan itu.
|
64 |
# Jika tidak, default ke 'en' (bahasa Inggris).
|
65 |
-
|
|
|
|
|
66 |
except Exception as e:
|
67 |
-
logger.warning(f"Language detection
|
68 |
return "en"
|
69 |
|
70 |
---
|
71 |
|
72 |
-
## Endpoint API untuk Terjemahan
|
|
|
|
|
73 |
|
74 |
```python
|
75 |
@app.post("/translate")
|
76 |
-
async def translate(text: str):
|
77 |
"""
|
78 |
Menerima teks dan mengembalikannya dalam bahasa Inggris.
|
79 |
-
Secara otomatis mendeteksi bahasa sumber.
|
80 |
"""
|
81 |
if not text:
|
82 |
raise HTTPException(status_code=400, detail="Text input is required.")
|
83 |
|
84 |
try:
|
85 |
-
#
|
86 |
-
|
87 |
-
|
|
|
|
|
|
|
|
|
88 |
|
89 |
# Jika bahasa sumber sudah Bahasa Inggris, kembalikan teks aslinya
|
90 |
if source_lang == "en":
|
91 |
-
logger.info("Source language is English, returning original text.")
|
92 |
return {"translated_text": text}
|
93 |
|
94 |
# Dapatkan translator yang sesuai dari kamus translators
|
95 |
translator = translators.get(source_lang)
|
96 |
|
97 |
-
# Jika tidak ada translator yang mendukung bahasa yang terdeteksi
|
98 |
if not translator:
|
99 |
logger.error(f"No translator found for language: '{source_lang}'.")
|
100 |
raise HTTPException(
|
@@ -103,7 +114,7 @@ async def translate(text: str):
|
|
103 |
)
|
104 |
|
105 |
# Lakukan terjemahan
|
106 |
-
logger.info(f"Translating from {source_lang}...")
|
107 |
result = translator(text)
|
108 |
translated_text = result[0]["translation_text"]
|
109 |
logger.info(f"Translation successful. Original: '{text[:50]}...', Translated: '{translated_text[:50]}...'")
|
|
|
3 |
import langdetect
|
4 |
import logging
|
5 |
import os
|
6 |
+
from typing import Optional # Import Optional untuk parameter di FastAPI
|
7 |
|
8 |
# Set environment variables for Hugging Face cache
|
9 |
# Ini penting agar model di-cache di lokasi yang benar di dalam container Hugging Face Space
|
|
|
45 |
|
46 |
---
|
47 |
|
48 |
+
## Fungsi Deteksi Bahasa yang Ditingkatkan (dengan Logging Detail)
|
49 |
|
50 |
+
Ini adalah bagian kunci perbaikannya dengan penambahan *logging* yang lebih informatif.
|
51 |
|
52 |
```python
|
53 |
def detect_language(text: str) -> str:
|
54 |
try:
|
55 |
detected_lang = langdetect.detect(text)
|
56 |
+
# Log ini SANGAT PENTING untuk debugging! Ini menunjukkan hasil mentah dari langdetect.
|
57 |
+
logger.info(f"langdetect raw result: '{detected_lang}' for text: '{text[:50]}...'")
|
58 |
|
59 |
# Normalisasi untuk bahasa Mandarin:
|
60 |
# Jika langdetect mengembalikan 'zh-cn', 'zh-tw', 'zh-hk', dll.,
|
61 |
# kita paksa menjadi 'zh' agar sesuai dengan kunci di MODEL_MAP.
|
62 |
if detected_lang.startswith('zh'):
|
63 |
+
logger.info(f"Normalizing '{detected_lang}' to 'zh' for Mandarin.")
|
64 |
return 'zh'
|
65 |
|
66 |
# Jika bahasa terdeteksi ada di MODEL_MAP, gunakan itu.
|
67 |
# Jika tidak, default ke 'en' (bahasa Inggris).
|
68 |
+
final_lang = detected_lang if detected_lang in MODEL_MAP else "en"
|
69 |
+
logger.info(f"Final determined language: '{final_lang}'. (Based on raw detected: '{detected_lang}')")
|
70 |
+
return final_lang
|
71 |
except Exception as e:
|
72 |
+
logger.warning(f"Language detection FAILED for text: '{text[:50]}...'. Error: {str(e)}. Defaulting to English.")
|
73 |
return "en"
|
74 |
|
75 |
---
|
76 |
|
77 |
+
## Endpoint API untuk Terjemahan (dengan Opsi Override Bahasa Sumber)
|
78 |
+
|
79 |
+
Ini adalah bagian kunci perbaikannya dengan penambahan parameter `source_lang_override`.
|
80 |
|
81 |
```python
|
82 |
@app.post("/translate")
|
83 |
+
async def translate(text: str, source_lang_override: Optional[str] = None):
|
84 |
"""
|
85 |
Menerima teks dan mengembalikannya dalam bahasa Inggris.
|
86 |
+
Secara otomatis mendeteksi bahasa sumber, atau bisa di-override oleh pengguna.
|
87 |
"""
|
88 |
if not text:
|
89 |
raise HTTPException(status_code=400, detail="Text input is required.")
|
90 |
|
91 |
try:
|
92 |
+
# Tentukan bahasa sumber: gunakan override jika diberikan dan valid, kalau tidak, deteksi otomatis.
|
93 |
+
if source_lang_override and source_lang_override in MODEL_MAP:
|
94 |
+
source_lang = source_lang_override
|
95 |
+
logger.info(f"Source language overridden by user to: '{source_lang_override}'.")
|
96 |
+
else:
|
97 |
+
source_lang = detect_language(text)
|
98 |
+
logger.info(f"Determined source language for translation: '{source_lang}'.")
|
99 |
|
100 |
# Jika bahasa sumber sudah Bahasa Inggris, kembalikan teks aslinya
|
101 |
if source_lang == "en":
|
102 |
+
logger.info("Source language is English or unrecognized, returning original text.")
|
103 |
return {"translated_text": text}
|
104 |
|
105 |
# Dapatkan translator yang sesuai dari kamus translators
|
106 |
translator = translators.get(source_lang)
|
107 |
|
108 |
+
# Jika tidak ada translator yang mendukung bahasa yang terdeteksi/di-override
|
109 |
if not translator:
|
110 |
logger.error(f"No translator found for language: '{source_lang}'.")
|
111 |
raise HTTPException(
|
|
|
114 |
)
|
115 |
|
116 |
# Lakukan terjemahan
|
117 |
+
logger.info(f"Translating text from '{source_lang}' to English...")
|
118 |
result = translator(text)
|
119 |
translated_text = result[0]["translation_text"]
|
120 |
logger.info(f"Translation successful. Original: '{text[:50]}...', Translated: '{translated_text[:50]}...'")
|