Update sentiment_app.py

sentiment_app.py

@@ -5,9 +5,9 @@ import pandas as pd
 import matplotlib.pyplot as plt
 import seaborn as sns
 from datetime import datetime
-import collections
 
 # --- SETUP MODEL ---
+# Menggunakan model RoBERTa Bahasa Indonesia untuk analisis sentimen
 MODEL_NAME = "w11wo/indonesian-roberta-base-sentiment-classifier"
 device = 0 if torch.cuda.is_available() else -1
 sentiment_pipeline = pipeline("sentiment-analysis", model=MODEL_NAME, device=device)
@@ -15,7 +15,7 @@ sentiment_pipeline = pipeline("sentiment-analysis", model=MODEL_NAME, device=dev
 # --- DATABASE SEDERHANA (In-Memory) ---
 all_messages = [] 
 
-# Mapping Label untuk Tampilan UI
+# Mapping Label untuk Tampilan UI agar lebih mudah dipahami manusia
 label_map = {
     "POSITIVE": "Pujian/Apresiasi", 
     "NEGATIVE": "Keluhan/Kritik", 
@@ -26,13 +26,13 @@ def process_submission(text):
     if not text or text.strip() == "":
         return "⚠️ Mohon isi komentar Anda terlebih dahulu.", gr.update()
     
-    # 1. Analisis Sentimen
+    # 1. Analisis Sentimen menggunakan Pipeline Hugging Face
     result = sentiment_pipeline(text)[0]
     label = result['label'].upper()
     
-    # 2. Simpan ke Database Lokal
+    # 2. Simpan ke Database Lokal dengan Timestamp
     new_entry = {
-        "Waktu": datetime.now().strftime("%Y-%m-%d %H:%M"),
+        "Waktu": datetime.now().strftime("%Y-%m-%d %H:%M:%S"),
         "Pesan": text.strip(),
         "Sentimen": label
     }
@@ -44,13 +44,13 @@ def process_submission(text):
 
 def get_admin_dashboard(filter_val):
     if not all_messages:
-        return None, pd.DataFrame(columns=["Pesan", "Sentimen", "Jumlah"]), "Belum ada data."
+        return None, pd.DataFrame(columns=["Waktu", "Pesan", "Sentimen"]), "Belum ada data."
     
     df_all = pd.DataFrame(all_messages)
     
     # --- LOGIKA FILTER ---
     if filter_val != "SEMUA":
-        # Balik mapping untuk filter data asli
+        # Balik mapping untuk mencari label asli (POSITIVE/NEGATIVE/NEUTRAL)
         rev_map = {v: k for k, v in label_map.items()}
         target = rev_map.get(filter_val)
         df_filtered = df_all[df_all['Sentimen'] == target]
@@ -58,37 +58,44 @@ def get_admin_dashboard(filter_val):
         df_filtered = df_all
 
     if df_filtered.empty:
-        return None, pd.DataFrame(columns=["Pesan", "Sentimen", "Jumlah"]), f"Tidak ada data untuk kategori: {filter_val}"
+        return None, pd.DataFrame(columns=["Waktu", "Pesan", "Sentimen"]), f"Tidak ada data untuk kategori: {filter_val}"
 
-    # --- VISUALISASI TOTAL ---
+    # --- VISUALISASI TOTAL (Pie Chart atau Bar Plot) ---
     fig, ax = plt.subplots(figsize=(8, 5))
-    color_map = {"POSITIVE": "#4CAF50", "NEGATIVE": "#F44336", "NEUTRAL": "#FFC107"}
     counts = df_all['Sentimen'].value_counts()
+    # Mengubah index angka/label asli ke label buatan kita (Pujian/Keluhan/dll)
     counts.index = [label_map.get(i, i) for i in counts.index]
     
     sns.barplot(x=counts.index, y=counts.values, palette="viridis", ax=ax)
     ax.set_title("Proporsi Pesan Masuk (Total)", fontsize=12, fontweight='bold')
-    
-    # --- TABEL TOP 10 DENGAN KOLOM SENTIMEN ---
-    top_df = df_filtered.groupby(['Pesan', 'Sentimen']).size().reset_index(name='Jumlah')
-    top_df = top_df.sort_values(by='Jumlah', ascending=False).head(10)
-    top_df['Sentimen'] = top_df['Sentimen'].map(label_map) # Percantik label di tabel
-    
-    return fig, top_df, f"Menampilkan {len(df_filtered)} pesan ({filter_val})"
+    ax.set_ylabel("Jumlah Pesan")
+
+    # --- TABEL DENGAN KOLOM WAKTU/TANGGAL ---
+    display_df = df_filtered[["Waktu", "Pesan", "Sentimen"]].copy()
+    display_df['Sentimen'] = display_df['Sentimen'].map(label_map) # Percantik label di tabel
+    display_df = display_df.sort_values(by="Waktu", ascending=False) # Urutkan: Terbaru di atas
+
+    return fig, display_df, f"Menampilkan {len(df_filtered)} pesan ({filter_val})"
 
 # --- INTERFACE GRADIO ---
 with gr.Blocks(theme=gr.themes.Soft(primary_hue="emerald"), title="PoskoLog Dashboard") as demo:
     gr.Markdown("# 📦 PoskoLog: Suara Pengungsi")
+    gr.Markdown("Sistem analisis sentimen otomatis untuk memprioritaskan laporan darurat pasca-bencana.")
     
     with gr.Tabs():
-        # TAB USER
+        # --- TAB USER ---
         with gr.Tab("📝 Sampaikan Pesan"):
             with gr.Column(variant="panel"):
-                user_input = gr.Textbox(label="Komentar Anda", placeholder="Tulis masukan di sini...", lines=4)
+                gr.Markdown("### Laporkan kondisi atau berikan masukan Anda")
+                user_input = gr.Textbox(
+                    label="Komentar Anda", 
+                    placeholder="Contoh: Bantuan air bersih belum sampai di tenda C...", 
+                    lines=4
+                )
                 submit_btn = gr.Button("Kirim Pesan", variant="primary")
                 user_feedback = gr.Markdown("")
 
-        # TAB ADMIN
+        # --- TAB ADMIN ---
         with gr.Tab("📊 Dashboard Admin"):
             with gr.Row():
                 sentiment_filter = gr.Dropdown(
@@ -96,20 +103,36 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="emerald"), title="PoskoLog Dash
                     value="SEMUA", 
                     label="Filter Sentimen"
                 )
-                refresh_btn = gr.Button("🔄 Refresh & Filter", variant="secondary")
+                refresh_btn = gr.Button("🔄 Refresh & Filter Data", variant="secondary")
             
             with gr.Row():
-                with gr.Column():
+                with gr.Column(scale=1):
                     plot_output = gr.Plot(label="Grafik Distribusi")
-                with gr.Column():
-                    gr.Markdown("#### Pesan Berdasarkan Filter")
-                    table_output = gr.Dataframe(headers=["Pesan", "Sentimen", "Jumlah"], interactive=False)
+                with gr.Column(scale=2):
+                    gr.Markdown("#### Daftar Laporan Masuk")
+                    # Tabel diperbarui dengan kolom Waktu
+                    table_output = gr.Dataframe(
+                        headers=["Waktu", "Pesan", "Sentimen"], 
+                        interactive=False,
+                        wrap=True
+                    )
             
-            status_txt = gr.Markdown("")
+            status_txt = gr.Markdown("Klik 'Refresh' untuk memuat data terbaru.")
 
-    # Binding Events
-    submit_btn.click(fn=process_submission, inputs=user_input, outputs=[user_feedback, user_input])
-    refresh_btn.click(fn=get_admin_dashboard, inputs=sentiment_filter, outputs=[plot_output, table_output, status_txt])
+    # --- BINDING EVENTS ---
+    # Saat klik kirim: proses teks, beri feedback, dan kosongkan textbox
+    submit_btn.click(
+        fn=process_submission, 
+        inputs=user_input, 
+        outputs=[user_feedback, user_input]
+    )
+    
+    # Saat klik refresh: update grafik dan tabel berdasarkan filter
+    refresh_btn.click(
+        fn=get_admin_dashboard, 
+        inputs=sentiment_filter, 
+        outputs=[plot_output, table_output, status_txt]
+    )
 
 if __name__ == "__main__":
     demo.launch(server_name="0.0.0.0", server_port=7860)