Update app.py

app.py

@@ -68,7 +68,6 @@ def n_weeks_before(date_string, n):
 
 
 def get_stock_data(stock_symbol, steps):
-
     stock_data = yf.download(stock_symbol, steps[0], steps[-1])
     if len(stock_data) == 0:
         raise gr.Error(f"Failed to download stock price data for symbol {stock_symbol} from yfinance!")
@@ -81,16 +80,19 @@ def get_stock_data(stock_symbol, steps):
     for date in steps[:-1]:
         for i in range(len(stock_data)):
             if available_dates[i] >= date:
-                prices.append(stock_data['Close'][i])
+                prices.append(stock_data['Close'].iloc[i])  # Use .iloc here
                 dates.append(datetime.strptime(available_dates[i], "%Y-%m-%d"))
                 break
 
+    # Append the last date and price
     dates.append(datetime.strptime(available_dates[-1], "%Y-%m-%d"))
-    prices.append(stock_data['Close'][-1])
+    prices.append(stock_data['Close'].iloc[-1])  # Use .iloc here as well
     
     return pd.DataFrame({
-        "Start Date": dates[:-1], "End Date": dates[1:],
-        "Start Price": prices[:-1], "End Price": prices[1:]
+        "Start Date": dates[:-1],
+        "End Date": dates[1:],
+        "Start Price": prices[:-1],
+        "End Price": prices[1:]
     })