Update app.py

app.py

@@ -359,22 +359,11 @@ with st.container():
     st.markdown('</div>', unsafe_allow_html=True)
 
 
-
 import streamlit as st
 import plotly.graph_objects as go
 import numpy as np
 import pandas as pd
 
-import streamlit as st
-import plotly.graph_objects as go
-import numpy as np
-import pandas as pd
-
-# =================== OBJECTIVE 1 - Company Reporting Activity (Polar Bar Chart) ===================
-import streamlit as st
-import plotly.graph_objects as go
-import numpy as np
-import pandas as pd
 # =================== OBJECTIVE 1 - Company Reporting Activity (Polar Bar Chart) ===================
 st.markdown(
     """
@@ -502,18 +491,22 @@ if 'temuan_kode_distrik' in df_local.columns:
             current_angle += a
         
         fig = go.Figure()
-        fig.add_trace(go.Barpolar(
-            r=ratios,
-            theta=mid_angles,
-            width=angles,
-            marker_color=colors,
-            marker_line_color="white",
-            marker_line_width=1.2,
-            opacity=0.9,
-            hovertemplate="<b>%{text}</b><br>Avg Ratio: %{r:.2f}<extra></extra>",
-            text=companies,
-            name='Companies',  # Nama trace untuk legend
-        ))
+        
+        # 🔥 Tambahkan satu trace untuk setiap perusahaan agar muncul di legend
+        for i, (comp, ratio, color, angle) in enumerate(zip(companies, ratios, colors, angles)):
+            fig.add_trace(go.Barpolar(
+                r=[ratio],
+                theta=[mid_angles[i]],
+                width=[angle],
+                marker_color=[color],
+                marker_line_color="white",
+                marker_line_width=1.2,
+                opacity=0.9,
+                hovertemplate="<b>%{text}</b><br>Avg Ratio: %{r[0]:.2f}<extra></extra>",
+                text=[comp],
+                name=comp,  # 🔥 Nama perusahaan untuk legend
+                showlegend=True  # 🔥 Tampilkan di legend
+            ))
 
         fig.update_layout(
             title=f'{area_name} Area',
@@ -533,20 +526,23 @@ if 'temuan_kode_distrik' in df_local.columns:
                     linecolor="white",  # 🔥 Warna garis derajat jadi putih
                     gridcolor="rgba(255,255,255,0.2)"  # 🔥 Warna grid derajat jadi transparan putih
                 ),
-                showlegend=True,  # 🔥 Aktifkan legend
-                legend=dict(
-                    orientation="v",
-                    yanchor="top",     # 🔥 Atas
-                    y=1,
-                    xanchor="right",   # 🔥 Kanan
-                    x=1.02,
-                    font=dict(size=10)
-                ),
+            ),
+            showlegend=True,  # 🔥 Aktifkan legend
+            legend=dict(
+                orientation="v",
+                yanchor="top",     # 🔥 Atas
+                y=1,
+                xanchor="right",   # 🔥 Kanan
+                x=1.02,
+                font=dict(size=10)
+            ),
             margin=dict(t=40, b=20, l=20, r=20),
             height=400,
             paper_bgcolor="rgba(0,0,0,0)",
-            plot_bgcolor="rgba(0,0,0,0)"))
+            plot_bgcolor="rgba(0,0,0,0)"
+        )
 
+        return fig
 
     # Plot
     col1, col2 = st.columns(2)