Update app.py

app.py

@@ -14,12 +14,12 @@ def fetch_stock_data(symbol, start_date, end_date):
 
 # Function to create features for the model
 def create_features(data):
-    data['Date'] = data.index
-    data['Year'] = data['Date'].dt.year
-    data['Month'] = data['Date'].dt.month
-    data['Day'] = data['Date'].dt.day
-    data['Hour'] = data['Date'].dt.hour
-    data['Minute'] = data['Date'].dt.minute
+    data.index = pd.to_datetime(data.index)  # Convert index to datetime
+    data['Year'] = data.index.year
+    data['Month'] = data.index.month
+    data['Day'] = data.index.day
+    data['Hour'] = data.index.hour
+    data['Minute'] = data.index.minute
 
     return data
 
@@ -62,7 +62,7 @@ def main():
         model = train_model(stock_data)
 
         # Predict the stock price for a specific date (e.g., the last date in the dataset)
-        prediction_date = stock_data['Date'].iloc[-1]
+        prediction_date = stock_data.index[-1]
         prediction_features = [[
             prediction_date.year,
             prediction_date.month,