Spaces:
Running
Running
| import streamlit as st | |
| import folium | |
| from streamlit_folium import st_folium | |
| import psycopg2 | |
| st.set_page_config(layout="wide") | |
| 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 | |
| if 'point_type' not in st.session_state: | |
| st.session_state['point_type'] = 'start' | |
| # Custom CSS for colored buttons | |
| st.markdown( | |
| """ | |
| <style> | |
| .start-button { background-color: red; color: white; border: none; padding: 10px 20px; margin: 5px; border-radius: 5px; cursor: pointer; } | |
| .lost-button { background-color: orange; color: white; border: none; padding: 10px 20px; margin: 5px; border-radius: 5px; cursor: pointer; } | |
| .end-button { background-color: blue; color: white; border: none; padding: 10px 20px; margin: 5px; border-radius: 5px; cursor: pointer; } | |
| .reset-button { background-color: gray; color: white; border: none; padding: 10px 20px; margin: 5px; border-radius: 5px; cursor: pointer; } | |
| </style> | |
| """, | |
| unsafe_allow_html=True, | |
| ) | |
| st.sidebar.title("Step 1 - Add Markers") | |
| # Custom buttons for selecting point type | |
| col1, col2, col3 = st.sidebar.columns(3) | |
| with col1: | |
| if st.button("Start", key="start-button"): | |
| st.session_state['point_type'] = 'start' | |
| st.sidebar.markdown('<div class="start-button">Start</div>', unsafe_allow_html=True) | |
| with col2: | |
| if st.button("Lost", key="lost-button"): | |
| st.session_state['point_type'] = 'lost' | |
| st.sidebar.markdown('<div class="lost-button">Lost</div>', unsafe_allow_html=True) | |
| with col3: | |
| if st.button("End", key="end-button"): | |
| st.session_state['point_type'] = 'end' | |
| st.sidebar.markdown('<div class="end-button">End</div>', unsafe_allow_html=True) | |
| st.sidebar.write(f"Selected Point Type: {st.session_state['point_type'].capitalize()}") | |
| # Reset button | |
| if st.sidebar.button("Reset Markers", key="reset-button"): | |
| st.session_state['points'] = {'start': None, 'lost': None, 'end': None} | |
| st.experimental_rerun() | |
| 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=800) | |
| 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'][st.session_state['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=800) | |
| 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) | |