| import gradio as gr | |
| import compare_kit | |
| import pandas as pd | |
| def analyse_ifc(file1, file2, token): | |
| result1, result2 = compare_kit.ChatIFC_multiple(file1.name, file2.name, token) | |
| return result1, result2 | |
| def main(): | |
| input_file1 = gr.File(label="Please upload a json file") | |
| input_file2 = gr.File(label="Please upload a txt file") | |
| token = gr.Textbox(label="Please provide chat AI tokens (seperate by ';')") | |
| out_display = gr.Dataframe(label="Result table is below. Please wait several minutes", | |
| type="pandas", overflow_row_behaviour="show_ends") | |
| outp = gr.File(label="Please download the csv file here") | |
| demo = gr.Interface(fn = analyse_ifc, | |
| inputs = [input_file1, input_file2, token], | |
| outputs = [out_display, outp], | |
| title = "ChatIFC", | |
| description = """This is a localhost interface for checking whether the building has violated the regulations. \nYou can upload your ifc and law file to the left side. \nThe information will be output on the right side.""" | |
| ) | |
| demo.launch() | |
| if __name__ == '__main__': | |
| main() | |