feat: Rename engine to `ArchonMasterEngine`, add `grocer` and `utilities` to essential keywords, refine risk scoring and NBO messages, and introduce a new balance trend visualization.

app.py

@@ -13,15 +13,25 @@ client_ai = genai.Client(api_key=GOOGLE_API_KEY) if GOOGLE_API_KEY else None
 custom_css = """
 @import url('https://fonts.googleapis.com/css2?family=Plus+Jakarta+Sans:wght@400;600;700;800&display=swap');
 body, .gradio-container { font-family: 'Plus Jakarta Sans', sans-serif !important; background-color: #FFFFFF !important; }
-.nagari-header { background: linear-gradient(135deg, #05A4DE 0%, #82C3EB 100%); padding: 35px; border-radius: 15px; border-bottom: 6px solid #F7BD87; text-align: center; margin-bottom: 25px; }
-.nagari-header h1 { color: #FFFFFF !important; font-weight: 800 !important; margin: 0; font-size: 2.2em; }
-.card-sidebar { background: #E0EDF4; border-radius: 15px; padding: 25px; border: 1.5px solid #82C3EB; box-shadow: 0 4px 12px rgba(5, 164, 222, 0.1); }
+
+.nagari-header { 
+    background: linear-gradient(135deg, #05A4DE 0%, #82C3EB 100%); 
+    padding: 35px; border-radius: 15px; border-bottom: 6px solid #F7BD87; 
+    margin-bottom: 25px; text-align: center;
+}
+.nagari-header h1 { color: #FFFFFF !important; font-weight: 800 !important; margin: 0; font-size: 2.2em; text-shadow: 2px 2px 4px rgba(0,0,0,0.2); }
+
+.card-sidebar { 
+    background: #E0EDF4; border-radius: 15px; padding: 25px; 
+    border: 1.5px solid #82C3EB; box-shadow: 0 4px 12px rgba(5, 164, 222, 0.1); 
+}
 .health-badge { background: white; padding: 12px; border-radius: 8px; margin-bottom: 12px; border-left: 5px solid #05A4DE; font-size: 0.95em; }
+
 .report-card { background: white; border-radius: 12px; padding: 30px; border: 1px solid #E2E8F0; line-height: 1.8; color: #1e293b; }
 .nbo-box { background: #fffdf0; border: 2px solid #F7BD87; padding: 20px; border-radius: 10px; margin-top: 20px; }
 """
 
-class ArchonFinalEngine:
+class ArchonMasterEngine:
     def __init__(self):
         self.load_data()
 
@@ -38,73 +48,65 @@ class ArchonFinalEngine:
         u_txn = self.df_txn[self.df_txn['customer_id'] == cid].copy()
         u_bal = self.df_bal[self.df_bal['customer_id'] == cid].sort_values('month')
         u_rep = self.df_rep[self.df_rep['customer_id'] == cid]
-        u_info_list = self.df_cust[self.df_cust['customer_id'] == cid]
+        u_info_df = self.df_cust[self.df_cust['customer_id'] == cid]
 
-        if u_txn.empty or u_info_list.empty: return None
-        u_info = u_info_list.iloc[0]
+        if u_txn.empty or u_info_df.empty: return None
+        u_info = u_info_df.iloc[0]
 
-        # --- FASE 2: TRANSACTION INTELLIGENCE (Semantic) ---
-        essential_list = ['indomaret', 'alfamart', 'listrik', 'pdam', 'telkom', 'sekolah', 'rs ', 'obat', 'cicilan', 'pinjaman', 'gaji', 'asuransi']
+        # --- FASE 2: INTELLIGENCE (Semantic Parser) ---
+        essential_keywords = ['indomaret', 'alfamart', 'listrik', 'pdam', 'telkom', 'sekolah', 'rs ', 'obat', 'cicilan', 'pinjaman', 'gaji', 'asuransi', 'grocer', 'utilities']
         def classify_exp(row):
             desc = str(row.get('raw_description', '')).lower()
-            return 'essential' if any(k in desc for k in essential_list) else 'discretionary'
+            return 'essential' if any(k in desc for k in essential_keywords) else 'discretionary'
         u_txn['expense_type'] = u_txn.apply(classify_exp, axis=1)
 
