coba

app.py

@@ -1,6 +1,6 @@
 import gradio as gr
 from huggingface_hub import InferenceClient
-
+from translator import translate
 
 def respond(
     message,
@@ -65,6 +65,25 @@ with gr.Blocks() as demo:
         gr.LoginButton()
     chatbot.render()
 
+# ===== Translator =====
+def translate_ui(text, direction):
+    return translate(text, direction)
+
+# ===== Layout Gradio =====
+with gr.Blocks() as demo:
+    with gr.Sidebar():
+        gr.LoginButton()
+    
+    with gr.Tabs():
+        with gr.Tab("Chatbot"):
+            chatbot.render()
+        
+        with gr.Tab("Translator"):
+            input_text = gr.Textbox(lines=4, label="Input")
+            direction = gr.Dropdown(["ID → EN", "EN → ID"], label="Direction")
+            output_text = gr.Textbox(label="Output")
+            translate_btn = gr.Button("Translate")
+            translate_btn.click(fn=translate_ui, inputs=[input_text, direction], outputs=output_text)
 
 if __name__ == "__main__":
     demo.launch()

translator.py

@@ -0,0 +1,10 @@
+from transformers import pipeline
+
+translator_id_en = pipeline("translation", model="Helsinki-NLP/opus-mt-id-en")
+translator_en_id = pipeline("translation", model="Helsinki-NLP/opus-mt-en-id")
+
+def translate(text, direction):
+    if direction == "ID → EN":
+        return translator_id_en(text)[0]['translation_text']
+    else:
+        return translator_en_id(text)[0]['translation_text']