Update app.py

app.py

@@ -1,7 +1,7 @@
 import streamlit as st
 import pandas as pd
 import numpy as np
-#import matplotlib.pyplot as plt
+import matplotlib.pyplot as plt
 import openai
 
 # Load the solar data CSV file
@@ -38,15 +38,24 @@ def estimate_cost_and_savings(system_size, state_data, electricity_bill):
     return system_cost, savings_per_month, payback_period
 
 # Streamlit UI
+st.set_page_config(page_title="AI-based Solar Project Estimation Tool", layout="centered")
 st.title("AI-based Solar Project Estimation Tool")
 
-# User input form
-st.sidebar.header("Enter Your Details")
-location = st.sidebar.text_input("Enter your city or state", "")
-roof_size = st.sidebar.number_input("Enter your roof size (in sq meters)", min_value=1)
-electricity_bill = st.sidebar.number_input("Enter your monthly electricity bill (₹)", min_value=0)
+# Center all input widgets
+st.write("### Enter Your Details Below:")
 
-if location and roof_size > 0 and electricity_bill >= 0:
+with st.form("solar_form"):
+    # Get the list of unique states from the dataset
+    state_options = df['State'].dropna().unique()
+    
+    # Input widgets
+    location = st.selectbox("Select your State", options=sorted(state_options))
+    roof_size = st.number_input("Enter your roof size (in sq meters)", min_value=1)
+    electricity_bill = st.number_input("Enter your monthly electricity bill (₹)", min_value=0)
+    
+    submitted = st.form_submit_button("Get Estimate")
+
+if submitted and location and roof_size > 0 and electricity_bill >= 0:
     # Fetch state data from the dataset
     state_data = get_state_data(location)
     if state_data is not None:
@@ -69,9 +78,9 @@ if location and roof_size > 0 and electricity_bill >= 0:
         
         # Plot savings and payback period
         fig, ax = plt.subplots()
-        ax.bar(['Total System Cost', 'Monthly Savings', 'Payback Period'], 
+        ax.bar(['Total System Cost', 'Monthly Savings', 'Payback Period (months)'], 
                [system_cost, savings_per_month, payback_period*12], color=['blue', 'green', 'orange'])
-        ax.set_ylabel("Amount (₹) / Time (years)")
+        ax.set_ylabel("Amount (₹) / Time (Months)")
         ax.set_title("Solar System Estimation Breakdown")
         st.pyplot(fig)