Update app.py

app.py

@@ -9,11 +9,10 @@ from utils import (
     create_price_chart,
     create_technical_chart,
     create_prediction_chart,
-    # Import fungsi baru
     calculate_advanced_risk_metrics 
 )
 import warnings
-import numpy as np # Tambahkan import numpy
+import numpy as np 
 
 warnings.filterwarnings("ignore")
 
@@ -27,21 +26,17 @@ def analyze_stock(symbol, prediction_days=30):
             symbol = symbol.upper() + ".JK"
 
         stock = yf.Ticker(symbol)
-        # Menggunakan periode 1 tahun untuk metrik risiko yang lebih relevan
         data = stock.history(period="1y", interval="1d") 
 
         if data.empty:
             raise ValueError("No price data available for this stock.")
 
-        # Menghitung Indikator Teknikal (Ini juga akan mengisi kolom RSI/MACD ke 'data')
         indicators = calculate_technical_indicators(data)
         signals = generate_trading_signals(data, indicators)
         fundamental_info = get_fundamental_data(stock)
         
-        # Menghitung Metrik Risiko Tingkat Lanjut
         risk_metrics = calculate_advanced_risk_metrics(data.copy())
 
-        # Prediksi Chronos-2 dengan Covariates
         predictions = predict_prices(data, prediction_days=prediction_days)
 
         fig_price = create_price_chart(data, indicators)
@@ -51,19 +46,21 @@ def analyze_stock(symbol, prediction_days=30):
         # kalkulasi TP1, TP2, SL yang diperbarui berdasarkan quantiles/range prediksi
         last_price = data['Close'].iloc[-1]
         
-        # Target/SL yang lebih konservatif berdasarkan range prediksi Q0.1-Q0.9
-        q05 = predictions.get('values', [last_price])
-        q01 = predictions.get('q01', [last_price * 0.95])
-        q09 = predictions.get('q09', [last_price * 1.05])
+        # FIX 3: Dapatkan array prediksi dengan fallback ke array yang berisi harga terakhir
+        q05 = predictions.get('values', np.array([last_price]))
+        q01 = predictions.get('q01', np.array([last_price * 0.95]))
+        q09 = predictions.get('q09', np.array([last_price * 1.05]))
         
-        # TP1 (Target Konservatif): Midpoint antara harga terakhir dan median tertinggi
-        tp1 = (last_price + np.max(q05)) / 2
-        # TP2 (Target Agresif): Quantile 90% tertinggi
-        tp2 = np.max(q09)
-        # SL (Stop Loss): Quantile 10% terendah
-        sl = np.min(q01)
+        # FIX 3: Robust max/min calculation
+        q05_max = np.max(q05) if q05.size > 0 else last_price
+        q09_max = np.max(q09) if q09.size > 0 else last_price * 1.05
+        q01_min = np.min(q01) if q01.size > 0 else last_price * 0.95
+        
+        # Target/SL calculation
+        tp1 = (last_price + q05_max) / 2
+        tp2 = q09_max
+        sl = q01_min
         
-        # Pastikan TP1 < TP2 dan SL lebih rendah dari TP
         if tp1 > tp2:
             tp1, tp2 = tp2, tp1 
         if sl > last_price:
@@ -73,12 +70,10 @@ def analyze_stock(symbol, prediction_days=30):
         predictions["tp2"] = tp2
         predictions["sl"] = sl
 
-        # Menambahkan risk_metrics ke return
         return fundamental_info, indicators, signals, risk_metrics, fig_price, fig_technical, fig_prediction, predictions
 
     except Exception as e:
         print(f"Error analyzing {symbol}: {e}")
-        empty_fig = gr.Plot.update(value=None)
         
         try:
             stock = yf.Ticker(symbol)
@@ -97,8 +92,10 @@ def analyze_stock(symbol, prediction_days=30):
             "q01": [], "q09": [], 
             "tp1": default_tp1, "tp2": default_tp2, "sl": default_sl,
         }