-        # --- FASE 3 & 4: RISK SCORING (STRICT COMPLIANCE) ---
+        # --- FASE 3 & 4: RISK SCORING ---
         income_txn = u_txn[u_txn['transaction_type'] == 'credit']['amount'].sum()
         ref_income = max(income_txn, u_info['monthly_income'])
         expense = u_txn[u_txn['transaction_type'] == 'debit']['amount'].sum()
         er = expense / ref_income if ref_income > 0 else 1.0
         
-        # Scoring Rules
+        # Bobot 30/20/20/20/10
         er_s = 1.0 if er > 0.8 else (0.5 if er > 0.5 else 0.0)
         bt_s = 1.0 if len(u_bal) >= 2 and u_bal.iloc[-1]['avg_balance'] < u_bal.iloc[0]['avg_balance'] else 0.0
         od_s = 1.0 if (u_bal['min_balance'] <= 0).any() else 0.0
         mp_s = 1.0 if (u_rep['status'] == 'late').any() else 0.0
         
         score = (0.3 * er_s) + (0.2 * bt_s) + (0.2 * od_s) + (0.2 * mp_s) + 0.05
-        
-        # Classification (Sesuai dokumen)
-        if score >= 0.7: risk_lv = "HIGH"
-        elif score >= 0.4: risk_lv = "MEDIUM"
-        else: risk_lv = "LOW"
+        risk_lv = "HIGH" if score >= 0.7 else ("MEDIUM" if score >= 0.4 else "LOW")
 
-        # --- FASE 5: NBO ENGINE (Customer Oriented) ---
+        # --- FASE 5: NBO ENGINE ---
         if risk_lv == "HIGH" or mp_s == 1:
-            action = "Evaluasi Cicilan & Restrukturisasi"
-            nbo_msg = "Untuk membantu meringankan beban finansial, Bapak/Ibu dapat mengajukan perpanjangan tenor atau penyesuaian jadwal pembayaran cicilan."
-            steps = ["Ajukan peninjauan tenor pinjaman", "Konsolidasi tagihan harian", "Manfaatkan konsultasi dana darurat."]
+            action, nbo_msg = "Program Restrukturisasi", "Kami menyarankan penyesuaian jadwal pembayaran cicilan agar kondisi keuangan Anda tetap terjaga."
+            steps = ["Ajukan peninjauan tenor", "Konsolidasi tagihan harian", "Manfaatkan konsultasi dana."]
         elif er > 0.6:
-            action = "Pengaturan Limit Belanja (Spending Control)"
-            nbo_msg = "Kami mendeteksi pengeluaran bulanan yang cukup tinggi. Bapak/Ibu disarankan mengaktifkan fitur limit transaksi QRIS untuk menjaga tabungan."
-            steps = ["Atur limit harian belanja", "Aktifkan notifikasi saldo minimum", "Prioritaskan kebutuhan pokok (Essential)."]
+            action, nbo_msg = "Pengaturan Limit Belanja (Spending Control)", "Kami mendeteksi pola belanja yang tinggi. Anda disarankan mengatur limit transaksi harian agar tabungan tetap aman."
+            steps = ["Atur limit harian QRIS", "Aktifkan notifikasi saldo", "Prioritaskan kebutuhan pokok."]
         elif risk_lv == "LOW":
-            action = "Pengembangan Aset (Saving/Investment)"
-            nbo_msg = "Kondisi keuangan Bapak/Ibu sangat prima. Ini waktu yang tepat untuk menanamkan dana mengendap ke Deposito atau Tabungan Berjangka."
-            steps = ["Buka tabungan berjangka otomatis", "Pindahkan dana ke deposito bunga tinggi", "Eksplorasi produk reksa dana."]
+            action, nbo_msg = "Optimalkan Tabungan (Promote Saving)", "Kondisi keuangan Anda sangat prima. Ini waktu yang tepat untuk menumbuhkan aset Anda melalui Deposito."
+            steps = ["Buka tabungan berjangka", "Pindahkan dana ke deposito", "Eksplorasi produk reksa dana."]
         else:
