Update app.py

app.py

@@ -21,7 +21,7 @@ MODEL_NAME = "databricks/dolly-v2-3b"
 FINBERT_MODEL_NAME = "yiyanghkust/finbert-tone"
 SYMBOL = '^KLSE'  # KLCI index symbol
 START_DATE = '2020-01-01'
-END_DATE = '2024-07-31'
+END_DATE = '2024-10-31'
 
 # Initialize Alpaca API
 api = REST(ALPACA_API_KEY, ALPACA_SECRET_KEY, ALPACA_BASE_URL)
@@ -115,29 +115,37 @@ def plot_klci(data):
 
 # Step 7: Main Function to Run Pipeline
 def main():
-    # 1. Fetch and Prepare Data
-    klci_data = fetch_stock_data(SYMBOL, START_DATE, END_DATE)
+    try:
+        # 1. Fetch and Prepare Data
+        klci_data = fetch_stock_data(SYMBOL, START_DATE, END_DATE)
     
-    # 2. Run FinBERT Sentiment Analysis
-    sample_text = "The market sentiment is bullish for KLCI."  # Example text
-    sentiment = analyze_sentiment(sample_text)
-    print(f"Sentiment: {sentiment}")
+        # 2. Run FinBERT Sentiment Analysis
+        sample_text = "The market sentiment is bullish for KLCI."  # Example text
+        sentiment = analyze_sentiment(sample_text)
+        print(f"Sentiment: {sentiment}")
     
-    # 3. Generate Dolly Prediction
-    prompt = f"The market trend for KLCI with sentiment {sentiment} and indicators {klci_data[['RSI', 'MACD', 'BB_upper', 'BB_middle', 'BB_lower']].iloc[-1].to_dict()}"
-    prediction = generate_prediction(prompt)
-    print(f"Dolly Prediction: {prediction}")
+        # 3. Generate Dolly Prediction
+        indicators = ['RSI', 'MACD', 'BB_upper', 'BB_middle', 'BB_lower']
+        available_columns = [col for col in indicators if col in klci_data.columns]
+        indicator_data = klci_data[available_columns].iloc[-1].to_dict()
+
+        prompt = f"The market trend for KLCI with sentiment {sentiment} and indicators {klci_data[['RSI', 'MACD', 'BB_upper', 'BB_middle', 'BB_lower']].iloc[-1].to_dict()}"
+        prediction = generate_prediction(prompt)
+        print(f"Dolly Prediction: {prediction}")
     
-    # 4. Decide Buy/Sell based on Prediction and Execute Trade
-    if "buy" in prediction.lower():
-        execute_trade("buy", symbol=SYMBOL, qty=1)
-    elif "sell" in prediction.lower():
-        execute_trade("sell", symbol=SYMBOL, qty=1)
-    else:
-        print("No clear trade signal from prediction.")
+        # 4. Decide Buy/Sell based on Prediction and Execute Trade
+        if "buy" in prediction.lower():
+            execute_trade("buy", symbol=SYMBOL, qty=1)
+        elif "sell" in prediction.lower():
+            execute_trade("sell", symbol=SYMBOL, qty=1)
+        else:
+            print("No clear trade signal from prediction.")
     
-    # 5. Plot KLCI Data with Indicators
-    plot_klci(klci_data)
+        # 5. Plot KLCI Data with Indicators
+        plot_klci(klci_data)
+
+    except Exception as e:
+        print (f"An error occured: {e}")
 
 
 # Run the main function