Update app.py

app.py

@@ -1064,85 +1064,139 @@ with col_3d:
             insight_text = "<div class='ai-insight'>Insufficient data for insight.</div>"
         st.markdown(insight_text, unsafe_allow_html=True)
 
-
-
-# =================== OBJECTIVE 4 - What Unsafe Issues Appear Most Often? ===================
-st.markdown("<h3 class='section-title'>OBJECTIVE 4 — What Unsafe Issues Appear Most Often?</h3>", unsafe_allow_html=True)
-
-# Function for consistent title
-def create_consistent_title(title_text):
-    return f"<div style='font-family: Arial; font-size: 16px; font-weight: bold; color: #2c3e50; text-align: center; margin-bottom: 8px;'>{title_text}</div>"
-
-# Try to import wordcloud
 try:
     from wordcloud import WordCloud
     import matplotlib.pyplot as plt
     import plotly.express as px
-    import plotly.graph_objects as go
     WORDCLOUD_AVAILABLE = True
 except ImportError:
     WORDCLOUD_AVAILABLE = False
 
+st.markdown("<h3 class='section-title'>OBJECTIVE 4 - What Unsafe Issues Appear Most Often?</h3>", 
+            unsafe_allow_html=True)
+
+# 🔹 Fungsi untuk membuat judul seragam
+def create_consistent_title(title_text):
+    return f"<div style='font-family: Arial; font-size: 16px; font-weight: bold; color: #2c3e50; text-align: center; margin-bottom: 8px;'>{title_text}</div>"
+
 if WORDCLOUD_AVAILABLE:
-    # Prepare data
-    df_all = df_local.copy()
-    
-    # 2-column layout
-    col_chart, col_cloud = st.columns([1.2, 1])
 
-    # === PIE CHART (with custom legend logic) ===
-    with col_chart:
+    # 🔥 Data untuk pie chart: semua kategori
+    df_all_kategori = df_local.copy()
+
+    # 🔥 Data untuk wordcloud: hanya Non-Positive
+    df_filtered_kategori = df_local[df_local['temuan_kategori'] != 'Positive'] \
+        if 'temuan_kategori' in df_local.columns else df_local
+
+    # 2 Kolom
+    col1, col2 = st.columns(2)
+
+    # === PIE CHART: Semua temuan_kategori (Tanpa Legend Default, Custom Legend Khusus) ===
+    with col1:
         st.markdown(create_consistent_title("Distribution of All Issue Categories"), unsafe_allow_html=True)
-        
-        if 'temuan_kategori' not in df_all.columns:
-            st.warning("Column 'temuan_kategori' not found.")
-        else:
-            # Count categories
-            cat_counts = df_all['temuan_kategori'].value_counts()
-            total = cat_counts.sum()
-            percentages = (cat_counts / total * 100).round(1)
-
-            # Define colors (sesuai preferensi Anda)
-            color_map = {
-                'Positive': '#2E7D32',        # Hijau
-                'Unsafe Condition': '#EF5350',  # Merah tua (pastel red)
-                'Unsafe Action': '#F48FB1',    # Pink pastel
-                'Near Miss': '#BDBDBD',        # Abu-abu
-            }
-            # Default color for unknown
-            default_color = '#9E9E9E'
-
-            # Assign colors
-            colors = [color_map.get(cat, default_color) for cat in cat_counts.index]
-
-            # Create base pie chart — tanpa legend
-            fig = go.Figure(data=[go.Pie(
-                labels=cat_counts.index,
-                values=cat_counts.values,
-                marker=dict(colors=colors),
-                textinfo='percent+label',
-                textposition='inside',
-                hoverinfo='label+percent',
-                sort=False  # pertahankan urutan asli
-            )])
-            fig.update_layout(
-                height=450,
-                showlegend=False,  # ❌ Matikan legend default Plotly
-                margin=dict(t=20, b=20, l=20, r=20)
-            )
-            st.plotly_chart(fig, use_container_width=True)
 
-            # === 🔥 CUSTOM LEGEND: hanya tampilkan "Unsafe Action" & "Near Miss" sebagai garis jika <10% ===
-            st.markdown("#### Legend")
+        if 'temuan_kategori' in df_all_kategori.columns:
+            category_counts = df_all_kategori['temuan_kategori'].value_counts()
             
