import gradio as gr import psycopg2 from dataclasses import dataclass, field from typing import List, Optional from mysite.interpreter.process import no_process_file,process_file import os #### @dataclass class Ride: ride_id: Optional[int] = field(default=None) rideable_type: str = '' start_station_id: int = 0 start_station_name: str = '' end_station_id: int = 0 end_station_name: str = '' started_at: str = '' ended_at: str = '' member_casual: str = '' def connect_to_db(): conn = psycopg2.connect( dbname="neondb", user=os.getenv("postgre_user"), password=os.getenv("postgre_pass"), host=os.getenv("postgre_host"), port=5432, sslmode="require" ) return conn def create_ride(ride: Ride): conn = connect_to_db() cur = conn.cursor() cur.execute("INSERT INTO rides (rideable_type, start_station_id, start_station_name, end_station_id, end_station_name, started_at, ended_at, member_casual) VALUES (%s, %s, %s, %s, %s, %s, %s, %s) RETURNING ride_id", (ride.rideable_type, ride.start_station_id, ride.start_station_name, ride.end_station_id, ride.end_station_name, ride.started_at, ride.ended_at, ride.member_casual)) ride_id = cur.fetchone()[0] conn.commit() cur.close() conn.close() return ride_id def read_rides(): conn = connect_to_db() cur = conn.cursor() cur.execute("SELECT * FROM rides ORDER BY ride_id desc") rides = cur.fetchall() conn.close() return rides def read_ride(ride_id: int): conn = connect_to_db() cur = conn.cursor() cur.execute("SELECT * FROM rides WHERE ride_id = %s", (ride_id,)) ride = cur.fetchone() conn.close() return ride def update_ride(ride: Ride): conn = connect_to_db() cur = conn.cursor() no_process_file(ride.start_station_name,ride.end_station_name) cur.execute("UPDATE rides SET rideable_type = %s, start_station_id = %s, start_station_name = %s, end_station_id = %s, end_station_name = %s, started_at = %s, ended_at = %s, member_casual = %s WHERE ride_id = %s", (ride.rideable_type, ride.start_station_id, ride.start_station_name, ride.end_station_id, ride.end_station_name, ride.started_at, ride.ended_at, ride.member_casual, ride.ride_id)) conn.commit() cur.close() conn.close() def delete_ride(ride_id: int): conn = connect_to_db() cur = conn.cursor() cur.execute("DELETE FROM rides WHERE ride_id = %s", (ride_id,)) conn.commit() cur.close() conn.close() def test_set_lide(input="test",foldername="test"): ride = Ride( rideable_type="rideable_type", start_station_name=input, end_station_name=foldername, ) create_ride(ride) return [[r[0], r[1], r[2], r[3], r[4], r[5], r[6], r[7], r[8]] for r in read_rides()] #def crud_interface(): with gr.Blocks() as gradio_interface: with gr.Row(): rideable_type = gr.Textbox(label="Rideable Type") start_station_id = gr.Number(label="Start Station ID") start_station_name = gr.Textbox(label="Start Station Name") end_station_id = gr.Number(label="End Station ID") end_station_name = gr.Textbox(label="End Station Name") started_at = gr.Textbox(label="Started At") ended_at = gr.Textbox(label="Ended At") member_casual = gr.Textbox(label="Member Casual") ride_id = gr.Number(label="Ride ID (for update/delete)", value=-1, interactive=False) create_button = gr.Button("Create Ride") update_button = gr.Button("Update Ride") delete_button = gr.Button("Delete Ride") read_button = gr.Button("Read Rides") output = gr.Dataframe(headers=["Ride ID", "Rideable Type", "Start Station ID", "Start Station Name", "End Station ID", "End Station Name", "Started At", "Ended At", "Member Casual"]) def create_ride_click(rideable_type, start_station_id, start_station_name, end_station_id, end_station_name, started_at, ended_at, member_casual): ride = Ride( rideable_type=rideable_type, start_station_id=start_station_id, start_station_name=start_station_name, end_station_id=end_station_id, end_station_name=end_station_name, started_at=started_at, ended_at=ended_at, member_casual=member_casual ) create_ride(ride) return [[r[0], r[1], r[2], r[3], r[4], r[5], r[6], r[7], r[8]] for r in read_rides()] def update_ride_click(ride_id, rideable_type, start_station_id, start_station_name, end_station_id, end_station_name, started_at, ended_at, member_casual): ride = Ride( ride_id=int(ride_id), rideable_type=rideable_type, start_station_id=start_station_id, start_station_name=start_station_name, end_station_id=end_station_id, end_station_name=end_station_name, started_at=started_at, ended_at=ended_at, member_casual=member_casual ) update_ride(ride) return [[r[0], r[1], r[2], r[3], r[4], r[5], r[6], r[7], r[8]] for r in read_rides()] def delete_ride_click(ride_id): delete_ride(int(ride_id)) return [[r[0], r[1], r[2], r[3], r[4], r[5], r[6], r[7], r[8]] for r in read_rides()] def read_rides_click(): return [[r[0], r[1], r[2], r[3], r[4], r[5], r[6], r[7], r[8]] for r in read_rides()] def load_ride_details(evt: gr.SelectData): row_index = evt.index[0] if isinstance(evt.index, list) else evt.index rides = read_rides() selected_row = rides[row_index] ride_id = selected_row[0] ride = read_ride(ride_id) if ride: return ride[1], ride[2], ride[3], ride[4], ride[5], ride[6], ride[7], ride[8], ride[0] return "", 0, "", 0, "", "", "", "", -1 create_button.click(fn=create_ride_click, inputs=[rideable_type, start_station_id, start_station_name, end_station_id, end_station_name, started_at, ended_at, member_casual], outputs=output) update_button.click(fn=update_ride_click, inputs=[ride_id, rideable_type, start_station_id, start_station_name, end_station_id, end_station_name, started_at, ended_at, member_casual], outputs=output) delete_button.click(fn=delete_ride_click, inputs=ride_id, outputs=output) read_button.click(fn=read_rides_click, outputs=output) output.select(fn=load_ride_details, inputs=None, outputs=[rideable_type, start_station_id, start_station_name, end_station_id, end_station_name, started_at, ended_at, member_casual, ride_id]) #return interface #d1 = crud_interface() #d1.launch()