Update app.py

app.py

@@ -75,13 +75,13 @@ def plot_charts(data, symbol, interval):
     
     # Determine the number of subplots based on the interval
     if interval in minute_intervals:
-        rows = 3
-        subplot_titles = ('Price & Volume', 'RSI', 'MACD')
-        row_heights = [0.5, 0.25, 0.25]
+        rows = 4  # Added an extra row for OBV
+        subplot_titles = ('Price & Volume', 'RSI', 'MACD', 'On-Balance Volume (OBV)')
+        row_heights = [0.35, 0.15, 0.15, 0.35]
     else:
-        rows = 4
-        subplot_titles = ('Price & Volume', 'RSI', 'MACD', 'Technical Indicators')
-        row_heights = [0.4, 0.2, 0.2, 0.2]
+        rows = 5  # Added two extra rows for OBV and Bollinger Bands
+        subplot_titles = ('Price & Volume', 'RSI', 'MACD', 'Bollinger Bands', 'On-Balance Volume (OBV)')
+        row_heights = [0.3, 0.15, 0.15, 0.2, 0.2]
     
     fig = make_subplots(rows=rows, cols=1, 
                         shared_xaxes=True,
@@ -111,6 +111,16 @@ def plot_charts(data, symbol, interval):
         colors = ['red' if row['Open'] - row['Close'] >= 0 else 'green' for index, row in data.iterrows()]
         fig.add_trace(go.Bar(x=data.index, y=data['Volume'], name='Volume', marker_color=colors), row=1, col=1)
     
+    # Bollinger Bands
+    if rows == 5:
+        if 'BB_upper' in data.columns and 'BB_middle' in data.columns and 'BB_lower' in data.columns:
+            fig.add_trace(go.Scatter(x=data.index, y=data['BB_upper'], name='BB Upper',
+                                     line=dict(color='gray', dash='dash')), row=4, col=1)
+            fig.add_trace(go.Scatter(x=data.index, y=data['BB_middle'], name='BB Middle',
+                                     line=dict(color='gray')), row=4, col=1)
+            fig.add_trace(go.Scatter(x=data.index, y=data['BB_lower'], name='BB Lower',
+                                     line=dict(color='gray', dash='dash')), row=4, col=1)
+    
     # RSI
     if 'RSI' in data.columns:
         fig.add_trace(go.Scatter(x=data.index, y=data['RSI'], name='RSI', line=dict(color='purple')), row=2, col=1)
@@ -121,21 +131,15 @@ def plot_charts(data, symbol, interval):
     if 'MACD' in data.columns:
         fig.add_trace(go.Scatter(x=data.index, y=data['MACD'], name='MACD', line=dict(color='blue')), row=3, col=1)
     
-    if rows == 4:
-        # Bollinger Bands
-        if 'BB_upper' in data.columns and 'BB_middle' in data.columns and 'BB_lower' in data.columns:
-            fig.add_trace(go.Scatter(x=data.index, y=data['BB_upper'], name='BB Upper',
-                                     line=dict(color='gray', dash='dash')), row=4, col=1)
-            fig.add_trace(go.Scatter(x=data.index, y=data['BB_middle'], name='BB Middle',
-                                     line=dict(color='gray')), row=4, col=1)
-            fig.add_trace(go.Scatter(x=data.index, y=data['BB_lower'], name='BB Lower',
-                                     line=dict(color='gray', dash='dash')), row=4, col=1)
+    # On-Balance Volume (OBV)
+    if 'OBV' in data.columns:
+        fig.add_trace(go.Scatter(x=data.index, y=data['OBV'], name='OBV', line=dict(color='brown')), row=rows, col=1)
     
     # Update layout
     fig.update_layout(
         title=f'{symbol} Technical Analysis Dashboard',
         yaxis_title='Price',
-        height=800 if rows == 4 else 600,
+        height=1000 if rows == 5 else 800,
         showlegend=True,
         xaxis_rangeslider_visible=False
     )
@@ -200,7 +204,7 @@ def main():
                 else:
                     daily_change = 0.0  # Not enough data to calculate change
                 
-                col1, col2, col3, col4 = st.columns(4)
+                col1, col2, col3, col4, col5 = st.columns(5)
                 col1.metric("Current Price", f"₹{current_price:.2f}", f"{daily_change:.2f}%")
                 
                 if '52W_High' in data.columns and '52W_Low' in data.columns:
@@ -210,6 +214,9 @@ def main():
                 if 'RSI' in data.columns:
                     col4.metric("RSI", f"{data['RSI'][-1]:.2f}")
                 
+                if 'OBV' in data.columns:
+                    col5.metric("OBV", f"{data['OBV'][-1]:.2f}")
+                
                 # Plot interactive charts
                 fig = plot_charts(data, symbol, interval)
                 st.plotly_chart(fig, use_container_width=True)
@@ -219,6 +226,8 @@ def main():
                 
                 # Simple trading signals
                 signals = []
+                
+                # RSI Signals
                 if 'RSI' in data.columns:
                     rsi = data['RSI'][-1]
                     if rsi < 30:
@@ -226,6 +235,7 @@ def main():
                     elif rsi > 70:
                         signals.append("RSI indicates overbought conditions")
                 
+                # MACD Signals
                 if 'MACD' in data.columns:
                     macd = data['MACD'][-1]
                     if macd > 0:
@@ -233,6 +243,7 @@ def main():
                     else:
                         signals.append("MACD is negative - Bearish momentum")
                 
+                # SMA Signals
                 if 'SMA_20' in data.columns and 'SMA_50' in data.columns:
                     current_price = data['Close'][-1]
                     sma_20 = data['SMA_20'][-1]
@@ -243,6 +254,25 @@ def main():
                     elif current_price < sma_20 and current_price < sma_50:
                         signals.append("Price is below major moving averages - Bearish")
                 
+                # OBV Signals
+                if 'OBV' in data.columns:
+                    obv = data['OBV'][-1]
+                    obv_prev = data['OBV'][-2] if len(data) >= 2 else obv
+                    if obv > obv_prev:
+                        signals.append("OBV is increasing - Positive volume trend")
+                    elif obv < obv_prev:
+                        signals.append("OBV is decreasing - Negative volume trend")
+                
+                # Bollinger Bands Signals
+                if rows == 5:
+                    price = data['Close'][-1]
+                    bb_upper = data['BB_upper'][-1]
+                    bb_lower = data['BB_lower'][-1]
+                    if price > bb_upper:
+                        signals.append("Price has breached the upper Bollinger Band - Potential overbought condition")
+                    elif price < bb_lower:
+                        signals.append("Price has breached the lower Bollinger Band - Potential oversold condition")
+                
                 for signal in signals:
                     st.write(f"• {signal}")