Update app.py

app.py

@@ -26,7 +26,7 @@ def translate_to_english(prompt):
     translated_prompt = GoogleTranslator(source='auto', target='en').translate(prompt)
     return translated_prompt
 
-def translate_to_persian(response):
+def translate_to_persian_text(response):
     translated_response = GoogleTranslator(source='auto', target='fa').translate(response)
     return translated_response
 
@@ -84,7 +84,7 @@ def chat_inf(system_prompt, prompt, history, memory, client_choice, seed, temp,
         for response in stream:
             output += response.token.text
             if translate_to_persian:
-                output = translate_to_persian(output)
+                output = translate_to_persian_text(output)
             yield [(prompt, output)], memory
         history.append((prompt, output))
         memory.append((prompt, output))
@@ -133,8 +133,8 @@ with gr.Blocks() as app:
     client_choice.change(load_models, client_choice, [chat_b])
     app.load(load_models, client_choice, [chat_b])
 
-    chat_sub = inp.submit(check_rand, [rand, seed], seed).then(chat_inf, [sys_inp, inp, chat_b, memory, client_choice, seed, temp, tokens, top_p, rep_p, chat_mem, custom_prompt, translate_to_persian_checkbox], [chat_b, memory])
-    go = btn.click(check_rand, [rand, seed], seed).then(chat_inf, [sys_inp, inp, chat_b, memory, client_choice, seed, temp, tokens, top_p, rep_p, chat_mem, custom_prompt, translate_to_persian_checkbox], [chat_b, memory])
+    chat_sub = inp.submit(check_rand, [rand, seed], seed).then(chat_inf, [sys_inp, inp, chat_b, memory, client_choice, seed, temp, tokens, top_p, rep_p, chat_mem, custom_prompt, translate_to_persian_checkbox.value], [chat_b, memory])
+    go = btn.click(check_rand, [rand, seed], seed).then(chat_inf, [sys_inp, inp, chat_b, memory, client_choice, seed, temp, tokens, top_p, rep_p, chat_mem, custom_prompt, translate_to_persian_checkbox.value], [chat_b, memory])
 
     clear_btn.click(clear_fn, None, [inp, sys_inp, chat_b, memory])
 app.queue(default_concurrency_limit=10).launch()