Update app.py

app.py

@@ -5,11 +5,27 @@ import unicodedata
 register = pd.read_csv('register.csv', index_col='jp_name')
 register.en_name = register.en_name.str.replace(',', '')
 
-def translate(jp_name):
+def translate_name(jp_name):
   jp_name = unicodedata.normalize('NFKC', jp_name)
-  if jp_name not in register.index:
-    raise gr.Error(f'Name `{jp_name}` not in register')
-  return register.loc[jp_name].en_name
+  en_name = np.nan
+  
+  if jp_name in register.index:
+    en_name = register.loc[jp_name].en_name
+  elif jp_name in register.index.map(lambda name: name.split('.')[-1]):
+    tmp_register = register.copy()
+    tmp_register.index = tmp_register.index.map(lambda name: name.split('.')[-1])
+    en_name = tmp_register.loc[jp_name].en_name
 
-demo = gr.Interface(fn=translate, inputs="text", outputs="text")
+  if isinstance(en_name, pd.DataFrame):
+    unique_names = en_name.en_name.unique()
+    en_name = unique_names[0]
+  elif not isinstance(en_name, str):
+    en_name = np.nan
+
+  return en_name
+
+def translate_names(jp_names):
+  return '\n'.join([str(translate_name(jp_name)) for jp_name in jp_names.splitlines()])
+
+demo = gr.Interface(fn=translate_names, inputs="text", outputs=gr.Text(show_copy_button=True))
 demo.launch(debug=True)