Update app.py

app.py

@@ -3,6 +3,7 @@ import pandas as pd
 import google.generativeai as genai
 import os
 from dotenv import load_dotenv
+import plotly.graph_objects as go
 
 # Load environment variables from .env file
 load_dotenv()
@@ -85,6 +86,40 @@ if submitted and location and roof_size > 0 and electricity_bill >= 0:
             if ":" in point:
                 key, value = point.split(":", 1)
                 st.write(f"**{key.strip()}**: {value.strip()}")
+        
+            # Variables to store extracted values for graph
+            system_size_kw = None
+            monthly_savings_rs = None
+
+            for point in estimated_data:
+                if ":" in point:
+                    try:
+                        key, value = point.split(":", 1)
+                        key = key.strip()
+                        value = value.strip()
+
+                        st.write(f"**{key}**: {value}")
+
+                        if "Estimated solar system size" in key:
+                            system_size_kw = float(value.split()[0])
+                        if "Monthly savings" in key:
+                            monthly_savings_rs = float(value.split()[0])
+                    except ValueError:
+                        st.warning("There was an issue processing the response.")
+
+            # Show Graph if available
+            if system_size_kw is not None and monthly_savings_rs is not None:
+                st.subheader("📊 Visual Summary")
+                
+                fig = go.Figure(data=[
+                    go.Bar(name='Estimate',
+                           x=['Solar System Size (kW)', 'Monthly Savings (₹)'],
+                           y=[system_size_kw, monthly_savings_rs],
+                           marker_color=['#636EFA', '#00CC96'])
+                ])
+                fig.update_layout(title="Solar Project Estimates", yaxis_title="Values", xaxis_title="Parameters")
+                st.plotly_chart(fig)
+            
     else:
         st.error("Sorry, the location entered does not match any available data.")
 else: