Update app.py

app.py

@@ -156,28 +156,23 @@ for provider in mockData:
 
 df = pd.DataFrame(data)
 
-# 정렬 함수
-def sort_data(column, ascending):
-    sorted_df = df.sort_values(by=column, ascending=ascending)
+# Function to sort the data based on a column
+def sort_data(column, ascending=True):
+    return df.sort_values(by=column, ascending=ascending)
+
+# Gradio interface to display and interact with the data
+def update_table(column, ascending):
+    sorted_df = sort_data(column, ascending)
     return sorted_df
 
-# Gradio 인터페이스 설정
 with gr.Blocks() as demo:
     gr.Markdown("# Comparison of LLM (Large Language Model) Costs Based on One Million Tokens")
-    
-    with gr.Row():
-        column = gr.Dropdown(choices=df.columns.tolist(), label="Select column to sort")
-        ascending = gr.Checkbox(label="ascending order", value=True)
-        sort_button = gr.Button("Sort")
-    
-    table = gr.Dataframe(value=df, interactive=True, headers=["Provider", "URI", "Model", "**Total Price**", "Input Price", "Output Price"])
-    
-    def update_table(column, ascending):
-        sorted_df = sort_data(column, ascending)
-        return sorted_df
-    
-    sort_button.click(update_table, inputs=[column, ascending], outputs=table)
+    column = gr.Dropdown(label="Select column to sort by", choices=df.columns.tolist(), default="Total Price")
+    ascending = gr.Checkbox(label="Ascending order", value=True)
+    sort_button = gr.Button("Sort")
+    table = gr.Dataframe(df)
+    sort_button.click(fn=update_table, inputs=[column, ascending], outputs=table)
 
-# 애플리케이션 실행
+# Run the application
 if __name__ == "__main__":
     demo.launch()