Update app.py

app.py

@@ -1,7 +1,50 @@
 import gradio as gr
+import pandas as pd
+from pyvi import ViTokenizer, ViPosTagger
 
-def greet(name):
-    return "Hello " + name + "!!"
+pos_mapping = {
+    "A": ("Adjective", "Tính từ"),
+    "C": ("Coordinating conjunction", "Liên từ kết hợp"),
+    "E": ("Preposition", "Giới từ"),
+    "I": ("Interjection", "Thán từ"),
+    "L": ("Determiner", "Hạn định từ"),
+    "M": ("Numeral", "Số từ"),
+    "N": ("Common noun", "Danh từ chung"),
+    "Nc": ("Noun Classifier", "Lượng từ"),
+    "Ny": ("Noun abbreviation", "Từ viết tắt danh từ"),
+    "Np": ("Proper noun", "Danh từ riêng"),
+    "Nu": ("Unit noun", "Danh từ đơn vị"),
+    "P": ("Pronoun", "Đại từ"),
+    "R": ("Adverb", "Trạng từ"),
+    "S": ("Subordinating conjunction", "Liên từ phụ thuộc"),
+    "T": ("Auxiliary, modal words", "Trợ từ, động từ tình thái"),
+    "V": ("Verb", "Động từ"),
+    "X": ("Unknown", "Không xác định"),
+    "F": ("Filtered out (punctuation)", "Bị loại bỏ (dấu câu)")
+}
 
-demo = gr.Interface(fn=greet, inputs="text", outputs="text")
+def change(sentence):
+    resutlt = ViPosTagger.postagging(ViTokenizer.tokenize(sentence))
+    print(resutlt)
+    words, pos_tags = resutlt
+    pos_tags_engs = []
+    pos_tags_vies = []
+    for i in pos_tags:
+        print(pos_mapping[i][0])
+        print(pos_mapping[i][1])
+        pos_tags_engs.append(pos_mapping[i][0])
+        pos_tags_vies.append(pos_mapping[i][1])
+    # final_result = ''''''
+    # for word, pos_tag, pos_tags_eng, pos_tags_vie in zip(words, pos_tags, pos_tags_engs, pos_tags_vies):
+    #     final_result += f"Từ: {word}, Loại từ: {pos_tag} ({pos_tags_eng} - {pos_tags_vie})\n"
+    # return final_result
+
+    header = f"{'Từ':<15} {'Loại từ':<10} {'English POS':<20} {'Tiếng Việt POS'}\n"
+    header += "-" * 60 + "\n"
+    rows = [f"{word:<15} {pos_tag:<10} {pos_tags_eng:<20} {pos_tags_vie}"
+            for word, pos_tag, pos_tags_eng, pos_tags_vie in zip(words, pos_tags, pos_tags_engs, pos_tags_vies)]
+    final_result = header + "\n".join(rows)
+    return final_result
+
+demo = gr.Interface(fn=change, inputs="text", outputs="text")
 demo.launch()