Change to direct onMessage event handler

app.py

@@ -8,7 +8,7 @@ class State:
 state = State()
 
 js = """
-window.addEventListener('message', (event) => {alert('test')})
+window.onMessage = (event) => {alert('test')}
 """
 
 def update_values(context, question):