-            # Buat container HTML untuk legend custom
-            legend_items = []
+            if not category_counts.empty:
+                # Mapping warna
+                color_map = {
+                    'Positive': '#2E7D32',           # Hijau
+                    'Unsafe Condition': '#EF5350',    # Merah Muda
+                    'Unsafe Action': '#F48FB1',       # Pink
+                    'Near Miss': '#BDBDBD'            # Abu-abu
+                }
+                colors = [color_map.get(cat, '#9E9E9E') for cat in category_counts.index]
 
-            # Selalu tampilkan kategori lain sebagai blok biasa (opsional — bisa dihapus jika hanya ingin 2 line)
-            # Tapi sesuai permintaan: "hanya 2 garis untuk 2 kategori tsb"
-            # → Jadi kita tampilkan **hanya** UA & NM jika <10%, dalam bentuk garis
+                # Buat pie chart tanpa legend
+                fig_pie = px.pie(
+                    names=category_counts.index,
+                    values=category_counts.values,
+                    color_discrete_sequence=colors
+                )
+                fig_pie.update_traces(
+                    textposition='inside',
+                    textinfo='percent+label',
+                    marker=dict(line=dict(color='#FFFFFF', width=1))
+                )
+                fig_pie.update_layout(
+                    showlegend=False,  # ❌ matikan legend default
+                    height=450,
+                    margin=dict(t=20, b=20, l=40, r=40)
+                )
+                st.plotly_chart(fig_pie, use_container_width=True)
+
+                # ✅ CUSTOM LEGEND: Gabungan lingkaran + garis
+                st.markdown("<div style='text-align: center; font-size: 14px; margin-top: -10px;'>", unsafe_allow_html=True)
+                legend_items = []
+
+                for cat in category_counts.index:
+                    color = color_map.get(cat, '#9E9E9E')
+                    if cat in ['Unsafe Action', 'Near Miss']:
+                        # Gunakan garis horizontal kecil
+                        item = (
+                            f"<span style='display: inline-block; width: 16px; height: 4px; background-color: {color}; "
+                            f"margin: 0 8px; border-radius: 1px;'></span>"
+                            f"{cat}"
+                        )
+                    else:
+                        # Gunakan lingkaran kecil (default)
+                        item = (
+                            f"<span style='display: inline-block; width: 12px; height: 12px; background-color: {color}; "
+                            f"border-radius: 50%; margin: 0 8px;'></span>"
+                            f"{cat}"
+                        )
+                    legend_items.append(item)
+
+                # Gabungkan semua item
+                legend_html = " | ".join(legend_items)
+                st.markdown(f"<div>{legend_html}</div>", unsafe_allow_html=True)
+                st.markdown("</div>", unsafe_allow_html=True)
+
+            else:
+                st.warning("No data available for pie chart.")
+        else:
+            st.warning("Column 'temuan_kategori' not available.")
+
+    # === WORDCLOUD: Hanya Non-Positive (dari keyword_kategori) ===
+    with col2:
+        st.markdown(create_consistent_title("Unsafe Issues"), unsafe_allow_html=True)
+
+        if df_filtered_kategori.empty:
+            st.warning("No data available after filtering out 'Positive' category.")
+        else:
+            if 'keyword_kategori' in df_filtered_kategori.columns:
+                # Gabungkan text
+                import re
+                text = ' '.join(df_filtered_kategori['keyword_kategori'].dropna().astype(str))
+                text = re.sub(r'[^a-zA-Z\s]', ' ', text)
+
+                if text.strip():
+                    # Buat WordCloud
+                    wordcloud = WordCloud(
+                        width=1600,          # resolusi besar agar HD
+                        height=800,
+                        background_color='white',
+                        colormap='viridis',
+                        max_words=1000,
+                        random_state=42
+                    ).generate(text)
+
+                    # Output kecil tapi tajam
+                    fig, ax = plt.subplots(figsize=(3, 2), dpi=200)
+                    ax.imshow(wordcloud, interpolation='bilinear')
+                    ax.axis('off')
+                    plt.tight_layout()
+
+                    st.pyplot(fig, use_container_width=True)
+                else:
+                    st.warning("No valid text remaining after cleaning.")
+            else:
+                st.warning("Column 'keyword_kategori' not available.")
+
+else:
+    st.info("WordCloud library not installed. Install `wordcloud` and `matplotlib` to enable this feature.")
 
-            target
 # =================== 5. Matrix (Tetap Dipertahankan) ===================
 st.markdown("<h3 class='section-title'>OBJECTIVE 5 - Findings vs Lead Time: Which Divisions Move Slow?</h3>", unsafe_allow_html=True)