Update app.py

app.py

@@ -9,6 +9,7 @@ from process_data import parse_survey_stack_parameters, parse_survey_stack_data,
 
 global parsed_input_data
 global original_outputs, yaml_outputs
+global survey_submission_ID
 
 def process_survey(button_type, submission_id): 
     if button_type == 'data':
@@ -68,6 +69,12 @@ def display_parameter_survey():
     
     iframe_html = f'<iframe src="{survey_url}" width="100%" height="600px"></iframe>'
     return iframe_html
+
+def process_submission():
+    global survey_submission_ID
+    submission_id = window.parent.gradio_app_submissionId
+    print("Submission ID from parent window:", submission_id)
+    survey_submission_ID = submission_id
     
 def display_data_survey():
     base_url = "https://app.surveystack.io/groups/"
@@ -88,13 +95,10 @@ def display_data_survey():
         // Listen for messages from the iframe
         window.addEventListener("message", (event) => {{
             console.log("Received message from iframe:", event.data);
-            // Get the payload 
-            const submissionData = JSON.stringify(event.data);
-            document.getElementById('submissionData').innerText = submissionData;
-            window.parent.updateSubmission(submissionData);
+            const submissionData = event.data.payload.submissionId;
+            window.parent.gradio_app_submissionId = submissionId;
         }});
     </script>
-    <div id="submissionData" style="display:none;"></div>
     """
     return iframe_html
     
@@ -122,9 +126,9 @@ with gr.Blocks() as app:
     with gr.Row():
         data_survey_output = gr.HTML(value=display_data_survey())
 
-    survey_submission_ID = gr.Textbox(label="Before submitting your surveystack survey, PLEASE record your submission ID. Type it here.")
+    #survey_submission_ID = gr.Textbox(label="Before submitting your surveystack survey, PLEASE record your submission ID. Type it here.")
     
-    data_submit_button = gr.Button("Parse Data")
+    data_submit_button = gr.Button(label="Parse Data", fn=process_submission)
 
     get_survey_back = gr.Textbox(label="Data from SurveyStack")
 
@@ -157,5 +161,6 @@ with gr.Blocks() as app:
         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()