Update app.py

app.py

@@ -28,20 +28,73 @@ def load_data():
 
 lemmas = load_data()
 
+def create_noun_table(lemma, forms):
+    table = f"""
+    <table border="1">
+        <thead>
+            <tr>
+                <th colspan="2">Eintal</th>
+                <th colspan="2">Fleirtal</th>
+            </tr>
+            <tr>
+                <th>Óbundið</th>
+                <th>Bundið</th>
+                <th>Óbundið</th>
+                <th>Bundið</th>
+            </tr>
+        </thead>
+        <tbody>
+            <tr>
+                <td>{forms.get('Nsns', '')}</td>
+                <td>{forms.get('Nsnst', '')}</td>
+                <td>{forms.get('Npns', '')}</td>
+                <td>{forms.get('Npnst', '')}</td>
+            </tr>
+            <tr>
+                <td>{forms.get('Nsas', '')}</td>
+                <td>{forms.get('Nsast', '')}</td>
+                <td>{forms.get('Npas', '')}</td>
+                <td>{forms.get('Npast', '')}</td>
+            </tr>
+            <tr>
+                <td>{forms.get('Nsds', '')}</td>
+                <td>{forms.get('Nsdst', '')}</td>
+                <td>{forms.get('Npds', '')}</td>
+                <td>{forms.get('Npdt', '')}</td>
+            </tr>
+            <tr>
+                <td>{forms.get('Nsgs', '')}</td>
+                <td>{forms.get('Nsgst', '')}</td>
+                <td>{forms.get('Npgs', '')}</td>
+                <td>{forms.get('Npgst', '')}</td>
+            </tr>
+        </tbody>
+    </table>
+    """
+    return table
+
 def search_lemma(lemma):
     results = lemmas.get(lemma, None)
     if not results:
         return f"No results found for {lemma}"
-    response = f"Results for {lemma}:\n\n"
-    response += "PPOS\tPHON1\tPHON2\tCOMM\n"
+    
+    forms = {}
     for result in results:
-        response += f"{result['PPOS']}\t{result['PHON1']}\t{result['PHON2']}\t{result['COMM']}\n"
-    return response
+        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)
+    else:
+        table = "Only noun tables are currently supported."
+
+    return table
 
 iface = gr.Interface(
     fn=search_lemma,
     inputs="text",
-    outputs="text",
+    outputs="html",
     title="Lemma Search",
     description="Enter a lemma to search for its declensions and pronunciations."
 )