Spaces:
Sleeping
Sleeping
okay
Browse files- multilingual/multi.py +2 -13
multilingual/multi.py
CHANGED
|
@@ -33,16 +33,10 @@ class Query(BaseModel):
|
|
| 33 |
|
| 34 |
def translate(text: str, src: str, tgt: str) -> str:
|
| 35 |
try:
|
| 36 |
-
print(f"DEBUG: Translating from {src} ({lang_codes[src]}) to {tgt} ({lang_codes[tgt]})")
|
| 37 |
-
print(f"DEBUG: Input text: {text}")
|
| 38 |
out = translator(text, src_lang=lang_codes[src], tgt_lang=lang_codes[tgt], max_length=400)
|
| 39 |
-
|
| 40 |
-
print(f"DEBUG: Translation result: {result}")
|
| 41 |
-
return result
|
| 42 |
except Exception as e:
|
| 43 |
-
|
| 44 |
-
print(f"DEBUG: {error_msg}")
|
| 45 |
-
return error_msg
|
| 46 |
|
| 47 |
def query_milestone1(prompt: str) -> str:
|
| 48 |
try:
|
|
@@ -107,22 +101,17 @@ def multilingual_ask(query: Query):
|
|
| 107 |
# Translate to English if not English
|
| 108 |
if user_lang != "en":
|
| 109 |
english_text = translate(user_text, user_lang, "en")
|
| 110 |
-
print(f"DEBUG: Translated to English: {english_text}")
|
| 111 |
else:
|
| 112 |
english_text = user_text
|
| 113 |
-
print(f"DEBUG: Using original English text: {english_text}")
|
| 114 |
|
| 115 |
# Get response from Milestone 1
|
| 116 |
english_response = query_milestone1(english_text)
|
| 117 |
-
print(f"DEBUG: Milestone 1 response: {english_response}")
|
| 118 |
|
| 119 |
# Translate response back to original language
|
| 120 |
if user_lang != "en":
|
| 121 |
final_response = translate(english_response, "en", user_lang)
|
| 122 |
-
print(f"DEBUG: Translated back to {user_lang}: {final_response}")
|
| 123 |
else:
|
| 124 |
final_response = english_response
|
| 125 |
-
print(f"DEBUG: Using English response: {final_response}")
|
| 126 |
|
| 127 |
return {
|
| 128 |
"language": lang_names[user_lang],
|
|
|
|
| 33 |
|
| 34 |
def translate(text: str, src: str, tgt: str) -> str:
|
| 35 |
try:
|
|
|
|
|
|
|
| 36 |
out = translator(text, src_lang=lang_codes[src], tgt_lang=lang_codes[tgt], max_length=400)
|
| 37 |
+
return out[0]["translation_text"]
|
|
|
|
|
|
|
| 38 |
except Exception as e:
|
| 39 |
+
return f"Translation error: {str(e)}"
|
|
|
|
|
|
|
| 40 |
|
| 41 |
def query_milestone1(prompt: str) -> str:
|
| 42 |
try:
|
|
|
|
| 101 |
# Translate to English if not English
|
| 102 |
if user_lang != "en":
|
| 103 |
english_text = translate(user_text, user_lang, "en")
|
|
|
|
| 104 |
else:
|
| 105 |
english_text = user_text
|
|
|
|
| 106 |
|
| 107 |
# Get response from Milestone 1
|
| 108 |
english_response = query_milestone1(english_text)
|
|
|
|
| 109 |
|
| 110 |
# Translate response back to original language
|
| 111 |
if user_lang != "en":
|
| 112 |
final_response = translate(english_response, "en", user_lang)
|
|
|
|
| 113 |
else:
|
| 114 |
final_response = english_response
|
|
|
|
| 115 |
|
| 116 |
return {
|
| 117 |
"language": lang_names[user_lang],
|