Update app.py

app.py

@@ -1,20 +1,21 @@
 from __future__ import annotations
 
 import json
-import shutil
-import subprocess
 import tempfile
-from datetime import datetime, timedelta
-from functools import lru_cache
 from pathlib import Path
-from huggingface_hub import hf_hub_download
 
 import gradio as gr
+from huggingface_hub import hf_hub_download
 
-from modular_graph_and_candidates import build_graph_json, generate_html, build_timeline_json, generate_timeline_html, filter_graph_by_threshold
+from modular_graph_and_candidates import (
+    build_graph_json,
+    generate_html,
+    build_timeline_json,
+    generate_timeline_html,
+    filter_graph_by_threshold,
+)
 
 def _escape_srcdoc(text: str) -> str:
-    """Escape for inclusion inside an <iframe srcdoc="…"> attribute."""
     return (
         text.replace("&", "&amp;")
             .replace("\"", "&quot;")
@@ -23,12 +24,10 @@ def _escape_srcdoc(text: str) -> str:
             .replace(">", "&gt;")
     )
 
-
 HF_MAIN_REPO = "https://github.com/huggingface/transformers"
-
 CACHE_REPO = "Molbap/hf_cached_embeds_log"
 
-def _fetch_from_cache_repo(kind: str, sim_method: str, threshold: float, multimodal: bool):
+def _fetch_from_cache_repo(kind: str, sim_method: str, threshold: float, multimodal: bool, *, height_vh: int = 85):
     repo_id = CACHE_REPO
     latest_fp = hf_hub_download(repo_id=repo_id, filename="latest.json", repo_type="dataset")
     info = json.loads(Path(latest_fp).read_text(encoding="utf-8"))
@@ -40,49 +39,42 @@ def _fetch_from_cache_repo(kind: str, sim_method: str, threshold: float, multimo
     filtered_data = filter_graph_by_threshold(raw_data, threshold)
 
     if kind == "timeline":
-        from modular_graph_and_candidates import generate_timeline_html
         raw_html = generate_timeline_html(filtered_data)
     else:
         raw_html = generate_html(filtered_data)
 
-    iframe_html = f'<iframe style="width:100%;height:85vh;border:none;" srcdoc="{_escape_srcdoc(raw_html)}"></iframe>'
+    iframe_html = f'<iframe style="width:100%;height:{height_vh}vh;border:none;" srcdoc="{_escape_srcdoc(raw_html)}"></iframe>'
     tmp = Path(tempfile.mkstemp(suffix=("_timeline.json" if kind == "timeline" else ".json"))[1])
     tmp.write_text(json.dumps(filtered_data), encoding="utf-8")
     return iframe_html, str(tmp)
 
-
-
-def run_loc(sim_method: str, multimodal: bool):
+def run_loc(sim_method: str, multimodal: bool, *, height_vh: int = 85):
     latest_fp = hf_hub_download(repo_id=CACHE_REPO, filename="latest.json", repo_type="dataset")
     info = json.loads(Path(latest_fp).read_text(encoding="utf-8"))
     sha = info["sha"]
     key = f"{sha}/{sim_method}-m{int(multimodal)}"
     html_fp = hf_hub_download(repo_id=CACHE_REPO, filename=f"loc/{key}.html", repo_type="dataset")
     raw_html = Path(html_fp).read_text(encoding="utf-8")
-    iframe_html = f'<iframe style="width:100%;height:85vh;border:none;" srcdoc="{_escape_srcdoc(raw_html)}"></iframe>'
+    iframe_html = f'<iframe style="width:100%;height:{height_vh}vh;border:none;" srcdoc="{_escape_srcdoc(raw_html)}"></iframe>'
     return iframe_html
 
+def run_graph(repo_url: str, threshold: float, multimodal: bool, sim_method: str, *, height_vh: int = 85):
+    return _fetch_from_cache_repo("graph", sim_method, threshold, multimodal, height_vh=height_vh)
 
-def run_graph(repo_url: str, threshold: float, multimodal: bool, sim_method: str):
-    return _fetch_from_cache_repo("graph", sim_method, threshold, multimodal)
-
-
-def run_timeline(repo_url: str, threshold: float, multimodal: bool, sim_method: str):
-    return _fetch_from_cache_repo("timeline", sim_method, threshold, multimodal)
-
-
+def run_timeline(repo_url: str, threshold: float, multimodal: bool, sim_method: str, *, height_vh: int = 85):
+    return _fetch_from_cache_repo("timeline", sim_method, threshold, multimodal, height_vh=height_vh)
 
 # ───────────────────────────── UI ────────────────────────────────────────────────
 
-
 CUSTOM_CSS = """
 #graph_html iframe, #timeline_html iframe {height:85vh !important; width:100% !important; border:none;}
 """
+
 TAB_INDEX = {"timeline": 0, "loc": 1, "graph": 2}
 
 with gr.Blocks(css=CUSTOM_CSS) as demo:
     header = gr.Markdown("## 🔍 Modular-candidate explorer for 🤗 Transformers")
-    # capture the Tabs as a component so we can control selection
+
     with gr.Tabs() as tabs:
         with gr.Tab("Chronological Timeline", id="timeline"):
             with gr.Row():
@@ -124,33 +116,38 @@ with gr.Blocks(css=CUSTOM_CSS) as demo:
                 [repo_in, thresh, multi_cb],
                 [graph_html_out, graph_json_out],
             )
-        embed_html = gr.HTML(visible=False) 
+
+    # make embed_html a sibling of Tabs (not a child), so we can hide Tabs but show this
+    embed_html = gr.HTML(visible=False)
 
     def _on_load(req: gr.Request):
         qp = req.query_params or {}
         tab_key = (qp.get("tab") or "").lower()
         embed = (qp.get("embed") == "1")
-    
         tab_sel = TAB_INDEX.get(tab_key, 0)
-    
+
         if embed:
+            # shorter iframe inside article view
             if tab_key == "graph":
-                html, _ = run_graph(HF_MAIN_REPO, 0.7, False, "jaccard")
+                html, _ = run_graph(HF_MAIN_REPO, 0.7, False, "jaccard", height_vh=60)
             elif tab_key == "timeline":
-                html, _ = run_timeline(HF_MAIN_REPO, 0.7, False, "jaccard")
-            else:  # "loc" or anything else
-                html = run_loc("jaccard", False)
-    
+                html, _ = run_timeline(HF_MAIN_REPO, 0.7, False, "jaccard", height_vh=60)
+            else:
+                html = run_loc("jaccard", False, height_vh=60)
+
             return (
-                gr.update(visible=False),        # header
-                gr.update(visible=False),        # tabs
-                gr.update(value=html, visible=True),  # embed_html
+                gr.Markdown.update(visible=False),                    # header
+                gr.Tabs.update(visible=False),                        # tabs
+                gr.HTML.update(value=html, visible=True),             # embed_html
             )
-    
+
         return (
-            gr.update(visible=True),           # header
-            gr.update(visible=True, selected=tab_sel),  # tabs
-            gr.update(visible=False),          # embed_html
+            gr.Markdown.update(visible=True),
+            gr.Tabs.update(visible=True, selected=tab_sel),
+            gr.HTML.update(visible=False),
         )
-    
+
     demo.load(_on_load, outputs=[header, tabs, embed_html])
+
+if __name__ == "__main__":
+    demo.launch(allowed_paths=["static"])