Update app.py

app.py

@@ -112,6 +112,9 @@ function custom(){
 
 def run(cmd):
     return str(subprocess.run(cmd, shell=True, capture_output=True, env=None).stdout)
+
+def xpath(str,pattern):
+    return fromstring(str).xpath(pattern)[0].text_content().lower().strip()
     
 @gpu(string(string,string),device=True,inline=True)
 def translate(text,lang):
@@ -142,10 +145,10 @@ def translate(text,lang):
     ).content)
     translated = text
     try:
-        src_lang = fromstring(content).xpath('//*[@class="source-language"]')[0].text_content().lower().strip()
-        trgt_lang = fromstring(content).xpath('//*[@class="target-language"]')[0].text_content().lower().strip()
-        src_text = fromstring(content).xpath('//*[@id="tw-source-text"]/*')[0].text_content().lower().strip()
-        trgt_text = fromstring(content).xpath('//*[@id="tw-target-text"]/*')[0].text_content().lower().strip()
+        src_lang = xpath(content,'//*[@class="source-language"]')
+        trgt_lang = xpath(content,'//*[@class="target-language"]')
+        src_text = xpath(content,'//*[@id="tw-source-text"]/*')
+        trgt_text = xpath(content,'//*[@id="tw-target-text"]/*')
         if trgt_lang == lang:
             translated = trgt_text
     except: