Update src/streamlit_app.py

src/streamlit_app.py

@@ -5,6 +5,7 @@ import pandas as pd
 import numpy as np
 from io import BytesIO
 import uuid
+from tvDatafeed import TvDatafeed, Interval
 
 from huggingface_hub import snapshot_download
 
@@ -65,6 +66,7 @@ def main():
                 index=0
             )
 
+
         with col2:
             global brokerages
             brokerages = st.number_input(label="Brokerages",step=0.01,value=0.2644 ,format="%.4f")
@@ -91,16 +93,36 @@ def main():
 
         with col3:
             # File uploader
-            uploaded_file = st.file_uploader("Upload your CSV file", type=["csv"])
+
+            if Exchange == 'Indian':
+                symbol = ['NIFTY', 'BANKNIFTY', 'FINNIFTY', 'NIFTYIT', 'NIFTYFMCG', 'NIFTYMETAL', 'NIFTYPSU', 'NIFTYAUTO', 'NIFTYMEDIA', 'NIFTYPVTBANK', 'NIFTYREALTY', 'NIFTYCONSUMER', 'NIFTYENERGY', 
+                          'NIFTYHEALTHCARE', 'NIFTYINFRA', 'NIFTYPHARMA', 'RELIANCE', 'TATAMOTORS', 'HDFCBANK', 'ICICIBANK', 'INFY', 'HINDUNILVR', 'LT', 'TCS', 'HDFC', 'KOTAKBANK', 'AXISBANK', 'ITC', 'SBIN', 'MARUTI', 'BAJFINANCE']
+                exchange_list = ['NSE', 'BSE', 'MCX', 'NFO', 'CDS']
+            else:
+                symbol = ['ETHUSDT', 'BTCUSDT', 'BNBUSDT', 'XRPUSDT', 'SOLUSDT', 'DOGEUSDT', 'ADAUSDT', 'MATICUSDT', 'DOTUSDT', 'TRXUSDT']
+                exchange_list = ['BINANCE', 'BITFINEX', 'COINBASE', 'BITSTAMP', 'OKEX', 'BYBIT', 'GEMINI']
+
+            ## Add search functionality in selectbox
+            # symbol = st.selectbox("Select Symbol", symbol, index=0, key="symbol_select")
+            symbol = st.selectbox("Select Symbol", symbol, index=0, key="symbol_select", help="Search for a symbol by typing in the box")
+            symbol_exchange = st.selectbox("Select Exchange", exchange_list, index=0, key="exchange_select", help="Search for an exchange by typing in the box")
+            
+
+            # symbol = st.selectbox("Select Symbol", symbol, index=0)
+            # symbol_exchange = st.selectbox("Select Exchange", exchange_list, index=0)
+            
+            # uploaded_file = st.file_uploader("Upload your CSV file", type=["csv"])
 
     if selected_script == "Old":
-        from main4 import get_buy_signal, parse_date, get_sell_signal, year_wise_analysis, calculate_yearly_returns, calculate_mtf_returns, crypto_year_wise_analysis, crypto_calculate_mtf_returns, crypto_calculate_yearly_returns, find_sequences, generate_trade_signals
+        from main4 import get_buy_signal, parse_date, get_sell_signal, year_wise_analysis, calculate_yearly_returns, calculate_mtf_returns, crypto_year_wise_analysis, crypto_calculate_mtf_returns, crypto_calculate_yearly_returns, find_sequences, generate_trade_signals, data_collection
     else:
-        from main6 import get_buy_signal, parse_date, get_sell_signal, year_wise_analysis, calculate_yearly_returns, calculate_mtf_returns, crypto_year_wise_analysis, crypto_calculate_mtf_returns, crypto_calculate_yearly_returns, find_sequences, generate_trade_signals
+        from main6 import get_buy_signal, parse_date, get_sell_signal, year_wise_analysis, calculate_yearly_returns, calculate_mtf_returns, crypto_year_wise_analysis, crypto_calculate_mtf_returns, crypto_calculate_yearly_returns, find_sequences, generate_trade_signals, data_collection
     
+    HISTORICAL_DATA = data_collection(symbol = symbol, exchange = symbol_exchange)
+
     # st.divider()
-    if uploaded_file is not None:
-        df = pd.read_csv(uploaded_file)
+    if HISTORICAL_DATA is not None:
+        df = HISTORICAL_DATA.copy()
         try:
             df = df[['Date','Open','High', 'Low', 'Close']]
         except:
@@ -544,7 +566,7 @@ def main():
                 buy_trades_mtf_yearly_compound.to_excel(writer, sheet_name='BUY MTF CMP Yearly', index=False)
 
 
-            file_name_ = f"{uploaded_file.name} Trailing {trail_offset} Target {target_sl}"
+            file_name_ = f"{symbol} Trailing {trail_offset} Target {target_sl}"
 
             excel_buffer.seek(0)
             st.download_button(