davideuler commited on
Commit
2ebcead
·
1 Parent(s): 2527e5c

fix for switching between Google and OpenAI Compatible LLM as translator

Browse files
Files changed (1) hide show
  1. app.py +18 -15
app.py CHANGED
@@ -359,9 +359,16 @@ def main():
359
  )
360
  target_lang_code = LANGUAGE_OPTIONS[target_lang]
361
 
362
- # API Configuration based on translator type
363
- st.subheader("API Settings")
364
- if TRANSLATOR_CONFIG["type"] == "OpenAI":
 
 
 
 
 
 
 
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
- api_base = st.text_input(
387
- "API Base URL",
388
- value=TRANSLATOR_CONFIG["google"]["default_api_base"]
389
- )
390
  st.session_state.api_settings.update({
391
- 'api_base': 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 global config
417
- if TRANSLATOR_CONFIG["type"] == "Google":
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
- TRANSLATOR_CONFIG["type"],
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 global config
486
- if TRANSLATOR_CONFIG["type"] == "Google":
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=TRANSLATOR_CONFIG["type"],
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()