Spaces:

ngocdang83
/

HachimiMT-demo

Running

App Files Files Community

ngocdang83 commited on Jul 25

Commit

53304ea

verified ·

1 Parent(s): c5201a6

Fix glossary row addition on Space

Browse files

Files changed (2) hide show

hachimimt-local.zip +2 -2
src/app.py +70 -5

hachimimt-local.zip CHANGED Viewed

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:14595200248513ec2a2fa9fc503b21252e077aacb4162cb4a270c173fc9a00d4
-size 136004

 version https://git-lfs.github.com/spec/v1
+oid sha256:a3511d1f3a5a8e0e44de5cbd24daedc7e9226c3bd9bf0efb0d7688972b48a0d9
+size 136689

src/app.py CHANGED Viewed

@@ -200,10 +200,32 @@ def render_gpu_hint_html() -> str:
 # theme vì Gradio bỏ qua nó ở đây và CSS đã đủ.
 # Preload font trong <head> để giảm nhấp nháy khi tải.
-HEAD_HTML = """
-<link rel="preconnect" href="https://fonts.googleapis.com">
-<link rel="preconnect" href="https://fonts.gstatic.com" crossorigin>
-"""
 CUSTOM_CSS = """
 @import url('https://fonts.googleapis.com/css2?family=Fraunces:ital,opsz,wght@0,9..144,400;0,9..144,600;0,9..144,700;1,9..144,500&family=Literata:ital,opsz,wght@0,7..72,400;0,7..72,500;0,7..72,600;1,7..72,400&family=Noto+Serif+SC:wght@400;600&display=swap');
@@ -1328,6 +1350,28 @@ def export_translation_ui(
     return export_translation(full_text, "hachimimt", request)
 def import_glossary_ui(file_obj) -> tuple[list[list[object]], str]:
     if file_obj is None:
         raise gr.Error("Chọn file glossary .tsv hoặc .json.")
@@ -2129,6 +2173,15 @@ def build_ui() -> gr.Blocks:
                 label="Glossary của phiên hiện tại",
                 elem_id="glossary-table",
             )
             with gr.Row():
                 glossary_file = gr.File(
                     label="Nhập glossary (.tsv / .json)",
@@ -2219,8 +2272,20 @@ mục `hachimimt`, chạy `setup.bat` (cài thư viện + tải model mặc đ
             on_backend_change,
             inputs=[backend_select, model_select],
             outputs=[engine_hint, model_badge],
-        )
         auto_batch.change(on_auto_batch_toggle, inputs=[auto_batch], outputs=[manual_batch])
         glossary_import_btn.click(
             import_glossary_ui,
             inputs=[glossary_file],

 # theme vì Gradio bỏ qua nó ở đây và CSS đã đủ.
 # Preload font trong <head> để giảm nhấp nháy khi tải.
+HEAD_HTML = """
+<link rel="preconnect" href="https://fonts.googleapis.com">
+<link rel="preconnect" href="https://fonts.gstatic.com" crossorigin>
+<script>
+window.__hachimimtRefreshGlossaryTable = () => {
+    window.dispatchEvent(new Event("resize"));
+    document
+        .querySelector("#glossary-table .virtual-table-viewport")
+        ?.dispatchEvent(new Event("scroll"));
+};
+document.addEventListener("click", (event) => {
+    const target = event.target instanceof Element
+        ? event.target.closest(
+            '[role="menuitem"][aria-label="Add row above"], ' +
+            '[role="menuitem"][aria-label="Add row below"]'
+        )
+        : null;
+    if (!target) return;
+    requestAnimationFrame(() => {
+        window.__hachimimtRefreshGlossaryTable();
+        requestAnimationFrame(window.__hachimimtRefreshGlossaryTable);
+    });
+});
+</script>
+"""
 CUSTOM_CSS = """
 @import url('https://fonts.googleapis.com/css2?family=Fraunces:ital,opsz,wght@0,9..144,400;0,9..144,600;0,9..144,700;1,9..144,500&family=Literata:ital,opsz,wght@0,7..72,400;0,7..72,500;0,7..72,600;1,7..72,400&family=Noto+Serif+SC:wght@400;600&display=swap');
     return export_translation(full_text, "hachimimt", request)
+def add_glossary_row_ui(rows) -> list[list[object]]:
+    """Append one editable glossary row without relying on Gradio's cell menu."""
+    if isinstance(rows, dict):
+        rows = rows.get("data", [])
+    elif hasattr(rows, "tolist"):
+        rows = rows.tolist()
+    blank_row: list[object] = ["", "", "", "", True]
+    table_rows: list[list[object]] = []
+    if isinstance(rows, (list, tuple)):
+        for row in rows:
+            if not isinstance(row, (list, tuple)):
+                continue
+            normalized = blank_row.copy()
+            for index, value in enumerate(row[: len(blank_row)]):
+                normalized[index] = value
+            table_rows.append(normalized)
+    table_rows.append(blank_row.copy())
+    return table_rows
 def import_glossary_ui(file_obj) -> tuple[list[list[object]], str]:
     if file_obj is None:
         raise gr.Error("Chọn file glossary .tsv hoặc .json.")
                 label="Glossary của phiên hiện tại",
                 elem_id="glossary-table",
             )
+            with gr.Row():
+                glossary_add_row_btn = gr.Button(
+                    "＋ Thêm hàng",
+                    variant="secondary",
+                    size="sm",
+                    scale=0,
+                    min_width=140,
+                    elem_id="glossary-add-row",
+                )
             with gr.Row():
                 glossary_file = gr.File(
                     label="Nhập glossary (.tsv / .json)",
             on_backend_change,
             inputs=[backend_select, model_select],
             outputs=[engine_hint, model_badge],
+        )
         auto_batch.change(on_auto_batch_toggle, inputs=[auto_batch], outputs=[manual_batch])
+        glossary_add_row_event = glossary_add_row_btn.click(
+            add_glossary_row_ui,
+            inputs=[glossary_table],
+            outputs=[glossary_table],
+            queue=False,
+        )
+        glossary_add_row_event.then(
+            fn=None,
+            js="() => window.__hachimimtRefreshGlossaryTable?.()",
+            queue=False,
+            api_name=False,
+        )
         glossary_import_btn.click(
             import_glossary_ui,
             inputs=[glossary_file],