Update app.py

app.py

@@ -1047,14 +1047,10 @@ with col_3d:
         st.markdown(insight_text, unsafe_allow_html=True)
 
 
-
-####OBJECTIVE 4
-# ================= WORDCLOUD (HIGH QUALITY) ==================
 try:
     from wordcloud import WordCloud
     import matplotlib.pyplot as plt
     import plotly.express as px
-    import numpy as np
     WORDCLOUD_AVAILABLE = True
 except ImportError:
     WORDCLOUD_AVAILABLE = False
@@ -1074,40 +1070,26 @@ if WORDCLOUD_AVAILABLE:
     # 2 Kolom
     col1, col2 = st.columns(2)
 
-    # === PIE CHART: Semua temuan_kategori (Positive Hijau, Lainnya Merah Pastel) ===
+    # === PIE CHART: Semua temuan_kategori (Warna Sesuai Kategori) ===
     with col1:
         if 'temuan_kategori' in df_all_kategori.columns:
             # Hitung jumlah per kategori
             category_counts = df_all_kategori['temuan_kategori'].value_counts()
             
             if not category_counts.empty:
-                # Buat warna custom
+                # Buat warna sesuai kategori
                 colors = []
                 for cat in category_counts.index:
                     if cat == 'Positive':
                         colors.append('#2E7D32')  # Hijau
+                    elif cat == 'Unsafe Condition':
+                        colors.append('#EF5350')  # Merah Muda
+                    elif cat == 'Unsafe Action':
+                        colors.append('#F48FB1')  # Pink
+                    elif cat == 'Near Miss':
+                        colors.append('#BDBDBD')  # Abu-abu
                     else:
-                        # Gradasi merah pastel: dari gelap ke terang
-                        idx = list(category_counts.index).index(cat)
-                        total_non_positive = len([c for c in category_counts.index if c != 'Positive'])
-                        
-                        if total_non_positive > 1:
-                            step = idx / (total_non_positive - 1)
-                        else:
-                            step = 0
-                        
-                        # Rentang merah pastel: dari #c62828 ke #ffebee
-                        r = int(198 + (255 - 198) * step)
-                        g = int(40 + (235 - 40) * step)
-                        b = int(40 + (238 - 40) * step)
-                        
-                        # Pastikan nilai tidak melebihi 255
-                        r = min(255, max(0, r))
-                        g = min(255, max(0, g))
-                        b = min(255, max(0, b))
-                        
-                        color_hex = f"#{r:02x}{g:02x}{b:02x}"
-                        colors.append(color_hex)
+                        colors.append('#9E9E9E')  # Default abu-abu
 
                 # Buat pie chart
                 fig_pie = px.pie(
@@ -1148,7 +1130,7 @@ if WORDCLOUD_AVAILABLE:
 
                     # Output kecil tapi tajam
                     fig, ax = plt.subplots(figsize=(3, 2), dpi=200)
-                    ax.set_title("Unsafe Issues", fontsize=7)
+                    ax.set_title("Unsafe Issues", fontsize=10)
                     ax.imshow(wordcloud, interpolation='bilinear')
                     ax.axis('off')
                     plt.tight_layout()
@@ -1161,7 +1143,6 @@ if WORDCLOUD_AVAILABLE:
 
 else:
     st.info("WordCloud library not installed. Install `wordcloud` and `matplotlib` to enable this feature.")
-
 # =================== 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)