Update app.py

app.py

@@ -125,16 +125,33 @@ 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) 
 
-    def _select_tab_on_load(req: gr.Request):
-        qp = req.query_params or {}
-        key = (qp.get("tab") or "").lower()
-        idx = TAB_INDEX.get(key)
-        if idx is not None:
-            return gr.Tabs.update(selected=idx)
-        return gr.Tabs.update()
-    
-    demo.load(_select_tab_on_load, outputs=[tabs])
-
-if __name__ == "__main__":
-    demo.launch(allowed_paths=["static"])
+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:
+        if tab_key == "graph":
+            html, _ = run_graph(HF_MAIN_REPO, 0.7, False, "jaccard")
+        elif tab_key == "timeline":
+            html, _ = run_timeline(HF_MAIN_REPO, 0.7, False, "jaccard")
+        else:  # "loc" or anything else
+            html = run_loc("jaccard", False)
+
+        return (
+            gr.update(visible=False),        # header
+            gr.update(visible=False),        # tabs
+            gr.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
+    )
+
+demo.load(_on_load, outputs=[header, tabs, embed_html])