Update src/streamlit_app.py

src/streamlit_app.py

@@ -66,7 +66,6 @@ def main():
                 index=0
             )
 
-
         with col2:
             global brokerages
             brokerages = st.number_input(label="Brokerages",step=0.01,value=0.2644 ,format="%.4f")
@@ -106,7 +105,7 @@ def main():
             # 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)
@@ -118,17 +117,21 @@ def main():
     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, data_collection
     
-    HISTORICAL_DATA = data_collection(symbol = symbol, exchange = symbol_exchange)
 
-    # st.divider()
-    if HISTORICAL_DATA is not None:
-        df = HISTORICAL_DATA.copy()
-        try:
-            df = df[['Date','Open','High', 'Low', 'Close']]
-        except:
-            df = df[['Date','Open','High', 'Low', 'Price']]
+    if st.button("Start Analysis"):
+        HISTORICAL_DATA = data_collection(symbol = symbol, exchange = symbol_exchange)
+
 
-        if st.button("Start Analysis"):
+
+        # st.divider()
+        if HISTORICAL_DATA is not None:
+            df = HISTORICAL_DATA.copy()
+            try:
+                df = df[['Date','Open','High', 'Low', 'Close']]
+            except:
+                df = df[['Date','Open','High', 'Low', 'Price']]
+
+        # if st.button("Start Analysis"):
             # Data preprocessing
             df.columns = df.columns.str.strip()
             df['Date'] = parse_date(df['Date'])
@@ -575,6 +578,9 @@ def main():
                 file_name=f"Analysis of {file_name_}.xlsx",
                 mime="application/vnd.openxmlformats-officedocument.spreadsheetml.sheet"
             )
+            
+        else:
+            st.error("No historical data found for the selected symbol and exchange. Please check the inputs and try again.")
 
 if __name__ == "__main__":
     main()