Update app.py

app.py

@@ -6,11 +6,15 @@ from datetime import datetime, timedelta
 import pytz
 
 
-# Function to fetch historical stock data using yfinance
-def fetch_stock_data(ticker, start_date, end_date):
-    stock_data = yf.download(ticker, start=start_date, end=end_date)
+
+
+# Function to fetch stock data
+def fetch_stock_data(ticker, start_datetime, end_datetime):
+    stock_data = yf.download(ticker, start=start_datetime, end=end_datetime)
     return stock_data
 
+
+
 # Function to detect head and shoulder patterns
 def detect_head_shoulder(df, window=3):
     roll_window = window
@@ -204,19 +208,20 @@ def main():
     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'))
 
-    preferred_timezone = st.timezone_input('Select Preferred Timezone', pytz.UTC)
+    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))
 
-    start_datetime = st.time_input('Select Start Time', datetime(2022, 1, 1, 0, 0))
-    end_datetime = st.time_input('Select End Time', datetime(2022, 1, 1, 23, 59))
+    start_datetime = datetime.combine(start_date, start_time)
+    end_datetime = datetime.combine(end_date, end_time)
 
-    start_datetime = datetime.combine(start_date, start_datetime.time())
-    end_datetime = datetime.combine(end_date, end_datetime.time())
+    start_datetime = pytz.timezone(preferred_timezone).localize(start_datetime)
+    end_datetime = pytz.timezone(preferred_timezone).localize(end_datetime)
 
-    start_datetime = preferred_timezone.localize(start_datetime)
-    end_datetime = preferred_timezone.localize(end_datetime)
-    
     if st.button('Detect Patterns'):
-        stock_data = fetch_stock_data(ticker, start_date, end_date)
+        stock_data = fetch_stock_data(ticker, start_datetime, end_datetime)
         stock_data = detect_head_shoulder(stock_data)
         stock_data = detect_multiple_tops_bottoms(stock_data)
         stock_data = calculate_support_resistance(stock_data)
@@ -230,4 +235,5 @@ def main():
         st.write(stock_data)
 
 if __name__ == "__main__":
-    main()
+    main()
+