-        # Mengembalikan dictionary kosong untuk risk_metrics saat error
-        return {}, {}, {}, {}, empty_fig, empty_fig, empty_fig, empty_predictions
+        empty_risk = {"error": "Prediction Model Failed to Load/Run. See console for details."}
+        
+        # FIX 4: Mengembalikan None untuk output plot Gradio untuk membersihkan plot
+        return {}, {}, {}, empty_risk, None, None, None, empty_predictions
 
 
 def update_analysis(symbol, prediction_days):
@@ -106,25 +103,27 @@ def update_analysis(symbol, prediction_days):
         fundamental_info,
         indicators,
         signals,
-        risk_metrics, # Tambahan metrik risiko
+        risk_metrics,
         fig_price,
         fig_technical,
         fig_prediction,
         predictions,
     ) = analyze_stock(symbol, prediction_days)
 
-    if not fundamental_info:
-        error_msg = f"Unable to fetch stock data for {symbol.upper()}. Please check the symbol."
-        tp_sl_info = f"<b>TP1:</b> Rp{predictions.get('tp1', 0):,.2f}<br><b>TP2:</b> Rp{predictions.get('tp2', 0):,.2f}<br><b>Stop Loss:</b> Rp{predictions.get('sl', 0):,.2f}<br><br><b>Model Insight:</b><br>Data fetching failed. Cannot proceed with analysis."
+    # FIX 4: Cek apakah ada plot yang None (berarti ada error)
+    if fig_price is None: 
+        error_msg = f"Unable to run AI prediction or fetch data for {symbol.upper()}. Check the model logs for details."
+        tp_sl_info = f"<b>TP1:</b> Rp{predictions.get('tp1', 0):,.2f}<br><b>TP2:</b> Rp{predictions.get('tp2', 0):,.2f}<br><b>Stop Loss:</b> Rp{predictions.get('sl', 0):,.2f}<br><br><b>Model Insight:</b><br>{predictions.get('summary', 'Data fetching or model execution failed. Cannot proceed with analysis.')}"
         
         return (
             f"""<div style="color: red; padding: 10px; border: 1px solid red; border-radius: 5px;">{error_msg}</div><br>{tp_sl_info}""",
-            gr.Plot.update(value=None),
-            gr.Plot.update(value=None),
-            gr.Plot.update(value=None),
+            None, # fig_price
+            None, # fig_technical
+            None, # fig_prediction
         )
 
     # --- FUNDAMENTALS ---
+    # ... (code for fundamentals remains the same)
     fundamentals = f"""
     <h4>COMPANY FUNDAMENTALS</h4>
     <b>Name:</b> {fundamental_info.get('name', 'N/A')} ({symbol.upper()})<br>
@@ -136,6 +135,7 @@ def update_analysis(symbol, prediction_days):
     """
 
     # --- TECHNICAL SIGNAL ---
+    # ... (code for trading_signal remains the same)
     details_list = "".join(
         [f"<li>{line.strip()}</li>" for line in signals.get("details", "").split("\n") if line.strip()]
     )
@@ -154,6 +154,7 @@ def update_analysis(symbol, prediction_days):
     """
     
     # --- RISK METRICS ---
+    # ... (code for risk_report remains the same)
     risk_details = ""
     if "error" in risk_metrics:
         risk_details = f"<b style='color: red;'>{risk_metrics['error']}</b>"
@@ -167,8 +168,12 @@ def update_analysis(symbol, prediction_days):
     """
     
     # --- AI FORECAST ---
-    band_min = float(min(predictions.get('q01', [0]))) if predictions.get('q01') and predictions.get('q01').size > 0 else 0
-    band_max = float(max(predictions.get('q09', [0]))) if predictions.get('q09') and predictions.get('q09').size > 0 else 0
+    # Perlu memastikan q01/q09 tidak kosong
+    q01_values = predictions.get('q01', [])
+    q09_values = predictions.get('q09', [])
+    
+    band_min = float(min(q01_values)) if q01_values and len(q01_values) > 0 else 0
+    band_max = float(max(q09_values)) if q09_values and len(q09_values) > 0 else 0
 
     prediction = f"""
     <h4>{prediction_days}-DAY AI FORECAST (CHRONOS-2 + COVARIATES)</h4>