Update app.py

app.py

@@ -29,9 +29,9 @@ def process_survey(button_type, submission_id):
         print(data)
 
         if button_type == 'data': 
-            return data, gr.HTML(visible=True), gr.Textbox(visible=False), gr.Textbox(visible=False), gr.Button(visible=False), gr.Textbox(visible=False)
+            return data, gr.HTML(visible=True), gr.Textbox(visible=False), gr.Textbox(visible=False), gr.Button(visible=False), gr.Textbox(visible=False), gr.Textbox(visible=True), gr.Textbox(visible=True)
         else:
-            return data
+            return data, gr.Textbox(visible=True)
         
     except Exception as e:
         return f"An error occurred while fetching the survey data: {e}"
@@ -78,11 +78,6 @@ with gr.Blocks() as app:
         parameter_survey_output = gr.HTML(value=display_parameter_survey(), visible=False)
         
     #data_submit_button.click(fn=process_survey, inputs=[survey_submission_ID, survey_number], outputs=[get_survey_back, parameter_survey_output, data_survey_output, survey_submission_ID, data_submit_button, get_survey_back])
-    data_submit_button.click(
-        lambda button_type: process_survey("data", button_type), 
-        inputs=[survey_submission_ID], 
-        outputs=[get_survey_back, parameter_survey_output, data_survey_output, survey_submission_ID, data_submit_button]
-    )
     
     survey2_submission_ID = gr.Textbox(label="Before submitting your surveystack survey, PLEASE record your submission ID. Type it here.", visible=False)
 
@@ -95,7 +90,13 @@ with gr.Blocks() as app:
     param_submit_button.click(
         lambda button_type: process_survey("param", button_type), 
         inputs=survey2_submission_ID, 
-        outputs=get_survey_back2
+        outputs=[get_survey_back2, get_survey_back2]
+    )
+
+    data_submit_button.click(
+        lambda button_type: process_survey("data", button_type), 
+        inputs=[survey_submission_ID], 
+        outputs=[get_survey_back, parameter_survey_output, data_survey_output, survey_submission_ID, data_submit_button, survey2_submission_ID, param_submit_button]
     )
 
 app.launch()