Update app.py

app.py

@@ -5,12 +5,24 @@ import os
 from io import StringIO
 import csv
 
+import streamlit as st
+import pandas as pd
+import google.generativeai as genai
+import os
+from io import StringIO
+import csv
+
 # Set page configuration first
 st.set_page_config(page_title="AI-based Solar Project Estimation Tool", layout="centered")
 
-# Initialize Gemini
-genai.configure(api_key=os.getenv("AIzaSyCVRGVxIe1vESoAgykgHWOej-jZxiU-RKE"))  # Ensure your Gemini API key is set as an environment variable
-model = genai.GenerativeModel("gemini-1.5-flash")
+# Get API key from environment variable
+api_key = os.getenv("GOOGLE_API_KEY")
+if not api_key:
+    st.error("Google API Key not set. Please set the GOOGLE_API_KEY environment variable.")
+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")
 
 # Load solar data
 @st.cache_data
@@ -84,7 +96,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..."):
@@ -103,3 +115,5 @@ if st.button("Generate Batch & Download CSV"):
         )
     else:
         st.warning("Please fill out all fields to generate batch estimates.")
+
+