Update app.py

app.py

@@ -3,6 +3,7 @@ import pandas as pd
 from prophet import Prophet
 from datetime import datetime, timedelta
 import numpy as np
+import plotly.graph_objects as go
 
 # Prepare data for Prophet
 def prepare_prophet_data(usage_series):
@@ -13,7 +14,7 @@ def prepare_prophet_data(usage_series):
         'ds': dates,
         'y': usage_series
     })
-    prophet_df['cap'] = 60  # Max observed usage
+    prophet_df['cap'] = 60
     prophet_df['floor'] = 0
     return prophet_df
 
@@ -103,6 +104,57 @@ def main():
         st.write(f"**For 7 Days**: Order {order_7} additional units.")
         st.write(f"**For 14 Days**: Order {order_14} additional units.")
         st.write(f"**For 30 Days**: Order {order_30} additional units.")
+        
+        # Forecast visualization
+        st.header("Forecast Visualization")
+        forecast_data = pd.DataFrame({
+            'Period': ['7 Days', '14 Days', '30 Days'],
+            'Units': [forecast_7, forecast_14, forecast_30]
+        })
+        fig_forecast = go.Figure()
+        fig_forecast.add_trace(go.Scatter(
+            x=forecast_data['Period'],
+            y=forecast_data['Units'],
+            mode='lines+markers',
+            name='Forecasted Units',
+            line=dict(color='blue'),
+            marker=dict(size=10)
+        ))
+        fig_forecast.update_layout(
+            title='Consumable Usage Forecast',
+            xaxis_title='Time Period',
+            yaxis_title='Units',
+            template='plotly_white'
+        )
+        st.plotly_chart(fig_forecast)
+
+        # Threshold alerts bar chart
+        st.header("Threshold Alerts Visualization")
+        alert_data = pd.DataFrame({
+            'Category': ['Current Stock', '7-Day Forecast', '14-Day Forecast', '30-Day Forecast'],
+            'Units': [current_stock, forecast_7, forecast_14, forecast_30],
+            'Alert': [
+                False,
+                current_stock < forecast_7,
+                current_stock < forecast_14,
+                current_stock < forecast_30
+            ]
+        })
+        fig_alerts = go.Figure()
+        fig_alerts.add_trace(go.Bar(
+            x=alert_data['Category'],
+            y=alert_data['Units'],
+            marker_color=['green'] + ['red' if alert else 'blue' for alert in alert_data['Alert'][1:]],
+            text=[f"🚩" if alert else "" for alert in alert_data['Alert']],
+            textposition='auto'
+        ))
+        fig_alerts.update_layout(
+            title='Stock vs Forecast with Alerts (🚩 indicates low stock)',
+            xaxis_title='Category',
+            yaxis_title='Units',
+            template='plotly_white'
+        )
+        st.plotly_chart(fig_alerts)
 
 if __name__ == "__main__":
     main()