Update app.py

app.py

@@ -113,6 +113,7 @@ if submitted and location:
         system_size_kw = None
         monthly_savings_rs = None
         total_system_cost = None
+        payback_period_years = None
         
         for point in estimated_data:
             if ":" in point:
@@ -129,26 +130,52 @@ if submitted and location:
                         monthly_savings_rs = float(value.split()[0])
                     if "Total system cost" in key:
                         total_system_cost = float(value.split()[0])
+                    if "Payback period" in key:
+                        payback_period_years = float(value.split()[0])
 
                 except ValueError:
                     st.warning("There was an issue processing the response. Please try again.")
 
         # Show Graph if values are available
-        if total_system_cost is not None and monthly_savings_rs is not None:
+        if total_system_cost is not None and monthly_savings_rs is not None and payback_period_years is not None:
             st.subheader("📊 Visual Summary")
-            
-            fig = go.Figure(data=[
-                go.Bar(
-                    x=["Total System Cost (₹)", "Monthly Savings (₹)"],
-                    y=[total_system_cost, monthly_savings_rs],
-                    marker_color=['#636EFA', '#00CC96']
-                )
-            ])
-            fig.update_layout(title="Solar Project Financial Estimates", yaxis_title="Amount (₹)", xaxis_title="Parameters")
+
+            # Prepare data for Area Chart
+            months = int(payback_period_years * 12)
+            savings_cumulative = [monthly_savings_rs * month for month in range(1, months + 1)]
+
+            fig = go.Figure()
+
+            # Area plot
+            fig.add_trace(go.Scatter(
+                x=list(range(1, months + 1)),
+                y=savings_cumulative,
+                mode='lines',
+                fill='tozeroy',
+                name='Cumulative Savings (₹)',
+                line=dict(color='#00CC96')
+            ))
+
+            # Line for Total Cost
+            fig.add_trace(go.Scatter(
+                x=[1, months],
+                y=[total_system_cost, total_system_cost],
+                mode='lines',
+                name='Total System Cost (₹)',
+                line=dict(color='red', dash='dash')
+            ))
+
+            fig.update_layout(
+                title="Cumulative Savings vs Total System Cost Over Time",
+                xaxis_title="Months",
+                yaxis_title="Amount (₹)",
+                legend_title="Legend",
+                template="plotly_white"
+            )
+
             st.plotly_chart(fig, use_container_width=True, key="solar_graph")
 
     else:
         st.error("Sorry, the location entered does not match any available data.")
 else:
     st.info("Please fill out all fields to get your solar project estimate.")
-