Update app.py

app.py

@@ -4,19 +4,22 @@ import google.generativeai as genai
 import os
 from io import StringIO
 import csv
+from dotenv import load_dotenv
 
+# Load environment variables from .env file
+load_dotenv()
 
 # Set page configuration first
 st.set_page_config(page_title="AI-based Solar Project Estimation Tool", layout="centered")
 
-# Get API key from environment variable
-api_key = os.getenv("AIzaSyCVRGVxIe1vESoAgykgHWOej-jZxiU-RKE")
-if not api_key:
-    st.error("Google API Key not set. Please set the GOOGLE_API_KEY environment variable.")
+# Initialize Gemini with the API key loaded from the .env file
+api_key = os.getenv("GOOGLE_API_KEY")
+if api_key:
+    genai.configure(api_key=api_key)
 else:
-    # Initialize Gemini with the API key
-    genai.configure(api_key=api_key)  # Ensure your Gemini API key is set correctly
-    model = genai.GenerativeModel("gemini-1.5-flash")
+    st.error("API key is missing. Please set the GOOGLE_API_KEY environment variable.")
+
+model = genai.GenerativeModel("gemini-1.5-flash")
 
 # Load solar data
 @st.cache_data
@@ -90,7 +93,7 @@ if st.button("Generate Batch & Download CSV"):
     if location and roof_size > 0 and electricity_bill >= 0:
         csv_buffer = StringIO()
         writer = csv.writer(csv_buffer)
-        writer.writerow(["Sequence_no", "Solar Estimate"] )
+        writer.writerow(["Sequence_no", "Solar Estimate"])
         
         # Call Gemini API once for the batch generation
         with st.spinner("Generating batch estimates..."):
@@ -109,5 +112,3 @@ if st.button("Generate Batch & Download CSV"):
         )
     else:
         st.warning("Please fill out all fields to generate batch estimates.")
-
-