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()
|