Update app.py

app.py

@@ -11,8 +11,6 @@ from sklearn.linear_model import LinearRegression
 import plotly.graph_objects as go
 from plotly.subplots import make_subplots
 import yfinance as yf
-
-# 🔥 เพิ่ม FinBERT
 import torch
 from transformers import AutoTokenizer, AutoModelForSequenceClassification
 
@@ -312,19 +310,43 @@ def main():
         fig.add_trace(
             go.Scatter(
                 x=future_dates, y=future_preds,
-                name="Predicted Sentiment", mode="lines+markers", line=dict(dash="dash")
+                name="Predicted Sentiment", mode="lines+markers", line=dict(color="#02a1f7", dash="dash")
             ),
             row=1, col=1, secondary_y=True
         )
 
+    # ---------------------------------------------------------
+    # เส้นเชื่อม Actual -> Predicted
+    # ---------------------------------------------------------
+    last_actual_date = plot_data["date_day"].max()
+    last_actual_value = plot_data["avg_sentiment"].iloc[-1]
+    first_pred_date = future_dates[0]
+    first_pred_value = future_preds[0]
+
+    fig.add_trace(
+        go.Scatter(
+            x=[last_actual_date, first_pred_date],
+            y=[last_actual_value, first_pred_value],
+            mode="lines",
+            line=dict(color="purple", dash="dot"),
+            name="Connector Actual→Predicted"
+        ),
+        row=1, col=1, secondary_y=True
+    )
+
+    
     # จำนวนข่าว
     for col in ["neutral", "negative", "positive"]:
         if col not in plot_data.columns:
             plot_data[col] = 0
 
-    fig.add_trace(go.Bar(x=plot_data["date_day"], y=plot_data["neutral"], name="Neutral"), row=2, col=1)
-    fig.add_trace(go.Bar(x=plot_data["date_day"], y=plot_data["negative"], name="Negative"), row=2, col=1)
-    fig.add_trace(go.Bar(x=plot_data["date_day"], y=plot_data["positive"], name="Positive"), row=2, col=1)
+    fig.add_trace(go.Bar(x=plot_data["date_day"], y=plot_data["neutral"], name="Neutral",
+                     marker_color='rgba(128, 128, 128, 0.7)'), row=2, col=1)
+    fig.add_trace(go.Bar(x=plot_data["date_day"], y=plot_data["negative"], name="Negative",
+                     marker_color='rgba(255, 0, 0, 0.7)'), row=2, col=1)
+    fig.add_trace(go.Bar(x=plot_data["date_day"], y=plot_data["positive"], name="Positive",
+                     marker_color='rgba(0, 128, 0, 0.7)'), row=2, col=1)
+
 
     fig.update_layout(
         title=f"แนวโน้มอารมณ์ข่าว + ราคาหุ้น ({symbol})",