Update Home.py

Home.py

@@ -5,6 +5,8 @@ import numpy as np
 import re
 from eldar import Query
 import altair as alt
+from itertools import combinations
+from collections import Counter
 
 
 
@@ -236,17 +238,49 @@ try:
             st.pyplot(fig)
             st.write(" ")
 
+            
+            #heatmap institusi
+            pairs = []
+
+            for row in key_df_j["Institution"]:
+                cities = [c.strip() for c in row.replace(";", ",").split(",")]
+                for c1, c2 in combinations(cities, 2):
+                    pairs.append(tuple(sorted([c1, c2])))
+            
+            # Hitung frekuensi
+            counter = Counter(pairs)
+            
+            # Ubah jadi DataFrame pasangan
+            df_pairs = pd.DataFrame(
+                [(c1, c2, count) for (c1, c2), count in counter.items()],
+                columns=["Institusi 1", "Institusi 2", "Jumlah"]
+            )
+            
+            # Heatmap Altair
+            heatmap = (
+                alt.Chart(df_pairs)
+                .mark_rect()
+                .encode(
+                    x=alt.X("Institusi 1:N", sort=None),
+                    y=alt.Y("Institusi 2:N", sort=None),
+                    color=alt.Color("Jumlah:Q", scale=alt.Scale(scheme="blues")),
+                    tooltip=["Institusi 1", "Institusi 2", "Jumlah"]
+                )
+            )
+            
+            st.altair_chart(heatmap, use_container_width=True)
+
 
 
             # layout
             col1, col2 = st.columns(2)
-            with col1.container(border=True, gap="medium"):
+            with col1.container(border=True, gap="medium", height=520):
                 st.bar_chart(jrnl_counts_df, use_container_width=True, horizontal=True, width=500, height=500)
-            with col2.container(border=True, gap="medium"):
+            with col2.container(border=True, gap="medium", height=520):
                 st.line_chart(year_counts_df, use_container_width=True, width=500, height=500)
-            with col1.container(border=True, gap="medium"):
+            with col1.container(border=True, gap="medium", height=520):
                 st.altair_chart(chart_ins, use_container_width=True)
-            with col2.container(border=True, gap="medium"):
+            with col2.container(border=True, gap="medium", height=520):
                 st.altair_chart(chart_aut, use_container_width=True)