Update app.py

app.py

@@ -132,13 +132,15 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
                     o9 = gr.Radio(["總是", "經常","一半一半","偶而","完全不曾"], label="10. 刮風的狀況？")
                     o10 = gr.Radio(["總是", "經常","一半一半","偶而","完全不曾"], label="12. 區域使用空調？")
                     all_osdi = [o1, o2, o3, o4, o5, o6, o7, o8, o9, o10]
-                
-                submit_btn = gr.Button("生成診斷報告", variant="primary")
+                with gr.Row():
+                    back_to_ccmq = gr.Button("返回 CCMQ")
+                    submit_btn = gr.Button("送出診斷並生成報告", variant="primary")              
 
     with gr.Column(visible=False) as stage_2:
         gr.Markdown("## AI 診斷分析報告")
         with gr.Row():
             res_table = gr.Dataframe(headers=["項目", "回答", "狀態"], interactive=False)
+            back_to_edit = gr.Button("重新修改問卷")
             with gr.Column():
                 res_prob = gr.Label(label="預測機率")
                 res_title = gr.Markdown("### 診斷結果")
@@ -150,6 +152,8 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
 
     all_inputs = all_ccmq + all_osdi
     btn_next.click(fn=lambda: gr.Tabs(selected=1), outputs=survey_tabs)
+    back_to_ccmq.click(fn=lambda: gr.Tabs(selected=0), outputs=survey_tabs)
+    back_to_edit.click(fn=lambda: (gr.update(visible=True), gr.update(visible=False)), outputs=[stage_1, stage_2])
     submit_btn.click(fn=analyze_and_predict, inputs=all_inputs, outputs=[stage_1, stage_2, res_title, res_desc, res_prob, res_table, plot_1, plot_2])
     finish_btn.click(fn=reset_system, outputs=[stage_1, stage_2, survey_tabs] + all_inputs)