Update app.py

app.py

@@ -28,6 +28,8 @@ MODEL_INFO: Dict[str, Tuple[str, str, str]] = {
     "grc-perseus-sm": ("Ancient Greek (Perseus SM)", "grc_perseus_sm", "grecy"),
 }
 
+# --- *** THE FIX IS HERE *** ---
+# Added the 'table_headers' key to each language
 UI_TEXT = {
     "de": {
         "title": "# 🔍 Mehrsprachiger Morpho-Syntaktischer Analysator",
@@ -41,11 +43,12 @@ UI_TEXT = {
         "tab_graphic": "Grafische Darstellung",
         "tab_table": "Tabelle",
         "tab_json": "JSON",
-        "tab_ner": "Entitäten", # <-- ADDED
+        "tab_ner": "Entitäten",
         "html_label": "Abhängigkeitsparsing",
         "table_label": "Morphologische Analyse",
+        "table_headers": ["Wort", "Lemma", "POS", "Tag", "Morphologie", "Abhängigkeit"], # <-- WAS MISSING
         "json_label": "JSON-Ausgabe",
-        "ner_label": "Benannte Entitäten", # <-- ADDED
+        "ner_label": "Benannte Entitäten",
         "error_message": "Fehler: "
     },
     "en": {
@@ -60,11 +63,12 @@ UI_TEXT = {
         "tab_graphic": "Graphic View",
         "tab_table": "Table",
         "tab_json": "JSON",
-        "tab_ner": "Entities", # <-- ADDED
+        "tab_ner": "Entities",
         "html_label": "Dependency Parsing",
         "table_label": "Morphological Analysis",
+        "table_headers": ["Word", "Lemma", "POS", "Tag", "Morphology", "Dependency"], # <-- WAS MISSING
         "json_label": "JSON Output",
-        "ner_label": "Named Entities", # <-- ADDED
+        "ner_label": "Named Entities",
         "error_message": "Error: "
     },
     "es": {
@@ -79,14 +83,16 @@ UI_TEXT = {
         "tab_graphic": "Vista Gráfica",
         "tab_table": "Tabla",
         "tab_json": "JSON",
-        "tab_ner": "Entidades", # <-- ADDED
+        "tab_ner": "Entidades",
         "html_label": "Análisis de Dependencias",
         "table_label": "Análisis Morfológico",
+        "table_headers": ["Palabra", "Lema", "POS", "Etiqueta", "Morfología", "Dependencia"], # <-- WAS MISSING
         "json_label": "Salida JSON",
-        "ner_label": "Entidades Nombradas", # <-- ADDED
+        "ner_label": "Entidades Nombradas",
         "error_message": "Error: "
     }
 }
+# --- *** END FIX *** ---
 
 MODELS: Dict[str, Optional[spacy.Language]] = {}
 
@@ -124,9 +130,9 @@ def install_grecy_model_from_github(model_name: str) -> bool:
 
     if model_name == "grc_proiel_trf":
         wheel_filename = "grc_proiel_trf-3.7.5-py3-none-any.whl"
-    # --- UPDATED: Added grc_ner_trf to this list ---
     elif model_name in ["grc_perseus_trf", "grc_proiel_lg", "grc_perseus_lg", 
                         "grc_proiel_sm", "grc_perseus_sm", "grc_ner_trf"]:
+        # Note: Wheel name uses underscore (grc_ner_trf), not hyphen
         wheel_filename = f"{model_name}-0.0.0-py3-none-any.whl"
     else:
         print(f"✗ Unknown grecy model: {model_name}")
@@ -329,11 +335,11 @@ def update_ui(ui_lang: str):
         gr.Tab(label=ui_config["tab_graphic"]),
         gr.Tab(label=ui_config["tab_table"]),
         gr.Tab(label=ui_config["tab_json"]),
-        gr.Tab(label=ui_config["tab_ner"]), # <-- ADDED
+        gr.Tab(label=ui_config["tab_ner"]),
         gr.HTML(label=ui_config["html_label"]),
         gr.DataFrame(label=ui_config["table_label"], headers=ui_config["table_headers"], interactive=False),
         gr.JSON(label=ui_config["json_label"]),
-        gr.HTML(label=ui_config["ner_label"]) # <-- ADDED
+        gr.HTML(label=ui_config["ner_label"])
     ]
 
 def create_interface():
@@ -358,22 +364,19 @@ def create_interface():
         with gr.Tabs():
             with gr.Tab(config["tab_graphic"]) as tab_graphic:
                 html_dep_out = gr.HTML(label=config["html_label"])
-            # --- ADDED NEW TAB ---
             with gr.Tab(config["tab_ner"]) as tab_ner:
                 html_ner_out = gr.HTML(label=config["ner_label"])
-            # --- END ADDITION ---
             with gr.Tab(config["tab_table"]) as tab_table:
+                # This is the line that was crashing
                 df_out = gr.DataFrame(label=config["table_label"], headers=config["table_headers"], interactive=False)
             with gr.Tab(config["tab_json"]) as tab_json:
                 json_out = gr.JSON(label=config["json_label"])
         
-        # --- UPDATED OUTPUTS ---
         analyze_button.click(fn=get_analysis, 
                              inputs=[ui_lang_radio, model_lang_radio, text_input],
                              outputs=[df_out, json_out, html_dep_out, html_ner_out, analyze_button], 
                              api_name="get_morphology")
         
-        # --- UPDATED OUTPUTS ---
         ui_lang_radio.change(fn=update_ui, 
                              inputs=ui_lang_radio,
                              outputs=[markdown_title, markdown_subtitle, ui_lang_radio, model_lang_radio,