Update app.py

app.py

@@ -69,17 +69,17 @@ def generate_output(file, column1, column2, choice, bins):
     plt.savefig(image_path)
     # plt.close()
 
-    return df.head(), gr.update(visible=True), image_path, gr.update(visible=True)
+    return df.head(), gr.update(visible=True), image_path, gr.update(visible=True), gr.update(visible=True)
     
 with gr.Blocks() as demo:
     with gr.Row():
         with gr.Column():
             file_input = gr.File(label="上传表格文件（支持CSV、XLS、XLSX等格式", file_types=["csv", "xls", "xlsx"])
-            col1_dropdown = gr.Dropdown(label="选择列1", visible=False)
-            col2_dropdown = gr.Dropdown(label="选择列2", visible=False)
-            choice_radio = gr.Radio(["是", "否"], label="是否选择", visible=False)
-            slider = gr.Slider(minimum=3, maximum=7, step=1, label="选择数字", visible=False, value=4)
-            submit_button = gr.Button("提交")
+            col1_dropdown = gr.Dropdown(label="请选择特征列", visible=False)
+            col2_dropdown = gr.Dropdown(label="选择标签列", visible=False)
+            choice_radio = gr.Radio(["是", "否"], label="特征列是否为连续值", default="是", visible=False)
+            slider = gr.Slider(minimum=3, maximum=7, step=1, label="选择将特征列分组的分组数", visible=False, value=4)
+            submit_button = gr.Button("查看结果", visible=False)
         with gr.Column():
             df_display = gr.Dataframe(visible=False)
             output_image = gr.Image(visible=False)
@@ -91,6 +91,6 @@ with gr.Blocks() as demo:
     choice_radio.change(update_slider, inputs=choice_radio, outputs=slider)
     
     # 点击提交按钮时调用 generate_output 函数
-    submit_button.click(generate_output, inputs=[file_input, col1_dropdown, col2_dropdown, choice_radio, slider], outputs=[df_display, df_display, output_image, output_image])
+    submit_button.click(generate_output, inputs=[file_input, col1_dropdown, col2_dropdown, choice_radio, slider], outputs=[df_display, df_display, output_image, output_image, submit_button])
 
 demo.launch(share=True)