Update app.py

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 from gradio_client import Client
 from huggingface_hub import InferenceClient
 import random
+from deep_translator import GoogleTranslator
 ss_client = Client("https://omnibus-html-image-current-tab.hf.space/")
 
 models=[
@@ -121,6 +122,7 @@ with gr.Blocks() as app:
                     custom_prompt=gr.Textbox(label="Modify Prompt Format", info="For testing purposes. 'USER_INPUT' is where 'SYSTEM_PROMPT, PROMPT' will be placed", lines=5,value="<start_of_turn>userUSER_INPUT<end_of_turn><start_of_turn>model")
             with gr.Column(scale=1):
                 with gr.Group():
+                    translate_fa = gr.Checkbox(label="Translate to Persian", value=True)
                     rand = gr.Checkbox(label="Random Seed", value=True)
                     seed=gr.Slider(label="Seed", minimum=1, maximum=1111111111111111,step=1, value=rand_val)
                     tokens = gr.Slider(label="Max new tokens",value=1600,minimum=0,maximum=8000,step=64,interactive=True, visible=True,info="The maximum number of tokens")
@@ -132,7 +134,12 @@ with gr.Blocks() as app:
     
     client_choice.change(load_models,client_choice,[chat_b])
     app.load(load_models,client_choice,[chat_b])
-    
+    if(len(inp) > 2000):
+        translatedtext1 = GoogleTranslator(source='auto', target='en').translate(inp[0:2000])
+        translatedtext2 = GoogleTranslator(source='auto', target='en').translate(inp[2000:(len(inp))])
+        inp = translatedtext1 + translatedtext2
+    else:
+        inp = GoogleTranslator(source='auto', target='en').translate(inp)
     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],[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],[chat_b,memory])