jeevitha-app commited on
Commit
7160af6
·
verified ·
1 Parent(s): 280a6e7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -7
app.py CHANGED
@@ -90,17 +90,19 @@ def save_output_to_file(detected_lang, translated, sem_results, bleu_score):
90
  f.write(f"\nBLEU Score: {bleu_score}")
91
  return f.name
92
 
93
- # Main Pipeline
94
  def full_pipeline(user_input_text, target_lang_code, human_ref=""):
95
  if not user_input_text.strip():
96
- return "Empty input", "", [], "", ""
 
 
 
97
 
98
  detected_lang = detect_language(user_input_text)
99
  src_nllb = xlm_to_nllb.get(detected_lang, "eng_Latn")
100
 
101
  translated = translate(user_input_text, src_nllb, target_lang_code)
102
  if not translated:
103
- return detected_lang, " Translation failed", [], "", ""
104
 
105
  sem_results = search_semantic(translated)
106
  result_list = [f"{i+1}. {txt} (Score: {score:.2f})" for i, (txt, score) in enumerate(sem_results)]
@@ -125,12 +127,8 @@ def full_pipeline(user_input_text, target_lang_code, human_ref=""):
125
  bleu = corpus_bleu([translated], [[human_ref]])
126
  bleu_score = f"{bleu.score:.2f}"
127
 
128
- # Create downloadable result file
129
  download_file_path = save_output_to_file(detected_lang, translated, sem_results, bleu_score)
130
  return detected_lang, translated, "\n".join(result_list), plot_path, bleu_score, download_file_path
131
- if len(user_input_text) > 2048:
132
- return "Input too long", "Please enter shorter text (under 2000 characters).", [], "", "", None
133
-
134
 
135
 
136
  # Gradio Interface
 
90
  f.write(f"\nBLEU Score: {bleu_score}")
91
  return f.name
92
 
 
93
  def full_pipeline(user_input_text, target_lang_code, human_ref=""):
94
  if not user_input_text.strip():
95
+ return "Empty input", "", [], "", "", None
96
+
97
+ if len(user_input_text) > 2048:
98
+ return " Input too long", "Please enter shorter text (under 2000 characters).", [], "", "", None
99
 
100
  detected_lang = detect_language(user_input_text)
101
  src_nllb = xlm_to_nllb.get(detected_lang, "eng_Latn")
102
 
103
  translated = translate(user_input_text, src_nllb, target_lang_code)
104
  if not translated:
105
+ return detected_lang, " Translation failed", [], "", "", None
106
 
107
  sem_results = search_semantic(translated)
108
  result_list = [f"{i+1}. {txt} (Score: {score:.2f})" for i, (txt, score) in enumerate(sem_results)]
 
127
  bleu = corpus_bleu([translated], [[human_ref]])
128
  bleu_score = f"{bleu.score:.2f}"
129
 
 
130
  download_file_path = save_output_to_file(detected_lang, translated, sem_results, bleu_score)
131
  return detected_lang, translated, "\n".join(result_list), plot_path, bleu_score, download_file_path
 
 
 
132
 
133
 
134
  # Gradio Interface