Update app.py

app.py

@@ -1,3 +1,4 @@
+import os
 import gradio as gr
 import requests
 from urllib.parse import urljoin, urlencode, quote
@@ -11,6 +12,9 @@ global parsed_input_data
 global original_outputs, yaml_outputs
 
 def process_survey(button_type, survey_submission_ID): 
+    my_variable = os.getenv("mySubmissionData")
+    print("HERE IS THE DATA " + my_variable)
+    
     if button_type == 'data':
         survey_id = "671fc52152e3225d7de349e6"
         
@@ -67,6 +71,18 @@ def display_parameter_survey():
     print(survey_url)
     
     iframe_html = f'<iframe src="{survey_url}" width="100%" height="600px"></iframe>'
+    iframe_html = f"""
+    <iframe src="{survey_url}" width="100%" height="600px"></iframe>
+    <script>
+        window.addEventListener("message", (event) => {{
+            const submissionData = JSON.stringify(event.data);
+            console.log("Submission Data", submissionData);
+            window.huggingface.variables.mySubmissionData = submissionData;
+        }});
+    </script>
+    <div id="submissionData" style="display:none;"></div>
+    """
+    
     return iframe_html
     
 def display_data_survey():