Gunroar commited on
Commit
427d810
·
1 Parent(s): 691bbb8
Files changed (1) hide show
  1. 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
- result = out[0]["translation_text"]
40
- print(f"DEBUG: Translation result: {result}")
41
- return result
42
  except Exception as e:
43
- error_msg = f"Translation error: {str(e)}"
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],