Update app.py

app.py

@@ -18,22 +18,25 @@ def get_data():
         }
         response = requests.get('https://api.coinex.com/v1/market/kline/BTCUSDT', params=params)
         data = response.json()
-        df = pd.DataFrame(data['data'])
-        df['open_time'] = pd.to_datetime(df['open_time'], unit='ms')  # Replace 'date' with 'open_time'
-        df.set_index('open_time', inplace=True)
-        df.drop(['vol', 'amount'], axis=1, inplace=True)
-        df_list.append(df)
+        
+        try:
+            df = pd.DataFrame(data['data'])
+            df['time'] = pd.to_datetime(df['time'], unit='ms')  # Assuming the timestamp key is 'time'
+            df.set_index('time', inplace=True)
+            df.drop(['vol', 'amount'], axis=1, inplace=True)
+            df_list.append(df)
+        except KeyError as e:
+            st.error(f"KeyError: {e}. Please check the keys in the data returned by the API.")
+            return None
 
     df = pd.concat(df_list)
     return df
 
-
-    df = pd.concat(df_list)
-    return df
-
-if st.button("sosis"):
-# Retrieve historical data from Coinex API
+if st.button("Fetch historical data"):
+    # Retrieve historical data from Coinex API
     df = get_data()
 
-# Save historical data to btcusdt_data.pkl
-    df.to_pickle('btcusdt_data.pkl')
+    if df is not None:
+        # Save historical data to btcusdt_data.pkl
+        df.to_pickle('btcusdt_data.pkl')
+        st.write("Data fetched successfully and saved to btcusdt_data.pkl")