Update app.py

app.py

@@ -1096,17 +1096,8 @@ for company in companies:
     # Strategy B: Regression (only Tesla & Microsoft)
     # df_c['StrategyB_Daily'] = df_c['Predicted']*df_c['PctChangeDaily']*df_c[f'Close_{TICKERS[company]}']
     # df_c['StrategyB_Cumulative'] = df_c['StrategyB_Daily'].cumsum()
-
-    # --- CLEANUP & KEEP ONLY NEEDED COLUMNS ---
-    keep_cols = [
-        "Date", "Company", "Title", "Summary", 
-        "Sentiment", "Confidence", 
-        f"Close_{TICKERS[company]}", "PctChangeDaily", 
-        "StrategyA_Daily", "StrategyA_Cumulative"
-    ]
-    
+    df_c = df_c.drop(columns=["date", "date_merge"], errors="ignore")
     
-    df_c = df_c[keep_cols]
 
     dfs_final[company] = df_c