davideuler
commited on
Commit
·
2ebcead
1
Parent(s):
2527e5c
fix for switching between Google and OpenAI Compatible LLM as translator
Browse files
app.py
CHANGED
|
@@ -359,9 +359,16 @@ def main():
|
|
| 359 |
)
|
| 360 |
target_lang_code = LANGUAGE_OPTIONS[target_lang]
|
| 361 |
|
| 362 |
-
#
|
| 363 |
-
st.subheader("
|
| 364 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 365 |
api_key = st.text_input(
|
| 366 |
"API Key",
|
| 367 |
value=TRANSLATOR_CONFIG["openai"]["default_api_key"],
|
|
@@ -383,12 +390,9 @@ def main():
|
|
| 383 |
'model': model
|
| 384 |
})
|
| 385 |
else: # Google Translator
|
| 386 |
-
|
| 387 |
-
"API Base URL",
|
| 388 |
-
value=TRANSLATOR_CONFIG["google"]["default_api_base"]
|
| 389 |
-
)
|
| 390 |
st.session_state.api_settings.update({
|
| 391 |
-
'api_base':
|
| 392 |
})
|
| 393 |
|
| 394 |
# Main content area
|
|
@@ -413,8 +417,8 @@ def main():
|
|
| 413 |
st.header("Translated")
|
| 414 |
|
| 415 |
try:
|
| 416 |
-
# Initialize translator based on
|
| 417 |
-
if
|
| 418 |
translator = GoogleTranslator(
|
| 419 |
source=source_lang,
|
| 420 |
target=target_lang_code
|
|
@@ -436,7 +440,7 @@ def main():
|
|
| 436 |
pages_per_load,
|
| 437 |
translator,
|
| 438 |
text_color,
|
| 439 |
-
|
| 440 |
target_lang_code
|
| 441 |
)
|
| 442 |
|
|
@@ -482,8 +486,8 @@ def main():
|
|
| 482 |
disabled=st.session_state.all_translated,
|
| 483 |
use_container_width=True):
|
| 484 |
try:
|
| 485 |
-
# Initialize translator based on
|
| 486 |
-
if
|
| 487 |
translator = GoogleTranslator(
|
| 488 |
source=source_lang,
|
| 489 |
target=target_lang_code
|
|
@@ -507,7 +511,7 @@ def main():
|
|
| 507 |
st.empty(),
|
| 508 |
pages_per_load,
|
| 509 |
text_color=text_color,
|
| 510 |
-
translator_name=
|
| 511 |
target_lang=target_lang_code,
|
| 512 |
output_path=output_path
|
| 513 |
)
|
|
@@ -558,7 +562,6 @@ def main():
|
|
| 558 |
|
| 559 |
|
| 560 |
if __name__ == "__main__":
|
| 561 |
-
|
| 562 |
args = parse_args()
|
| 563 |
update_translator_config(args)
|
| 564 |
main()
|
|
|
|
| 359 |
)
|
| 360 |
target_lang_code = LANGUAGE_OPTIONS[target_lang]
|
| 361 |
|
| 362 |
+
# Add translator selection
|
| 363 |
+
st.subheader("Translator Settings")
|
| 364 |
+
translator_type = st.radio(
|
| 365 |
+
"Translator",
|
| 366 |
+
options=["Google", "OpenAI Compatible"],
|
| 367 |
+
index=0 if TRANSLATOR_CONFIG["type"] == "Google" else 1
|
| 368 |
+
)
|
| 369 |
+
|
| 370 |
+
# API Configuration based on translator selection
|
| 371 |
+
if translator_type == "OpenAI Compatible":
|
| 372 |
api_key = st.text_input(
|
| 373 |
"API Key",
|
| 374 |
value=TRANSLATOR_CONFIG["openai"]["default_api_key"],
|
|
|
|
| 390 |
'model': model
|
| 391 |
})
|
| 392 |
else: # Google Translator
|
| 393 |
+
# No configuration needed for Google Translator
|
|
|
|
|
|
|
|
|
|
| 394 |
st.session_state.api_settings.update({
|
| 395 |
+
'api_base': TRANSLATOR_CONFIG["google"]["default_api_base"]
|
| 396 |
})
|
| 397 |
|
| 398 |
# Main content area
|
|
|
|
| 417 |
st.header("Translated")
|
| 418 |
|
| 419 |
try:
|
| 420 |
+
# Initialize translator based on user selection
|
| 421 |
+
if translator_type == "Google":
|
| 422 |
translator = GoogleTranslator(
|
| 423 |
source=source_lang,
|
| 424 |
target=target_lang_code
|
|
|
|
| 440 |
pages_per_load,
|
| 441 |
translator,
|
| 442 |
text_color,
|
| 443 |
+
translator_type,
|
| 444 |
target_lang_code
|
| 445 |
)
|
| 446 |
|
|
|
|
| 486 |
disabled=st.session_state.all_translated,
|
| 487 |
use_container_width=True):
|
| 488 |
try:
|
| 489 |
+
# Initialize translator based on user selection
|
| 490 |
+
if translator_type == "Google":
|
| 491 |
translator = GoogleTranslator(
|
| 492 |
source=source_lang,
|
| 493 |
target=target_lang_code
|
|
|
|
| 511 |
st.empty(),
|
| 512 |
pages_per_load,
|
| 513 |
text_color=text_color,
|
| 514 |
+
translator_name=translator_type,
|
| 515 |
target_lang=target_lang_code,
|
| 516 |
output_path=output_path
|
| 517 |
)
|
|
|
|
| 562 |
|
| 563 |
|
| 564 |
if __name__ == "__main__":
|
|
|
|
| 565 |
args = parse_args()
|
| 566 |
update_translator_config(args)
|
| 567 |
main()
|