Update app.py

app.py

@@ -75,9 +75,11 @@ def display_parameter_survey():
     <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;
+            if (event.origin === "https://app.surveystack.io" && event.data.type === "SUBMISSION_SUBMIT_SUCCESS") {{
+                const submissionId = event.data.payload.submissionId;
+                console.log("Submission ID:", submissionId);
+                window.huggingface.variables.mySubmissionData = submissionId; 
+            }}
         }});
     </script>
     """