import gradio as gr
from main import main_fn

#Input structure
##Postal_,age_,town_,storey_,room_ = 680705, 30, 'CHOA CHU KANG', 12, '5 ROOM'

town_list = ['ANG MO KIO', 'BEDOK', 'BISHAN', 'BUKIT BATOK', 'BUKIT MERAH', 'BUKIT PANJANG', 'BUKIT TIMAH', 'CENTRAL AREA', 'CHOA CHU KANG', 'CLEMENTI', 'GEYLANG', 'HOUGANG', 'JURONG EAST', 'JURONG WEST', 'KALLANG/WHAMPOA', 'MARINE PARADE', 'PASIR RIS', 'PUNGGOL', 'QUEENSTOWN', 'SEMBAWANG', 'SENGKANG', 'SERANGOON', 'TAMPINES', 'TOA PAYOH', 'WOODLANDS', 'YISHUN']
room_list = ['1 ROOM', '2 ROOM', '3 ROOM', '4 ROOM', '5 ROOM', 'EXECUTIVE', 'MULTI-GENERATION']

iface = gr.Interface(
    fn=main_fn, 
    inputs= [
            gr.inputs.Number(default=680705, label='Postal Code'),
            gr.inputs.Number(default=25, label='Years since lease commencement (TOP)'),
            gr.inputs.Dropdown(choices=town_list, type="value", default=None, label='Town'),
            gr.inputs.Number(default=11, label='Floor'),
            gr.inputs.Dropdown(choices=room_list, type="value", default=None, label='Room')
        ], 
    outputs= [
            gr.Textbox(type="text", label='Predicted House Price ($)'),
            gr.Dataframe(row_count = (10, "dynamic"), col_count=(4, "fixed"), label="Past transactions")
        ]
    )
iface.launch()