Update from_lang logic

POST

@@ -0,0 +1,12 @@
+HTTP/1.1 405 Method Not Allowed
+Date: Sat, 06 Aug 2022 01:37:43 GMT
+Content-Type: application/json
+Content-Length: 31
+Connection: keep-alive
+Server: nginx
+X-Powered-By: huggingface-moon
+x-request-id: 23fbb637-9bd3-4eeb-9743-a4f6e73c1802
+allow: POST
+Link: <https://huggingface.co/spaces/mikeee/opus-mt>; rel="canonical"
+Strict-Transport-Security: max-age=31536000; includeSubDomains
+

app.py

@@ -1,6 +1,6 @@
 """Prep gradio API."""
 # pylint: diable=invalid-name
-from typing import List, Union
+from typing import List, Optional, Union
 
 import gradio as gr
 from easynmt import EasyNMT
@@ -12,12 +12,25 @@ translate = EasyNMT("opus-mt").translate
 
 def opusmt(
     text: str,
-    # from_lang: Optional[str] = None,
+    from_lang: Optional[str] = None,
     to_lang: str = "zh",
 ) -> Union[str, List[str]]:
     """Translate via easyntm-opus-mt."""
+    if to_lang in ["auto"]:
+        if from_lang in ["zh"]:
+            to_lang = "en"
+        else:
+            to_lang = "zh"
+
+    if from_lang in ["auto"]:
+        from_lang = None
+
     try:
-        res = translate(text, target_lang=to_lang)
+        res = translate(
+            text,
+            target_lang=to_lang,
+            source_lang=from_lang,
+        )
     except Exception as e:
         logger.error(e)
         res = f"errors: {e}"
@@ -31,8 +44,8 @@ inputs = [
         lines=7,
         max_lines=200,
     ),
-    # "textarea",
-    gr.Textbox(label="to_lang", value="zh"),
+    gr.Textbox(label="from-lang", value="auto"),
+    gr.Textbox(label="to-lang", value="zh"),
 ]
 outputs = [
     gr.Textbox(
@@ -52,8 +65,8 @@ iface = gr.Interface(
     title="opus mt",
     description=description,
     examples=[
-        ["This is a test.", "zh"],
-        ["This is a test.", "de"],
+        ["This is a test.", "en", "zh"],
+        ["This is a test.", "en", "de"],
     ],
     allow_flagging="never",
     cache_examples=False,