-            action = "Edukasi Pengelolaan Kas"
-            nbo_msg = "Mari optimalkan arus kas harian Bapak/Ibu dengan fitur budgeting di aplikasi mobile kami agar resiliensi tetap terjaga."
-            steps = ["Baca artikel cerdas belanja", "Gunakan fitur budgeting di aplikasi", "Review pengeluaran akhir bulan."]
+            action, nbo_msg = "Edukasi Pengelolaan Kas", "Mari perkuat daya tahan keuangan Anda dengan tips pengelolaan arus kas di aplikasi mobile kami."
+            steps = ["Baca artikel cerdas belanja", "Gunakan fitur budgeting", "Review pengeluaran bulanan."]
 
         return risk_lv, score, er, u_bal, u_txn, expense, ref_income, action, nbo_msg, steps, er_s, bt_s, od_s, mp_s
 
     def build_narrative(self, risk_lv, score, er, u_bal, exp, inc, action, nbo_msg, steps, cid, u_txn):
-        # FASE 6: EXPLAINABLE SUMMARY (NARRATIVE)
-        msg = f"### ANALISIS RESILIENSI FINANSIAL ANDA\n\n"
-        msg += f"Halo Bapak/Ibu, berdasarkan tinjauan transaksi Anda ({cid}), sistem kami menetapkan tingkat kesehatan finansial Anda pada kategori **{risk_lv}** (Skor: {score:.2f}).\n\n"
+        # FASE 6: EXPLAINABLE SUMMARY
+        msg = f"### LAPORAN ANALISIS RESILIENSI FINANSIAL ANDA\n\n"
+        msg += f"Halo Bapak/Ibu, sistem Archon menetapkan tingkat kesehatan finansial Anda ({cid}) pada kategori **{risk_lv}** (Skor: {score:.2f}).\n\n"
         
         msg += f"**1. Mengapa Skor Ini Muncul?**\n"
         msg += f"* **Efisiensi Belanja ({er:.1%})**: Anda menghabiskan Rp{exp:,.0f} dari pendapatan Rp{inc:,.0f}. "
-        msg += "Artinya, Bapak/Ibu membelanjakan lebih banyak dari yang diterima. Ini mengindikasikan penggunaan dana tabungan untuk konsumsi harian." if er > 1 else "Manajemen belanja Anda terpantau sehat."
+        msg += "Ini menunjukkan pengeluaran yang melebihi pendapatan (defisit)." if er > 1 else "Manajemen belanja Anda terpantau cukup terkendali."
         
         if not u_bal.empty:
-            msg += f"\n* **Analisis Saldo**: Saldo saat ini Rp{u_bal.iloc[-1]['avg_balance']:,.0f}. "
-            msg += "Terdeteksi tren penurunan saldo rata-rata, disarankan untuk mulai menabung kembali." if len(u_bal) > 1 and u_bal.iloc[-1]['avg_balance'] < u_bal.iloc[0]['avg_balance'] else "Saldo Anda tumbuh dengan stabil dan aman."
+            msg += f"\n* **Analisis Saldo**: Saldo rata-rata Anda Rp{u_bal.iloc[-1]['avg_balance']:,.0f}. "
+            msg += "Terdeteksi tren penurunan saldo, disarankan untuk mulai menabung kembali." if len(u_bal) > 1 and u_bal.iloc[-1]['avg_balance'] < u_bal.iloc[0]['avg_balance'] else "Saldo Anda tumbuh dengan stabil dan aman."
 
         msg += f"\n\n<div class='nbo-box'>REKOMENDASI UNTUK ANDA: {action}\n\n"
         msg += f"**Tujuan:** {nbo_msg}\n\n"
-        msg += f"**Langkah Implementasi:**\n"
+        msg += f"**Langkah-Langkah Implementasi:**\n"
         for i, step in enumerate(steps, 1): msg += f"{i}. {step}\n"
         msg += f"</div>"
         return msg
@@ -118,7 +120,7 @@ class ArchonFinalEngine:
         f1.add_trace(go.Bar(x=cf.index, y=cf.get('debit', 0), name='Pengeluaran (Outflow)', marker_color='#05A4DE'))
         f1.update_layout(title="Arus Kas Bulanan", barmode='group', template='plotly_white')
 
-        # 2. Pie Chart (FIXED COLORS)
+        # 2. Pie Chart
         exp_dist = u_txn[u_txn['transaction_type'] == 'debit'].groupby('expense_type')['amount'].sum().reset_index()
         color_map = {'essential': '#05A4DE', 'discretionary': '#F7BD87'}
         f2 = go.Figure(data=[go.Pie(
@@ -126,19 +128,24 @@ class ArchonFinalEngine:
             marker=dict(colors=[color_map.get(x, '#E0EDF4') for x in exp_dist['expense_type']])
         )])
         f2.update_layout(title="Komposisi Pengeluaran (Kebutuhan vs Gaya Hidup)")
