feat: integrate AMRToText and reactivate TextToAMR

app.py

@@ -16,6 +16,7 @@ sys.path.append("./amr-tst-indo")
 from text_to_amr import TextToAMR
 from style_detector import StyleDetector
 from style_rewriting import StyleRewriting
+from amr_to_text import AMRToTextWithTaufiqMethod
 
 amr_parsing_model_name = "mbart-en-id-smaller-indo-amr-parsing-translated-nafkhan"
 snapshot_download(
@@ -56,6 +57,20 @@ sr = StyleRewriting(
     maximize_style_words_expansion=False
 )
 
+amr_gen_model_name = "taufiq-indo-amr-generation-gold-uncased"
+model_path = f"./{amr_gen_model_name}"
+snapshot_download(
+    repo_id=f"abdiharyadi/{amr_gen_model_name}",
+    local_dir=model_path,
+    allow_patterns=[
+        "*checkpoint-3*"
+    ]
+)
+a2t = AMRToTextWithTaufiqMethod(
+    model_path=os.path.join(model_path, "checkpoint-3"),
+    lowercase=True,
+)
+
 def run(text, source_style):
     yield (
         "(Memproses ...)",
@@ -67,15 +82,7 @@ def run(text, source_style):
 
     start_time = time.time()
 
-    # source_amr, *_ = t2a([text])
-    source_amr = penman.decode("""
-    (z0 / dan
-        :op1 (z1 / bagus-01
-            :ARG1 (z2 / tempat)
-            :degree (z3 / sangat))
-        :op2 (z4 / bersih-01
-            :ARG1 z2))
-    """)
+    source_amr, *_ = t2a([text])
     source_amr.metadata = {}
     source_amr_display = penman.encode(source_amr)
     source_amr_display += f"\n\n({time.time() - start_time:.2f} s)"
@@ -88,10 +95,6 @@ def run(text, source_style):
     )
 
     triplets = sd.get_triplets(text)
-    # triplets = [
-    #     ("kamar", "sangat bagus", "positif"),
-    #     ("kamar", "bersih", "positif")
-    # ]
     triplets_display = "\n".join(f"({x[0]}, {x[1]}, {x[2]})" for x in triplets)
     triplets_display += f"\n\n({time.time() - start_time:.2f} s)"
     yield (
@@ -103,7 +106,6 @@ def run(text, source_style):
     )
 
     style_words = sd.get_style_words_from_triplets(triplets)
-    # style_words = ["bagus", "bersih"]
     style_words_display = ", ".join(style_words)
     style_words_display += f"\n\n({time.time() - start_time:.2f} s)"
     yield (
@@ -125,7 +127,7 @@ def run(text, source_style):
         "(Memproses ...)",
     )
 
-    result = f"dunia ({text=}, {source_style=})"
+    result, *_ = a2t([target_amr])
     result += f"\n\n({time.time() - start_time:.2f} s)"
     yield (
         source_amr_display,