Update modules/translator.py

modules/translator.py

@@ -35,12 +35,22 @@ def text_translator(input_text : str, language : str) -> str:
     return output_text
 
 # Interface
-with gr.Blocks() as demo:
-    gr.HTML("<h1 align = 'center'> Text Translator </h1>")
-    gr.HTML("<h4 align = 'center'> Translate to any language </h4>")
-    
-    inputs = [gr.Textbox(label = "Enter the text that you want to translate"), gr.Textbox(label = "Enter the language that you want it to translate to", placeholder = "Example : Hindi,French,Bengali,etc")]
-    generate_btn = gr.Button(value = 'Generate')
-    outputs  = [gr.Textbox(label = "Translated text")]
-    generate_btn.click(fn = text_translator, inputs= inputs, outputs = outputs)
-    
+def translator_interface():
+    with gr.Blocks() as demo:
+        gr.HTML("<h1 align = 'center'> Text Translator </h1>")
+        gr.HTML("<h4 align = 'center'> Translate to any language </h4>")
+        
+        with gr.Row():
+            input_text = gr.Textbox(label="Enter the text that you want to translate")
+            language = gr.Textbox(label="Enter the language that you want it to translate to", placeholder="Example: Hindi, French, Bengali, etc.")
+        
+        generate_btn = gr.Button(value='Generate')
+        output_text = gr.Textbox(label="Translated text")
+        
+        generate_btn.click(fn=text_translator, inputs=[input_text, language], outputs=output_text)
+    
+    return demo
+
+# สำหรับแสดง Interface ใน app.py
+def get_translator_tab():
+    return translator_interface()