Update app/webui/app.py

app/webui/app.py

@@ -93,9 +93,9 @@ def read_doc(file):
 
 def enable_sec(choice):
     if choice:
-        return gr.update(visible = True), gr.update(visible = True), gr.update(visible = True)
+        return gr.update(visible = True)
     else:
-        return gr.update(visible = False), gr.update(visible = False), gr.update(visible = False)
+        return gr.update(visible = False)
 
 def update_menu(visible):
     return not visible, gr.update(visible=not visible)
@@ -116,7 +116,7 @@ def switch(source_lang,source_text,target_lang,output_final):
 
 def closeBtnShow():
     return gr.update(visible=False), gr.update(visible=True)
-    
+
 def closeBtnHide(output_final):
     if output_final:
         return gr.update(visible=True), gr.update(visible=False)
@@ -198,17 +198,17 @@ with gr.Blocks(theme="soft", css=CSS, fill_height=True) as demo:
                 choices=["Groq","OpenAI","Cohere","TogetherAI","Ollama","Huggingface"],
                 value="Huggingface",
             )
-            choice = gr.Checkbox(label="Second Endpoint", info="Add second endpoint for reflection")
-            model = gr.Textbox(label="Model", value="mistralai/Mistral-7B-Instruct-v0.3",)
+            choice = gr.Checkbox(label="Additional Endpoint", info="Additional endpoint for reflection")
+            model = gr.Textbox(label="Model", value="mistralai/Mistral-7B-Instruct-v0.3", )
             api_key = gr.Textbox(label="API_KEY", type="password", )
-            endpoint2 = gr.Dropdown(
-                label="Endpoint 2",
-                choices=["Groq","OpenAI","Cohere","TogetherAI","Ollama","Huggingface"],
-                value="Groq",
-                visible=False,
-            )
-            model2 = gr.Textbox(label="Model 2", value="llama3-70b-8192", visible=False,)
-            api_key2 = gr.Textbox(label="API_KEY 2", type="password", visible=False,)
+            with gr.Column(visible=False) as AddEndpoint:
+                endpoint2 = gr.Dropdown(
+                    label="Additional Endpoint",
+                    choices=["Groq","OpenAI","Cohere","TogetherAI","Ollama","Huggingface"],
+                    value="Groq",
+                )
+                model2 = gr.Textbox(label="Model", value="llama3-70b-8192", )
+                api_key2 = gr.Textbox(label="API_KEY", type="password", )
             with gr.Row():
                 source_lang = gr.Textbox(
                     label="Source Lang",
@@ -279,17 +279,16 @@ with gr.Blocks(theme="soft", css=CSS, fill_height=True) as demo:
     menuBtn.click(fn=update_menu, inputs=visible, outputs=[visible, menubar], js=JS)
     endpoint.change(fn=update_model, inputs=[endpoint], outputs=[model])
     
-    choice.select(fn=enable_sec, inputs=[choice], outputs=[endpoint2, model2, api_key2])
+    choice.select(fn=enable_sec, inputs=[choice], outputs=[AddEndpoint])
     endpoint2.change(fn=update_model, inputs=[endpoint2], outputs=[model2])
-    
+   
     start_ta = submit.click(fn=huanik, inputs=[endpoint, model, api_key, choice, endpoint2, model2, api_key2, source_lang, target_lang, source_text, country, max_tokens, context_window, num_output, rpm], outputs=[output_init, output_reflect, output_final, output_diff])
-    
     upload.upload(fn=read_doc, inputs = upload, outputs = source_text)
     output_final.change(fn=export_txt, inputs=output_final, outputs=[export])
-    
+
     submit.click(fn=closeBtnShow, outputs=[clear, close])
     output_final.change(fn=closeBtnHide, inputs=output_final, outputs=[clear, close])
     close.click(fn=None, cancels=start_ta)
-    
+
 if __name__ == "__main__":
     demo.queue(api_open=False).launch(show_api=False, share=False)