Update app.py

app.py

@@ -19,15 +19,21 @@ def get_data():
         response = requests.get('https://api.coinex.com/v1/market/kline/BTCUSDT', params=params)
         data = response.json()
         
-        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
+        if 'data' in data:
+            keys = data['data'][0].keys()
+            st.write(keys)  # Print out the keys in the first data point
+            
+            try:
+                df = pd.DataFrame(data['data'])
+                # Adjust the key based on the actual keys in the data dictionary
+                timestamp_key = 'time' if 'time' in keys else 'open_time'
+                df[timestamp_key] = pd.to_datetime(df[timestamp_key], unit='ms')
+                df.set_index(timestamp_key, 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
@@ -39,4 +45,4 @@ if st.button("Fetch historical data"):
     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")
+        st.write("Data fetched successfully and saved to btcusdt_data.pkl")