Upload app.py

src/app.py

@@ -510,6 +510,15 @@ with tab_overview:
                 .update_layout(xaxis_title="Year", yaxis_title="Citations"),
                 use_container_width=True)
 
+        st.subheader("Overall intent distribution")
+        all_intents = events.groupby("primary_intent").size().to_dict()
+        ai_df = pd.DataFrame({"intent": ALLOWED_INTENTS,
+                               "count": [int(all_intents.get(i, 0)) for i in ALLOWED_INTENTS]})
+        fig2 = px.bar(ai_df, x="intent", y="count", color="intent",
+                      color_discrete_map=INTENT_COLORS)
+        fig2.update_layout(showlegend=False, xaxis_title="", yaxis_title="Count")
+        st.plotly_chart(fig2, use_container_width=True)
+
         st.subheader("Field distribution")
         fd = (seed_filtered.groupby("field", dropna=False).size()
               .reset_index(name="count").sort_values("count", ascending=False).head(20))
@@ -556,71 +565,58 @@ with tab_ontology:
 with tab_kg:
     st.subheader("Knowledge Graph — Selected Seed Paper")
     st.caption("kg_nodes + kg_edges에서 선택된 seed paper의 1-hop 서브그래프 (DuckDB 부분 쿼리)")
-    st.info("아래 버튼을 눌러 KG 데이터를 로드하세요. kg_nodes만 전체 로드, kg_edges는 필요한 부분만 쿼리합니다.")
 
     max_edges_kg = st.slider("Max edges", 20, 150, 80, key="kg_max_edges")
 
-    if st.button("KG 데이터 로드", key="kg_load"):
-        with st.spinner("kg_nodes 로딩 + kg_edges 경로 준비 중..."):
-            st.session_state["kg_loaded"] = True
+    try:
+        with st.spinner("KG 데이터 로딩 중... (최초 1회 후 캐시됩니다)"):
+            kg_nodes       = load_kg_nodes(data_dir_val)
+            kg_edges_path  = get_parquet_path("kg_edges.parquet", data_dir_val)
 
-    if st.session_state.get("kg_loaded"):
-        try:
-            with st.spinner("kg_nodes 로딩 중..."):
-                kg_nodes = load_kg_nodes(data_dir_val)
+        seed_doi = selected_seed["doi"]
+        if not seed_doi:
+            st.warning("선택된 seed paper의 DOI가 없어 KG 조회가 불가합니다.")
+        else:
+            node_id = f"seed:{seed_doi}"
+            with st.spinner("kg_edges 쿼리 중 (DuckDB)..."):
+                edges_sub = query_kg_edges_for_node(node_id, kg_edges_path, max_edges_kg)
 
-            seed_doi = selected_seed["doi"]
-            if not seed_doi:
-                st.warning("선택된 seed paper의 DOI가 없어 KG 조회가 불가합니다.")
+            if edges_sub.empty:
+                st.warning(f"KG에서 해당 노드의 엣지를 찾을 수 없습니다. (node_id: {node_id})")
             else:
-                node_id = f"seed:{seed_doi}"
-
-                with st.spinner("kg_edges 쿼리 중 (DuckDB)..."):
-                    kg_edges_path = get_parquet_path("kg_edges.parquet", data_dir_val)
-                    edges_sub = query_kg_edges_for_node(node_id, kg_edges_path, max_edges_kg)
-
-                if edges_sub.empty:
-                    st.warning(f"KG에서 해당 노드의 엣지를 찾을 수 없습니다. (node_id: {node_id})")
-                else:
-                    all_node_ids = set(edges_sub["source"].tolist()) | set(edges_sub["target"].tolist())
-                    nodes_sub = kg_nodes[kg_nodes["node_id"].isin(all_node_ids)]
-
-                    c1, c2, c3 = st.columns(3)
-                    c1.metric("Nodes", fmt_num(len(nodes_sub)))
-                    c2.metric("Edges", fmt_num(len(edges_sub)))
-                    c3.metric("Node types", fmt_num(nodes_sub["node_type"].nunique()))
-
-                    type_counts = nodes_sub["node_type"].value_counts().reset_index()
-                    type_counts.columns = ["node_type", "count"]
-                    st.plotly_chart(
-                        px.bar(type_counts, x="node_type", y="count",
-                               color="node_type", color_discrete_map=NODE_TYPE_COLORS,
-                               title="Node Type Distribution")
-                        .update_layout(showlegend=False, xaxis_title="", yaxis_title="Count"),
-                        use_container_width=True)
-
-                    st.caption("🖱 Scroll: zoom  |  Drag: pan  |  Click node: info  |  ⛶ button: fullscreen")
-                    components.html(pyvis_from_kg(nodes_sub, edges_sub), height=820, scrolling=True)
-        except Exception as e:
-            st.error(str(e))
+                all_node_ids = set(edges_sub["source"].tolist()) | set(edges_sub["target"].tolist())
+                nodes_sub = kg_nodes[kg_nodes["node_id"].isin(all_node_ids)]
+
+                c1, c2, c3 = st.columns(3)
+                c1.metric("Nodes",      fmt_num(len(nodes_sub)))
+                c2.metric("Edges",      fmt_num(len(edges_sub)))
+                c3.metric("Node types", fmt_num(nodes_sub["node_type"].nunique()))
+
+                type_counts = nodes_sub["node_type"].value_counts().reset_index()
+                type_counts.columns = ["node_type", "count"]
+                st.plotly_chart(
+                    px.bar(type_counts, x="node_type", y="count",
+                           color="node_type", color_discrete_map=NODE_TYPE_COLORS,
+                           title="Node Type Distribution")
+                    .update_layout(showlegend=False, xaxis_title="", yaxis_title="Count"),
+                    use_container_width=True)
+
+                st.caption("🖱 Scroll: zoom  |  Drag: pan  |  Click node: info  |  ⛶ button: fullscreen")
+                components.html(pyvis_from_kg(nodes_sub, edges_sub), height=820, scrolling=True)
+    except Exception as e:
+        st.error(str(e))
 
 
 # ═══ 5. KG EXPLORER ═════════════════════════════════════════════
 with tab_kg_exp:
     st.subheader("KG Explorer")
     st.caption("kg_nodes를 탐색하고 임의 노드의 연결 관계를 시각화합니다. kg_edges는 DuckDB로 필요한 부분만 쿼리합니다.")
-    st.info("아래 버튼을 눌러 KG 데이터를 로드하세요. kg_nodes만 전체 로드되고, kg_edges/enriched는 DuckDB로 쿼리합니다.")
-
-    if st.button("KG 데이터 로드", key="kg_exp_load"):
-        with st.spinner("kg_nodes 로딩 중..."):
-            st.session_state["kg_loaded"] = True
-
-    if st.session_state.get("kg_loaded"):
-        try:
-            with st.spinner("kg_nodes 로딩 중..."):
-                kg_nodes = load_kg_nodes(data_dir_val)
-                kg_edges_path   = get_parquet_path("kg_edges.parquet", data_dir_val)
-                enriched_path   = get_parquet_path("citation_events_enriched.parquet", data_dir_val)
+
+    try:
+        with st.spinner("KG 데이터 로딩 중... (최초 1회 후 캐시됩니다)"):
+            kg_nodes      = load_kg_nodes(data_dir_val)
+            kg_edges_path = get_parquet_path("kg_edges.parquet", data_dir_val)
+            enriched_path = get_parquet_path("citation_events_enriched.parquet", data_dir_val)
 
             # ── 전체 노드 타입 분포 (kg_nodes만으로 계산)
             col_a, col_b = st.columns([1,2])