Update app.py

app.py

@@ -1,35 +1,96 @@
 import streamlit as st
-from utils.preprocessing import clean_data
-from utils.models import predict_umkm
+import pandas as pd
 import matplotlib.pyplot as plt
+from utils.models import predict_umkm
+from utils.preprocessing import clean_data
 
-# Judul
+# Konfigurasi tampilan
+st.set_page_config(page_title="AI Supply Chain UMKM", layout="wide")
 st.title("🚀 AI Supply Chain UMKM")
 
+# Sidebar untuk parameter
+with st.sidebar:
+    st.header("Pengaturan Model")
+    prediction_days = st.slider("Jumlah Hari Prediksi", 1, 30, 7)
+    confidence_level = st.slider("Tingkat Kepercayaan (%)", 70, 95, 85)
+
 # Upload data
-uploaded_file = st.file_uploader("Upload CSV", type="csv")
+uploaded_file = st.file_uploader(
+    "Upload Data CSV (Format: tanggal, demand, supply)", 
+    type="csv"
+)
+
 if uploaded_file:
-    df = clean_data(uploaded_file)
-    
-    # Grafik supply-demand
-    fig, ax = plt.subplots()
-    ax.plot(df['tanggal'], df['demand'], label='Demand', color='blue')
-    ax.plot(df['tanggal'], df['supply'], label='Supply', color='red')
-    ax.fill_between(df['tanggal'], df['demand'], df['supply'], 
-                   where=(df['supply'] > df['demand']), 
-                   color='green', alpha=0.3, label='Surplus')
-    ax.legend()
-    st.pyplot(fig)
-    
-    # Rekomendasi AI
-    st.subheader("Rekomendasi")
-    recommendation = predict_umkm(df)
-    st.success(recommendation)
-    
-    # Parameter ROI
-    st.subheader("Parameter Keputusan")
-    st.json({
-        "ROI Estimasi": "15%",
-        "Anomali Terdeteksi": 0,
-        "Selisih (Supply - Demand)": 20
-    })
+    try:
+        # Preprocessing data
+        df = clean_data(uploaded_file)
+        
+        # Pastikan kolom ada
+        if not all(col in df.columns for col in ['tanggal', 'demand', 'supply']):
+            raise ValueError("Format kolom harus: tanggal, demand, supply")
+
+        # ===== 1. Tampilkan Grafik =====
+        st.subheader("Grafik Supply vs Demand")
+        fig, ax = plt.subplots(figsize=(12, 6))
+        
+        ax.plot(df['tanggal'], df['demand'], label='Demand', color='blue', marker='o')
+        ax.plot(df['tanggal'], df['supply'], label='Supply', color='red', marker='s')
+        
+        ax.fill_between(
+            df['tanggal'], 
+            df['demand'], 
+            df['supply'], 
+            where=(df['supply'] > df['demand']),
+            color='green', alpha=0.3, label='Surplus'
+        )
+        ax.fill_between(
+            df['tanggal'], 
+            df['demand'], 
+            df['supply'], 
+            where=(df['supply'] < df['demand']),
+            color='red', alpha=0.3, label='Defisit'
+        )
+        
+        ax.set_xlabel("Tanggal")
+        ax.set_ylabel("Jumlah")
+        ax.legend()
+        ax.grid(True)
+        st.pyplot(fig)
+
+        # ===== 2. Prediksi dengan AI =====
+        st.subheader("Rekomendasi AI")
+        
+        with st.spinner("Menganalisis data..."):
+            # Prediksi menggunakan GRANITE-TTM + Chronos-T5
+            recommendation = predict_umkm(
+                df,
+                prediction_length=prediction_days,
+                confidence=confidence_level/100
+            )
+            
+            # Tampilkan hasil
+            st.success(recommendation['text'])
+            
+            # Detail teknis
+            with st.expander("Detail Prediksi"):
+                st.json({
+                    "prediksi_demand": recommendation['predictions'],
+                    "roi_estimasi": f"{recommendation['roi']*100:.1f}%",
+                    "confidence_score": recommendation['confidence']
+                })
+
+    except Exception as e:
+        st.error(f"⚠️ Error: {str(e)}")
+        st.info("Pastikan data berformat:\n- Kolom: tanggal, demand, supply\n- Minimal 30 data historis")
+
+# Contoh data default jika tidak ada upload
+else:
+    st.info("""
+    **Contoh Format CSV:**
+    ```
+    tanggal,demand,supply
+    2024-01-01,100,120
+    2024-01-02,150,130
+    2024-01-03,200,180
+    ```
+    """)