Update app.py

app.py

@@ -1054,6 +1054,7 @@ 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
@@ -1073,19 +1074,43 @@ if WORDCLOUD_AVAILABLE:
     # 2 Kolom
     col1, col2 = st.columns(2)
 
-    # === PIE CHART: Semua temuan_kategori ===
+    # === PIE CHART: Semua temuan_kategori (Positive Hijau, Lainnya Merah Pastel) ===
     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
+                colors = []
+                for cat in category_counts.index:
+                    if cat == 'Positive':
+                        colors.append('#2E7D32')  # Hijau
+                    else:
+                        # Gradasi merah pastel: dari gelap ke terang
+                        # Urutan: index 0 = terbanyak
+                        idx = list(category_counts.index).index(cat)
+                        # Misal: 0 → merah gelap, 1 → sedikit lebih terang, dst.
+                        # Gunakan indeks untuk gradasi
+                        total_non_positive = len([c for c in category_counts.index if c != 'Positive'])
+                        if total_non_positive > 0:
+                            # Buat gradasi dari merah gelap ke terang
+                            step = idx / max(1, total_non_positive - 1) if total_non_positive > 1 else 0
+                            # Rentang merah pastel: #c62828 ke #ffebee
+                            r = int(198 - (198 - 255) * step)
+                            g = int(40 - (40 - 235) * step)
+                            b = int(40 - (40 - 238) * step)
+                            color_hex = f"#{r:02x}{g:02x}{b:02x}"
+                        else:
+                            color_hex = "#c62828"  # Default merah gelap
+                        colors.append(color_hex)
+
                 # Buat pie chart
                 fig_pie = px.pie(
                     names=category_counts.index,
                     values=category_counts.values,
                     title="Distribution of All Issue Categories",
-                    color_discrete_sequence=px.colors.sequential.Viridis
+                    color_discrete_sequence=colors
                 )
                 fig_pie.update_traces(textposition='inside', textinfo='percent+label')
                 fig_pie.update_layout(height=500)
@@ -1119,6 +1144,7 @@ if WORDCLOUD_AVAILABLE:
 
                     # Output kecil tapi tajam
                     fig, ax = plt.subplots(figsize=(3, 2), dpi=200)
+                    ax.set_title("Unsafe Issues", fontsize=10)
                     ax.imshow(wordcloud, interpolation='bilinear')
                     ax.axis('off')
                     plt.tight_layout()