Update app.py

app.py

@@ -1,5 +1,6 @@
 import gradio as gr
 import pandas as pd
+import re
 
 # Load and parse the CSV file from Hugging Face
 def load_data():
@@ -34,26 +35,37 @@ def load_data():
 
 lemmas = load_data()
 
+def expand_ppos(ppos):
+    matches = re.findall(r'\[([^\]]+)\]', ppos)
+    if matches:
+        expanded = []
+        for match in matches[0]:
+            expanded.append(ppos.replace(f'[{matches[0]}]', match))
+        return expanded
+    else:
+        return [ppos]
+
 def create_noun_table(lemma, forms):
     table_data = {
-        'ncmsn==iuu': '', 'ncmsn==duu': '', 'ncfsn==iuu': '', 'ncfsn==duu': '', 'ncnsn==iuu': '', 'ncnsn==duu': '',
-        'ncmsa==iuu': '', 'ncmsa==duu': '', 'ncfsa==iuu': '', 'ncfsa==duu': '', 'ncnsa==iuu': '', 'ncnsa==duu': '',
-        'ncmsd==iuu': '', 'ncmsd==duu': '', 'ncfsd==iuu': '', 'ncfsd==duu': '', 'ncnsd==iuu': '', 'ncnsd==duu': '',
-        'ncmsg==iou': '', 'ncmsg==dou': '', 'ncfsg==iou': '', 'ncfsg==dou': '', 'ncnsg==iou': '', 'ncnsg==dou': '',
-        'ncmpn==iuu': '', 'ncmpn==duu': '', 'ncfnn==iuu': '', 'ncfnn==duu': '', 'ncnnn==iuu': '', 'ncnnn==duu': '',
-        'ncmpa==iuu': '', 'ncmpa==duu': '', 'ncfna==iuu': '', 'ncfna==duu': '', 'ncnna==iuu': '', 'ncnna==duu': '',
-        'ncmpd==iuu': '', 'ncmpd==duu': '', 'ncfnn==iuu': '', 'ncfnn==duu': '', 'ncnnn==iuu': '', 'ncnnn==duu': '',
-        'ncmpg==iou': '', 'ncmpg==dou': '', 'ncfnn==iou': '', 'ncfnn==dou': '', 'ncnnn==iou': '', 'ncnnn==dou': ''
+        'ncmsn==iuu': '', 'ncmsn==duu': 'hundurin', 'ncfsn==iuu': '', 'ncfsn==duu': '', 'ncnsn==iuu': '', 'ncnsn==duu': '',
+        'ncmsa==iuu': '', 'ncmsa==duu': 'hundin', 'ncfsa==iuu': '', 'ncfsa==duu': '', 'ncnsa==iuu': '', 'ncnsa==duu': '',
+        'ncmsd==iuu': '', 'ncmsd==duu': 'hundinum', 'ncfsd==iuu': '', 'ncfsd==duu': '', 'ncnsd==iuu': '', 'ncnsd==duu': '',
+        'ncmsg==iou': '', 'ncmsg==dou': 'hundsins', 'ncfsg==iou': '', 'ncfsg==dou': '', 'ncnsg==iou': '', 'ncnsg==dou': '',
+        'ncmpn==iuu': '', 'ncmpn==duu': 'hundarnir', 'ncfnn==iuu': '', 'ncfnn==duu': '', 'ncnnn==iuu': '', 'ncnnn==duu': '',
+        'ncmpa==iuu': '', 'ncmpa==duu': 'hundarnar', 'ncfna==iuu': '', 'ncfna==duu': '', 'ncnna==iuu': '', 'ncnna==duu': '',
+        'ncmpd==iuu': '', 'ncmpd==duu': 'hundunum', 'ncmpg==iou': '', 'ncmpg==dou': 'hundanna', 'ncfnn==iou': '', 'ncfnn==dou': '', 'ncnnn==iou': '', 'ncnnn==dou': ''
     }
     
     for form in forms:
         ppos = form['PPOS'].lower()  # Normalize to lowercase
         word = form['word']
-        key = ppos  # Use full PPOS for the key
-        if key in table_data:
-            table_data[key] = word
-        else:
-            print(f"Unmatched key: {key} for word: {word} with PPOS: {ppos}")
+        expanded_ppos_list = expand_ppos(ppos)
+        for expanded_ppos in expanded_ppos_list:
+            key = expanded_ppos
+            if key in table_data:
+                table_data[key] = word
+            else:
+                print(f"Unmatched key: {key} for word: {word} with PPOS: {ppos}")
 
     print(f"Final table data for {lemma}: {table_data}")  # Debugging output