Update app.py

app.py

@@ -30,16 +30,17 @@ def calculate_volume_profile(data, row_layout):
     price_max = data['High'].max()
 
     bins = row_layout
-    bin_edges = np.linspace(price_min, price_max, bins)
+    bin_edges = np.linspace(price_min, price_max, bins).flatten()
 
-    volume_profile = pd.DataFrame(index=bin_edges[:-1], columns=['Total Volume'])
-    volume_profile['Total Volume'] = 0
+    # Create a DataFrame for volume profile
+    volume_profile = pd.DataFrame(0, index=bin_edges[:-1], columns=['Total Volume'])
 
     for index, row in data.iterrows():
         bin_indices = np.digitize([row['Low'], row['High']], bin_edges) - 1
         bin_indices = [max(0, min(bins-2, b)) for b in bin_indices]
 
-        volume_profile.iloc[bin_indices[0]:bin_indices[1] + 1, volume_profile.columns.get_loc('Total Volume')] += row['Volume']
+        # Update the volume for the respective bin ranges
+        volume_profile.iloc[bin_indices[0]:bin_indices[1] + 1, 0] += row['Volume']
 
     return volume_profile