Update app.py

app.py

@@ -3,7 +3,8 @@ import requests
 import geocoder
 
 # 👉 Put your API key here
-API_KEY = "API_KEY"
+import os
+API_KEY = os.getenv("API_KEY")
 
 # -------- FUNCTIONS --------
 def get_weather(city):
@@ -37,8 +38,12 @@ city = st.text_input("Enter City Name", city)
 if city:
     weather = get_weather(city)
 
-    if weather.get("cod") != 200:
+    if weather.get("cod") == 401:
+        st.error("❌ Invalid API Key")
+    elif weather.get("cod") == "404":
         st.error("❌ City not found")
+    elif weather.get("cod") != 200:
+        st.error(f"Error: {weather.get('message')}")
     else:
         st.subheader(f"📍 Weather in {city}")