Update app.py

app.py

@@ -13,7 +13,14 @@ def fetch_stock_data(ticker, start_datetime, end_datetime):
     stock_data = yf.download(ticker, start=start_datetime, end=end_datetime)
     return stock_data
 
+def fetch_stock_data(ticker, start_datetime, end_datetime):
+    # Fetch live data using yfinance
+    stock_data = yf.download(ticker, start=start_datetime, end=end_datetime)
+
+    # Convert DatetimeIndex to a column
+    df.reset_index(inplace=True)
 
+    return stock_data
 
 # Function to detect head and shoulder patterns
 def detect_head_shoulder(df, window=3):
@@ -203,23 +210,32 @@ def find_pivots(df):
 
 # Streamlit App
 def main():
-    st.title('Stock Pattern Detection App')
+    st.title('Live Stock Pattern Detection App')
     ticker = st.text_input('Enter Stock Ticker:', 'AAPL')
-    start_date = st.date_input('Start Date', pd.to_datetime('2020-01-01'))
-    end_date = st.date_input('End Date', pd.to_datetime('2022-01-01'))
+    
+    # Placeholder for start and end date
+    start_date_placeholder = st.empty()
+    end_date_placeholder = st.empty()
+
+    start_date = start_date_placeholder.date_input('Start Date', pd.to_datetime('2022-01-01'))
+    end_date = end_date_placeholder.date_input('End Date', pd.to_datetime('2022-02-01'))
 
     st.info("Select Preferred Timezone:")
     preferred_timezone = st.selectbox('Timezone', list(pytz.all_timezones))
 
-    start_time = st.time_input('Select Start Time', datetime(2022, 1, 1, 0, 0))
-    end_time = st.time_input('Select End Time', datetime(2022, 1, 1, 23, 59))
+    # Time selection components
+    start_hour, start_minute, start_second = st.slider('Select Start Time', 0, 23, 0), st.slider('', 0, 59, 0), st.slider('', 0, 59, 0)
+    end_hour, end_minute, end_second = st.slider('Select End Time', 0, 23, 23), st.slider('', 0, 59, 59), st.slider('', 0, 59, 59)
 
-    start_datetime = datetime.combine(start_date, start_time)
-    end_datetime = datetime.combine(end_date, end_time)
+    start_datetime = datetime(start_date.year, start_date.month, start_date.day, start_hour, start_minute, start_second)
+    end_datetime = datetime(end_date.year, end_date.month, end_date.day, end_hour, end_minute, end_second)
 
     start_datetime = pytz.timezone(preferred_timezone).localize(start_datetime)
     end_datetime = pytz.timezone(preferred_timezone).localize(end_datetime)
 
+    start_date_placeholder.info(f"Start Date (UTC): {start_datetime.strftime('%Y-%m-%d %H:%M:%S %Z')}")
+    end_date_placeholder.info(f"End Date (UTC): {end_datetime.strftime('%Y-%m-%d %H:%M:%S %Z')}")
+
     if st.button('Detect Patterns'):
         stock_data = fetch_stock_data(ticker, start_datetime, end_datetime)
         stock_data = detect_head_shoulder(stock_data)