Spaces:
Runtime error
Runtime error
File size: 2,919 Bytes
7005448 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 |
import gradio as gr
import psycopg2
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, start_point, lost_point, end_point):
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]
points = {'start': start_point, 'lost': lost_point, 'end': end_point}
for pointType, point in points.items():
if point:
cursor.execute(
"""
INSERT INTO public.gettinglost_geom (ID, PointType, geom)
VALUES (%s, %s, ST_SetSRID(ST_Point(%s, %s), 4326));
""",
(record_id, pointType, point[1], point[0])
)
conn.commit()
return "Data recorded successfully!"
except Exception as e:
conn.rollback()
return f"Error: {str(e)}"
finally:
cursor.close()
conn.close()
age = gr.Dropdown(choices=["0-10", "11-20", "21-30", "31-40", "41-50", "51-60", "61-70", "71-80", "81-90", "91-100"], label="Age")
gender = gr.Radio(choices=["M", "F", "O", "PNTS"], label="Gender")
transport = gr.Radio(choices=["Walk", "Car", "Bike", "Train", "Other", "Multi"], label="Mode of Transport")
multi_transport = gr.CheckboxGroup(choices=["Walk", "Car", "Bike", "Train", "Other"], label="If Multi, Select Modes Used", visible=False)
time_of_day = gr.Dropdown(choices=["Morning", "Afternoon", "Evening", "Night"], label="Time of Day")
day_of_week = gr.Dropdown(choices=["Monday", "Tuesday", "Wednesday", "Thursday", "Friday", "Saturday", "Sunday"], label="Day of the Week")
description = gr.Textbox(lines=4, placeholder="Why did you get lost?", label="Description")
start_point = gr.Map(label="Select Start Point")
lost_point = gr.Map(label="Select Getting Lost Point")
end_point = gr.Map(label="Select End Point")
def update_visibility(transport_mode):
if transport_mode == "Multi":
return gr.update(visible=True)
else:
return gr.update(visible=False)
transport.change(fn=update_visibility, inputs=transport, outputs=multi_transport)
iface = gr.Interface(
fn=submit_data,
inputs=[age, gender, transport, multi_transport, time_of_day, day_of_week, description, start_point, lost_point, end_point],
outputs="text",
live=True
)
if __name__ == "__main__":
iface.launch()
|