Update app.py

app.py

@@ -7,6 +7,12 @@ import csv
 from dotenv import load_dotenv
 
 
+import streamlit as st
+import pandas as pd
+import google.generativeai as genai
+import os
+from dotenv import load_dotenv
+
 # Load environment variables from .env file
 load_dotenv()
 
@@ -35,7 +41,7 @@ st.title("AI-based Solar Project Estimation Tool")
 st.write("### Enter Your Details Below:")
 
 with st.form("solar_form"):
-    state_options = df['states'].dropna().unique()  # Ensure using the correct column name 'states'
+    state_options = df['State'].dropna().unique()
     
     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)
@@ -62,11 +68,11 @@ def build_prompt(location, roof_size, electricity_bill, ghi, solar_cost_per_kw):
 
 # Generate the solar project estimate via Gemini
 if submitted and location and roof_size > 0 and electricity_bill >= 0:
-    state_data = df[df['states'].str.contains(location, case=False)].iloc[0]
+    state_data = df[df['State'].str.contains(location, case=False)].iloc[0]
     
     if state_data is not None:
         ghi = state_data['Avg_GHI (kWh/m²/day)']
-        solar_cost_per_kw = state_data['Solar_Cost_per_kW (₹)']  # Use the correct column name here
+        Solar_Cost_per_kW (₹) = state_data['Solar_Cost_per_kW (₹)']
         
         prompt_text = build_prompt(location, roof_size, electricity_bill, ghi, solar_cost_per_kw)