Update data_fetcher.py

data_fetcher.py

@@ -1,27 +1,38 @@
 import requests
 import pandas as pd
-from config import NREL_API_KEY
+import os
+
+# ✅ Read API key securely from environment variables
+NREL_API_KEY = os.getenv("NREL_API_KEY")
+
+if not NREL_API_KEY:
+    raise ValueError("❌ ERROR: NREL API Key is missing! Set it as an environment variable.")
 
-# Fetch data from NREL API
 def fetch_nrel_data():
-    url = f"https://developer.nrel.gov/api/hydrogen/v1/stations.json?api_key={L0l8YMg3shJjkOS9kBAUCsbDglzJ0uYd8wMCNBdT}"
-    response = requests.get(url)
-    
-    if response.status_code == 200:
+    """Fetch real-time hydrogen station data from NREL API."""
+    url = f"https://developer.nrel.gov/api/hydrogen/v1/stations.json?api_key={NREL_API_KEY}"
+
+    try:
+        response = requests.get(url, timeout=10)  # Timeout added for reliability
+        response.raise_for_status()  # Raise error for bad responses (4xx, 5xx)
         return response.json()
     
-    return {"error": "Failed to fetch data from NREL API"}
+    except requests.exceptions.RequestException as e:
+        print(f"⚠️ ERROR: Failed to fetch data from NREL API: {e}")
+        return None  # Return None if the API call fails
 
-# Function to clean and structure data
 def get_hydrogen_data():
+    """Process hydrogen station data into a DataFrame."""
     data = fetch_nrel_data()
     
-    if "error" in data:
-        return data
+    if data is None or "fuel_stations" not in data:
+        return None  # Return None if API data is unavailable
     
-    stations = data.get("fuel_stations", [])
+    stations = data["fuel_stations"]
     
     df = pd.DataFrame(stations)
-    df = df[["station_name", "state", "status", "capacity_kg_day", "street_address", "latitude", "longitude"]]
     
-    return df
+    if df.empty:
+        return None  # Return None if no data is available
+    
+    return df[["station_name", "state", "status", "capacity_kg_day", "street_address", "latitude", "longitude"]]