Update app.py

app.py

@@ -11,12 +11,13 @@ from dotenv import load_dotenv  # Import dotenv
 load_dotenv()
 
 # Check for API key environment variable
-api_key = os.getenv("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")
+api_key = os.getenv("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")
 
-if not api_key:
-    st.error("Groq API key is not set. Please configure it in your .env file.")
-else:
+if api_key:
     client = Groq(api_key=api_key)
+else:
+    client = None  # Set client to None if API key is not available
+    st.warning("Groq API key is not set. Features requiring Groq API will be disabled.")
 
 # App title
 st.title("Optimized EV Charging Scheduling in Microgrids")
@@ -28,22 +29,21 @@ uploaded_file = st.sidebar.file_uploader("Upload Excel File", type=["xlsx", "xls
 if uploaded_file:
     data = pd.ExcelFile(uploaded_file)
 
-    # Load data from sheets
     try:
+        # Load data from sheets
         historical_data = data.parse("Historical Data")
         renewable_forecast = data.parse("Renewable Energy Forecast")
         ev_profiles = data.parse("EV Charging Profiles")
 
-        # Display datasets
-        st.header("Uploaded Data")
-        st.subheader("Historical Data")
-        st.write(historical_data)
+        # Standardize column names
+        historical_data.columns = historical_data.columns.str.strip()
+        renewable_forecast.columns = renewable_forecast.columns.str.strip()
+        ev_profiles.columns = ev_profiles.columns.str.strip()
 
-        st.subheader("Renewable Energy Forecast")
-        st.write(renewable_forecast)
-
-        st.subheader("EV Charging Profiles")
-        st.write(ev_profiles)
+        # Rename specific columns if necessary
+        historical_data.rename(columns={"DATE": "Date", "date": "Date"}, inplace=True)
+        renewable_forecast.rename(columns={"DATE": "Date", "date": "Date"}, inplace=True)
+        ev_profiles.rename(columns={"DATE": "Date", "date": "Date"}, inplace=True)
 
         # Parameters
         rated_grid_capacity = 3500  # kW
@@ -91,14 +91,20 @@ if uploaded_file:
         )
 
         # Simulate Groq API
-        st.subheader("Groq AI Insights")
-        prompt = st.text_input("Ask Groq for insights (e.g., grid stability)")
-        if prompt:
-            response = client.chat.completions.create(
-                messages=[{"role": "user", "content": prompt}],
-                model="llama-3.3-70b-versatile",
-            )
-            st.write(response.choices[0].message.content)
+        if client:
+            st.subheader("Groq AI Insights")
+            prompt = st.text_input("Ask Groq for insights (e.g., grid stability)")
+            if prompt:
+                try:
+                    response = client.chat.completions.create(
+                        messages=[{"role": "user", "content": prompt}],
+                        model="llama-3.3-70b-versatile",
+                    )
+                    st.write(response.choices[0].message.content)
+                except Exception as e:
+                    st.error(f"Error communicating with Groq API: {e}")
+        else:
+            st.warning("Groq API key is missing. Insights feature is disabled.")
 
     except Exception as e:
         st.error(f"Error parsing file: {e}")