Update app.py

app.py

@@ -2,7 +2,7 @@ import gradio as gr
 import requests
 from urllib.parse import urljoin, urlencode, quote
 
-def process_survey(survey_number, submission_id): 
+def process_survey(button_type, submission_id): 
     survey_id = "671fc52152e3225d7de349e6"
     base_url = "https://app.surveystack.io/api/submissions"
     params = {
@@ -15,7 +15,7 @@ def process_survey(survey_number, submission_id):
 
     survey_url = f"{base_url}?{encoded_params}&match={match_param}"
 
-    print("SURVEY NUMBER: " + str(survey_number))
+    print("Button Type: " + button_type)
     print("PROCESS_SURVEY")
     print(survey_url)
 
@@ -25,7 +25,7 @@ def process_survey(survey_number, submission_id):
         data = response.json()
         #actual_data = data[0]['data']
 
-        if survey_number == 0: 
+        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)
         else:
             return data
@@ -71,14 +71,12 @@ with gr.Blocks() as app:
 
     get_survey_back = gr.Textbox(label="Data from SurveyStack")
 
-    survey_number = 0
-
     with gr.Row():
         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 number_input: process_survey(survey_number, number_input), 
+        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, get_survey_back]
     )
@@ -89,11 +87,10 @@ with gr.Blocks() as app:
 
     get_survey_back2 = gr.Textbox(label="Parameters from SurveyStack")
 
-    survey_number = 77
 
     #param_submit_button.click(fn=process_survey, inputs=[survey2_submission_ID, survey_number], outputs=[get_survey_back2])
     param_submit_button.click(
-        lambda number_input: process_survey(survey_number, number_input), 
+        lambda button_type: process_survey("param", button_type), 
         inputs=survey2_submission_ID, 
         outputs=get_survey_back2
     )