Update app.py

app.py

@@ -16,6 +16,7 @@ def load_data():
             lemmas[current_lemma] = []
         else:
             lemma_data = {
+                'word': row['#ORTO'].replace("ORTO:", "") if pd.notna(row['#ORTO']) else "",
                 'PPOS': row['#PPOS'].replace("PPOS:", "") if pd.notna(row['#PPOS']) else "",
                 'PHON1': row['#PHON1'].replace("PHON:", "") if pd.notna(row['#PHON1']) else "",
                 'PHON2': row['#PHON2'].replace("PHON:", "") if pd.notna(row['#PHON2']) else "",
@@ -28,6 +29,22 @@ def load_data():
 lemmas = load_data()
 
 def create_noun_table(lemma, forms):
+    # Initialize the table structure
+    table_data = {
+        'Nsns': '', 'Nsnst': '', 'Nsas': '', 'Nsast': '',
+        'Nsds': '', 'Nsdst': '', 'Nsgs': '', 'Nsgst': '',
+        'Npns': '', 'Npnst': '', 'Npas': '', 'Npast': '',
+        'Npds': '', 'Npdt': '', 'Npgs': '', 'Npgst': ''
+    }
+    
+    # Fill the table data based on PPOS
+    for form in forms:
+        ppos = form['PPOS']
+        word = form['word']
+        if ppos[1:5] in table_data:
+            table_data[ppos[1:5]] = word
+    
+    # Create the HTML table
     table = f"""
     <table border="1">
         <thead>
@@ -44,28 +61,28 @@ def create_noun_table(lemma, forms):
         </thead>
         <tbody>
             <tr>
-                <td>{forms.get('Nsns', '')}</td>
-                <td>{forms.get('Nsnst', '')}</td>
-                <td>{forms.get('Npns', '')}</td>
-                <td>{forms.get('Npnst', '')}</td>
+                <td>{table_data['Nsns']}</td>
+                <td>{table_data['Nsnst']}</td>
+                <td>{table_data['Npns']}</td>
+                <td>{table_data['Npnst']}</td>
             </tr>
             <tr>
-                <td>{forms.get('Nsas', '')}</td>
-                <td>{forms.get('Nsast', '')}</td>
-                <td>{forms.get('Npas', '')}</td>
-                <td>{forms.get('Npast', '')}</td>
+                <td>{table_data['Nsas']}</td>
+                <td>{table_data['Nsast']}</td>
+                <td>{table_data['Npas']}</td>
+                <td>{table_data['Npast']}</td>
             </tr>
             <tr>
-                <td>{forms.get('Nsds', '')}</td>
-                <td>{forms.get('Nsdst', '')}</td>
-                <td>{forms.get('Npds', '')}</td>
-                <td>{forms.get('Npdt', '')}</td>
+                <td>{table_data['Nsds']}</td>
+                <td>{table_data['Nsdst']}</td>
+                <td>{table_data['Npds']}</td>
+                <td>{table_data['Npdt']}</td>
             </tr>
             <tr>
-                <td>{forms.get('Nsgs', '')}</td>
-                <td>{forms.get('Nsgst', '')}</td>
-                <td>{forms.get('Npgs', '')}</td>
-                <td>{forms.get('Npgst', '')}</td>
+                <td>{table_data['Nsgs']}</td>
+                <td>{table_data['Nsgst']}</td>
+                <td>{table_data['Npgs']}</td>
+                <td>{table_data['Npgst']}</td>
             </tr>
         </tbody>
     </table>
@@ -77,14 +94,8 @@ def search_lemma(lemma):
     if not results:
         return f"No results found for {lemma}"
     
-    forms = {}
-    for result in results:
-        ppos = result['PPOS']
-        key = ppos[1:5]  # Extracting relevant part of PPOS
-        forms[key] = result['PPOS']
-    
     if 'N' in results[0]['PPOS']:
-        table = create_noun_table(lemma, forms)
+        table = create_noun_table(lemma, results)
     else:
         table = "Only noun tables are currently supported."