Update app.py

app.py

@@ -1,82 +1,11 @@
 import gradio as gr
-import phonetisaurus
-from collections import namedtuple    
-import re
-import shlex
-import tempfile
-import subprocess
-import khmercut
+import khmerphonemizer
 
-def phonemizer(text: str):
-    
-    if not text:
-        return ""
-    
-    
-    text = text.lower()
-    env = phonetisaurus.guess_environment()
-    model_path = "./model.fst"
-
-    with tempfile.NamedTemporaryFile(suffix=".txt", mode="w+") as temp_file:
-        print(text, file=temp_file)
-        
-        temp_file.seek(0)
-
-        phonetisaurus_cmd = [
-            "phonetisaurus-apply",
-            "--model",
-            shlex.quote(str(model_path)),
-            "--word_list",
-            shlex.quote(str(temp_file.name)),
-            "--nbest",
-            str(1),
-        ]
-
-        result_str = subprocess.check_output(phonetisaurus_cmd, env=env, universal_newlines=True)
-        phoneme = result_str.split('\t')[1].strip()
-        #phoneme = re.sub(r'\s', "", phoneme)
-        return phoneme
-
-
-def phonemizer_ipa(text: str):
-    
-    if not text:
-        return ""
-    
-    
-    text = text.lower()
-    env = phonetisaurus.guess_environment()
-    model_path = "./model_ipa.fst"
-
-    with tempfile.NamedTemporaryFile(suffix=".txt", mode="w+") as temp_file:
-        print(text, file=temp_file)
-        
-        temp_file.seek(0)
-
-        phonetisaurus_cmd = [
-            "phonetisaurus-apply",
-            "--model",
-            shlex.quote(str(model_path)),
-            "--word_list",
-            shlex.quote(str(temp_file.name)),
-            "--nbest",
-            str(1),
-        ]
-
-        result_str = subprocess.check_output(phonetisaurus_cmd, env=env, universal_newlines=True)
-        phoneme = result_str.split('\t')[1].strip()
-        #phoneme = re.sub(r'\s', "", phoneme)
-        return phoneme
-
-def processor(text: str):
-    tokens = khmercut.tokenize(text)
-    return "|".join(map(phonemizer, tokens)), "|".join(map(phonemizer_ipa, tokens))
-    
 iface = gr.Interface(
     title="Khmer Phonemizer",
     description="Convert Khmer & English into Phonetic Symbols",
-    fn=processor, inputs="text", 
-    outputs=["text", "text"],
+    fn=khmerphonemizer.phonemize, inputs="text", 
+    outputs=["text"],
     allow_flagging="never",
 )