+
+        # 3. Tren Saldo (Restored)
+        f3 = go.Figure()
+        f3.add_trace(go.Scatter(x=u_bal['month'], y=u_bal['avg_balance'], name='Saldo Rata-rata', line=dict(color='#F7BD87', width=4)))
+        f3.update_layout(title="Tren Pertumbuhan Saldo", template='plotly_white')
         
-        return f1, f2
+        return f1, f2, f3
 
 # --- UI LOGIC ---
-engine = ArchonFinalEngine()
+engine = ArchonMasterEngine()
 
 def run_app(cust_id):
     res = engine.analyze(cust_id)
-    if not res: return "ID Tidak Valid", "Gunakan C0001 - C0120", None, None
+    if not res: return "ID Tidak Valid", "Gunakan ID C0001 - C0120", None, None, None
     
     risk_lv, score, er, u_bal, u_txn, exp, inc, action, nbo_msg, steps, er_s, bt_s, od_s, mp_s = res
     report = engine.build_narrative(risk_lv, score, er, u_bal, exp, inc, action, nbo_msg, steps, cust_id, u_txn)
-    p1, p2 = engine.create_viz(u_bal, u_txn)
+    p1, p2, p3 = engine.create_viz(u_bal, u_txn)
     
     color = "#ef4444" if risk_lv == "HIGH" else ("#f59e0b" if risk_lv == "MEDIUM" else "#10b981")
     sidebar = f"""
@@ -150,14 +157,14 @@ def run_app(cust_id):
         <div class='health-badge'><b>Kesehatan Saldo:</b> {'🔻 Menurun' if bt_s == 1 else '🔺 Stabil'}</div>
     </div>
     """
-    return sidebar, report, p1, p2
+    return sidebar, report, p1, p2, p3
 
 with gr.Blocks(css=custom_css) as demo:
     gr.HTML("<div class='nagari-header'><h1>ARCHON-AI</h1></div>")
     with gr.Row():
         with gr.Column(scale=1):
             id_in = gr.Textbox(label="Customer ID", placeholder="C0001")
-            btn = gr.Button("ANALYISIS SEKARANG", variant="primary")
+            btn = gr.Button("ANALYZE CUSTOMER", variant="primary")
             out_side = gr.HTML()
         with gr.Column(scale=2):
             with gr.Tabs():
@@ -166,12 +173,16 @@ with gr.Blocks(css=custom_css) as demo:
                 with gr.Tab("Visualisasi Keuangan"):
                     gr.Markdown("### 1. Uang Masuk vs Uang Keluar")
                     plot_cf = gr.Plot()
-                    gr.HTML("<div style='background:#f1f5f9; padding:15px; border-radius:8px;'><b>Interpretasi:</b> Batang muda (Inflow) harus lebih tinggi dari batang tua (Outflow) untuk menjaga saldo tetap aman.</div>")
+                    gr.HTML("<div style='background:#f1f5f9; padding:15px; border-radius:8px;'><b>Interpretasi:</b> Batang muda (Inflow) idealnya lebih tinggi dari batang tua (Outflow).</div>")
                     gr.Markdown("---")
-                    gr.Markdown("### 2. Ke mana Uang Anda Pergi?")
+                    gr.Markdown("### 2. Komposisi Gaya Hidup")
                     plot_dist = gr.Plot()
                     gr.HTML("<div style='background:#f1f5f9; padding:15px; border-radius:8px;'><b>Warna:</b> Biru = Kebutuhan (Essential). Emas = Gaya Hidup (Discretionary).</div>")
+                    gr.Markdown("---")
+                    gr.Markdown("### 3. Tren Pertumbuhan Saldo")
+                    plot_bal = gr.Plot()
+                    gr.HTML("<div style='background:#f1f5f9; padding:15px; border-radius:8px;'><b>Interpretasi:</b> Menunjukkan daya tahan tabungan Anda terhadap krisis.</div>")
 
-    btn.click(fn=run_app, inputs=id_in, outputs=[out_side, out_report, plot_cf, plot_dist])
+    btn.click(fn=run_app, inputs=id_in, outputs=[out_side, out_report, plot_cf, plot_dist, plot_bal])
 
 demo.launch()