Upload app.py

app.py

@@ -1,4 +1,4 @@
-import os, re, string, json
+import os, re, string, json, tempfile, uuid
 import html
 import inspect
 import importlib.resources as importlib_resources
@@ -96,12 +96,12 @@ CSS = """
   color:#0b1b19 !important;
 }
 
-#lang_fo_off, #lang_en_off{
+#lang_fo_off, #lang_en_off, #btn_dl_main, #btn_dl_exp{
   background:#C6DAD6 !important;
   border-color:#6F9992 !important;
   color:#0b1b19 !important;
 }
-#lang_fo_off:hover, #lang_en_off:hover{
+#lang_fo_off:hover, #lang_en_off:hover, #btn_dl_main:hover, #btn_dl_exp:hover{
   background:#89AFA9 !important;
   border-color:#6F9992 !important;
   color:#0b1b19 !important;
@@ -109,12 +109,12 @@ CSS = """
 
 /* Dark mode: make the INACTIVE buttons match what you had before (darker, readable) */
 @media (prefers-color-scheme: dark){
-  #lang_fo_off, #lang_en_off{
+  #lang_fo_off, #lang_en_off, #btn_dl_main, #btn_dl_exp{
     background:#2a3b38 !important;
     border-color:#6F9992 !important;
     color:#e7eceb !important;
   }
-  #lang_fo_off:hover, #lang_en_off:hover{
+  #lang_fo_off:hover, #lang_en_off:hover, #btn_dl_main:hover, #btn_dl_exp:hover{
     background:#89AFA9 !important;
     border-color:#6F9992 !important;
     color:#0b1b19 !important;
