import streamlit as st import folium from streamlit_folium import st_folium import psycopg2 st.set_page_config(layout="wide") @st.cache(allow_output_mutation=True) def connect_to_db(): return psycopg2.connect( dbname="glprui_jloddr", user="glprui_jloddr", password="612ef773", host="db.qgiscloud.com", port="5432", sslmode="prefer" ) def submit_data(age, gender, transport, multi_transport, time_of_day, day_of_week, description, points): conn = connect_to_db() cursor = conn.cursor() try: cursor.execute( """ INSERT INTO public.gettinglost_tracking (Age, Gender, Transport, TimeOfDay, DayOfWeek, Description) VALUES (%s, %s, %s, %s, %s, %s) RETURNING ID; """, (age, gender, transport, time_of_day, day_of_week, description) ) record_id = cursor.fetchone()[0] for pointType, point in points.items(): cursor.execute( """ INSERT INTO public.gettinglost_geom (ID, PointType, geom) VALUES (%s, %s, ST_SetSRID(ST_Point(%s, %s), 4326)); """, (record_id, pointType, point[0], point[1]) ) conn.commit() st.success("Data recorded successfully!") except Exception as e: conn.rollback() st.error(f"Error: {str(e)}") finally: cursor.close() conn.close() def create_map(points): m = folium.Map(location=[51.5074, -0.1278], zoom_start=10, control_scale=True) for point_type, coords in points.items(): if coords: folium.Marker( location=coords, popup=point_type, icon=folium.Icon(color="red" if point_type == "start" else "blue" if point_type == "end" else "orange") ).add_to(m) return m if 'points' not in st.session_state: st.session_state['points'] = {'start': None, 'lost': None, 'end': None} if 'survey' not in st.session_state: st.session_state['survey'] = False # Custom CSS for colored buttons st.markdown( """ """, unsafe_allow_html=True, ) st.sidebar.title("Step 1 - Add Markers") # Custom buttons for selecting point type point_type = st.sidebar.radio( "Select Point Type:", ["start", "lost", "end"], format_func=lambda x: x.capitalize() ) if point_type == "start": st.sidebar.markdown('
Start
', unsafe_allow_html=True) elif point_type == "lost": st.sidebar.markdown('
Lost
', unsafe_allow_html=True) elif point_type == "end": st.sidebar.markdown('
End
', unsafe_allow_html=True) map_placeholder = st.empty() with map_placeholder.container(): folium_map = create_map(st.session_state['points']) map_output = st_folium(folium_map, width="100%", height=500) new_coords = None if map_output and 'last_clicked' in map_output and map_output['last_clicked'] is not None: new_coords = (map_output['last_clicked']['lat'], map_output['last_clicked']['lng']) if new_coords: st.session_state['points'][point_type] = new_coords map_placeholder.empty() with map_placeholder.container(): folium_map = create_map(st.session_state['points']) st_folium(folium_map, width="100%", height=500) if all(st.session_state['points'].values()) and not st.session_state['survey']: if st.sidebar.button("Proceed to Survey"): st.session_state['survey'] = True else: st.sidebar.warning("Please add start, lost, and end points to proceed.") if st.session_state['survey']: st.sidebar.title("Step 2 - Survey Questions") age = st.sidebar.selectbox("Age", list(range(10, 101, 10))) gender = st.sidebar.radio("Gender", ["Male", "Female", "Other", "Prefer not to say"]) transport = st.sidebar.radio("Mode of Transport", ["Walk", "Car", "Bike", "Train", "Other", "Multi"]) multi_transport = st.sidebar.multiselect("If Multi, select modes used", ["Walk", "Car", "Bike", "Train", "Other"]) if transport == "Multi" else [] time_of_day = st.sidebar.selectbox("Time of Day", ["Morning", "Afternoon", "Evening", "Night"]) day_of_week = st.sidebar.selectbox("Day of the Week", ["Monday", "Tuesday", "Wednesday", "Thursday", "Friday", "Saturday", "Sunday"]) description = st.sidebar.text_area("Why did you get lost?") if st.sidebar.button("Save"): submit_data(age, gender, transport, multi_transport, time_of_day, day_of_week, description, st.session_state['points']) st.session_state['points'] = {'start': None, 'lost': None, 'end': None} st.session_state['survey'] = False st.experimental_rerun() # st.sidebar.write("Current Points:") # st.sidebar.json(st.session_state['points']) st.sidebar.markdown("---") st.sidebar.markdown("For a more detailed survey, click the link or scan the QR code:") st.sidebar.markdown("[https://arcg.is/1GK5jP0](https://arcg.is/1GK5jP0)") st.sidebar.image("static/Getting Lost Survey.png", width=200)