Update app.py

app.py

@@ -23,43 +23,62 @@ if 'location_name' not in st.session_state:
 default_date = datetime.utcnow().date()
 selected_date = st.text_input("Date (YYYY-MM-DD)", value=str(default_date))
 
-# Layout: 2 columns for inputs
+# Layout: 3 columns for inputs
 col1, col2, col3 = st.columns(3)
 
 with col1:
-    latitude = st.text_input("Latitude", value=str(st.session_state['latitude']))
+    latitude_input = st.text_input("Latitude", value=str(st.session_state['latitude']))
 with col2:
-    longitude = st.text_input("Longitude", value=str(st.session_state['longitude']))
+    longitude_input = st.text_input("Longitude", value=str(st.session_state['longitude']))
 with col3:
     location_search = st.text_input("Search Location (English only)", value=st.session_state['location_name'])
 
 # Search button
 if st.button("Search"):
-    if location_search:
-     with st.spinner("🔎 Searching for location..."):
-        try:
+    with st.spinner("🔎 Searching and updating the location..."):
+        # First try location search
+        if location_search.strip() != "":
             location = geolocator.geocode(location_search, language="en")
             if location:
                 st.session_state['latitude'] = location.latitude
                 st.session_state['longitude'] = location.longitude
                 st.session_state['location_name'] = location.address
                 st.success(f"✅ Location found: {location.address}")
+                st.balloons()
             else:
-                st.error("❌ Location not found. Try another keyword.")
-        except Exception as e:
-            st.error(f"❌ Error while searching the location: {e}")
-
+                st.error("❌ Location not found. Please try another keyword.")
+        # If no location given, fall back to lat/lon
+        else:
+            try:
+                lat = float(latitude_input)
+                lon = float(longitude_input)
+                location = geolocator.reverse((lat, lon), language="en")
+                if location:
+                    st.session_state['location_name'] = location.address
+                    st.success(f"✅ Coordinates updated to: {location.address}")
+                    st.balloons()
+                else:
+                    st.warning("⚠️ Valid coordinates but no address found.")
+                st.session_state['latitude'] = lat
+                st.session_state['longitude'] = lon
+            except ValueError:
+                st.error("❌ Invalid latitude or longitude format.")
 
 # Map
-st.subheader("Map View:")
+st.subheader("🗺️ Map View:")
 
 # Create Map
-m = folium.Map(location=[st.session_state['latitude'], st.session_state['longitude']], zoom_start=5)
+m = folium.Map(
+    location=[st.session_state['latitude'], st.session_state['longitude']],
+    zoom_start=5,
+    control_scale=True,
+    tiles="OpenStreetMap"
+)
 
-# Add marker
+# Add marker in English popup
 folium.Marker(
     [st.session_state['latitude'], st.session_state['longitude']],
-    popup=st.session_state['location_name'] or "Selected Location",
+    popup=f"📍 {st.session_state['location_name'] or 'Selected Location'}",
     tooltip="Click for location",
 ).add_to(m)
 
@@ -80,4 +99,5 @@ if st_data and st_data.get("last_clicked"):
             st.session_state['location_name'] = ""
     except:
         st.session_state['location_name'] = ""
-    st.experimental_rerun()  # Rerun to update UI
+    st.experimental_rerun()
+