@@ -140,6 +140,26 @@ CSS = """
   min-width:120px !important;
   flex:0 0 auto !important;
 }
+
+/* Expanded-tags header: keep download button hard-right */
+#expanded_hdr{
+  display:flex !important;
+  align-items:center !important;
+  gap:12px !important;
+}
+#expanded_hdr > .gr-markdown{ flex:1 1 auto !important; }
+#expanded_buttons{
+  display:flex !important;
+  gap:10px !important;
+  justify-content:flex-end !important;
+  align-items:center !important;
+  flex-wrap:nowrap !important;
+}
+#expanded_buttons .gr-button, #expanded_buttons button{
+  width:auto !important;
+  min-width:120px !important;
+  flex:0 0 auto !important;
+}
 /* Remove the big Gradio panel/frame around the textbox (keep textarea normal) */
 #input_col,
 #input_col > div,
@@ -640,6 +660,35 @@ def render(rows_state, lang: str):
 
     return (_make_html_table(cols_main, out_main), _make_html_table(cols_mean, out_mean), build_overview(lang))
 
+
+def _write_tsv(df: pd.DataFrame, filename: str) -> str:
+    """Write a TSV file to a unique temp folder and return the absolute path."""
+    tmpdir = os.path.join(tempfile.gettempdir(), "marka_downloads", str(uuid.uuid4()))
+    os.makedirs(tmpdir, exist_ok=True)
+    path = os.path.join(tmpdir, filename)
+    df.to_csv(path, sep="\t", index=False, encoding="utf-8")
+    return path
+
+def build_download_main(rows_state) -> str:
+    """TSV with 2 columns: Faroese analysis, then English analysis."""
+    fo_vals, en_vals = [], []
+    for r in (rows_state or []):
+        vec = torch.tensor(r["vec"])
+        fo_vals.append(analysis_text(vec, "fo"))
+        en_vals.append(analysis_text(vec, "en"))
+    df = pd.DataFrame({UI["fo"]["s"]: fo_vals, UI["en"]["s"]: en_vals})
+    return _write_tsv(df, "Markað.tsv")
+
+def build_download_expanded(rows_state, lang: str) -> str:
+    """TSV with the Expanded tags column in the currently selected UI language."""
+    lang = "fo" if lang == "fo" else "en"
+    vals = []
+    for r in (rows_state or []):
+        vec = torch.tensor(r["vec"])
+        vals.append(expanded_text(vec, lang))
+    df = pd.DataFrame({UI[lang]["m"]: vals})
+    return _write_tsv(df, "Markað_útgreinað.tsv")
+
 with gr.Blocks(css=CSS, title="Marka") as demo:
     with gr.Row(equal_height=False):
         with gr.Column(scale=2, elem_id="input_col"):
@@ -663,10 +712,15 @@ with gr.Blocks(css=CSS, title="Marka") as demo:
             btn_lang_fo_off = gr.Button("Føroyskt", variant="secondary", elem_id="lang_fo_off", visible=False)
             btn_lang_en_on  = gr.Button("English",  variant="primary",   elem_id="lang_en_on",  visible=False)
             btn_lang_en_off = gr.Button("English",  variant="secondary", elem_id="lang_en_off", visible=False)
+            btn_dl_main    = gr.DownloadButton("Tak niður / Download", variant="secondary", elem_id="btn_dl_main", visible=False)
     out_df = gr.HTML(value="", elem_id="out_df", visible=False)
 
     expanded_acc = gr.Accordion("Útgreinað marking / Expanded tags", open=False, visible=False)
     with expanded_acc:
+        with gr.Row(elem_id="expanded_hdr"):
+            gr.Markdown(" ")
+            with gr.Row(elem_id="expanded_buttons"):
+                btn_dl_exp = gr.DownloadButton("Tak niður / Download", variant="secondary", elem_id="btn_dl_exp", visible=False)
         out_mean_df = gr.HTML(value="", elem_id="out_mean_df")
 
     overview_acc = gr.Accordion("Markayvirlit / Tag Overview", open=False, visible=True)
@@ -680,6 +734,10 @@ with gr.Blocks(css=CSS, title="Marka") as demo:
         show_fo = (lang_current == "fo")
         show_en = (lang_current == "en")
 
+        have_rows = bool(rows)
+        dl_main_path = build_download_main(rows) if have_rows else None
+        dl_exp_path = build_download_expanded(rows, lang_current) if have_rows else None
+
         return (
             rows,
             gr.update(value=df_main, visible=True),
@@ -690,6 +748,8 @@ with gr.Blocks(css=CSS, title="Marka") as demo:
             gr.update(visible=not show_fo),
             gr.update(visible=show_en),
             gr.update(visible=not show_en),
+            gr.update(value=dl_main_path, visible=have_rows),
+            gr.update(value=dl_exp_path, visible=have_rows),
             lang_current,
         )
 
@@ -699,6 +759,10 @@ with gr.Blocks(css=CSS, title="Marka") as demo:
         show_fo = (lang_value == "fo")
         show_en = (lang_value == "en")
 
+        have_rows = bool(rows)
+        dl_main_path = build_download_main(rows) if have_rows else None
+        dl_exp_path = build_download_expanded(rows, lang_value) if have_rows else None
+
         return (
             lang_value,
             gr.update(value=df_main),
@@ -708,8 +772,9 @@ with gr.Blocks(css=CSS, title="Marka") as demo:
             gr.update(visible=not show_fo),
             gr.update(visible=show_en),
             gr.update(visible=not show_en),
+            gr.update(value=dl_main_path, visible=have_rows),
+            gr.update(value=dl_exp_path, visible=have_rows),
         )
-
     def on_set_fo(rows):
         return on_set_lang(rows, "fo")
 
@@ -720,7 +785,8 @@ with gr.Blocks(css=CSS, title="Marka") as demo:
         on_tag,
         inputs=[inp, lang_state],
         outputs=[state, out_df, out_mean_df, overview_md, expanded_acc,
-                 btn_lang_fo_on, btn_lang_fo_off, btn_lang_en_on, btn_lang_en_off, lang_state],
+                 btn_lang_fo_on, btn_lang_fo_off, btn_lang_en_on, btn_lang_en_off,
+                 btn_dl_main, btn_dl_exp, lang_state],
         queue=False,
     )
 
@@ -728,28 +794,32 @@ with gr.Blocks(css=CSS, title="Marka") as demo:
         on_set_fo,
         inputs=[state],
         outputs=[lang_state, out_df, out_mean_df, overview_md,
-                 btn_lang_fo_on, btn_lang_fo_off, btn_lang_en_on, btn_lang_en_off],
+                 btn_lang_fo_on, btn_lang_fo_off, btn_lang_en_on, btn_lang_en_off,
+                 btn_dl_main, btn_dl_exp],
         queue=False,
     )
     btn_lang_fo_off.click(
         on_set_fo,
         inputs=[state],
         outputs=[lang_state, out_df, out_mean_df, overview_md,
-                 btn_lang_fo_on, btn_lang_fo_off, btn_lang_en_on, btn_lang_en_off],
+                 btn_lang_fo_on, btn_lang_fo_off, btn_lang_en_on, btn_lang_en_off,
+                 btn_dl_main, btn_dl_exp],
         queue=False,
     )
     btn_lang_en_on.click(
         on_set_en,
         inputs=[state],
         outputs=[lang_state, out_df, out_mean_df, overview_md,
-                 btn_lang_fo_on, btn_lang_fo_off, btn_lang_en_on, btn_lang_en_off],
+                 btn_lang_fo_on, btn_lang_fo_off, btn_lang_en_on, btn_lang_en_off,
+                 btn_dl_main, btn_dl_exp],
         queue=False,
     )
     btn_lang_en_off.click(
         on_set_en,
         inputs=[state],
         outputs=[lang_state, out_df, out_mean_df, overview_md,
-                 btn_lang_fo_on, btn_lang_fo_off, btn_lang_en_on, btn_lang_en_off],
+                 btn_lang_fo_on, btn_lang_fo_off, btn_lang_en_on, btn_lang_en_off,
+                 btn_dl_main, btn_dl_exp],
         queue=False,
     )