Update app.py

app.py

@@ -1,6 +1,7 @@
 #ref: https://github.com/kram254/Mixture-of-Agents-running-on-Groq/tree/main
 import streamlit as st
 import json
+import threading
 from typing import Iterable
 from moa.agent import MOAgent
 from moa.agent.moa import ResponseChunk
@@ -65,12 +66,18 @@ layer_agent_config_rec = {
 
 def stream_response(messages: Iterable[ResponseChunk]):
     layer_outputs = {}
+    lock = threading.Lock()
+    
+    def safe_append(layer, delta):
+        with lock:
+            if layer not in layer_outputs:
+                layer_outputs[layer] = []
+            layer_outputs[layer].append(delta)
+
     for message in messages:
         if message['response_type'] == 'intermediate':
             layer = message['metadata']['layer']
-            if layer not in layer_outputs:
-                layer_outputs[layer] = []
-            layer_outputs[layer].append(message['delta'])
+            safe_append(layer, message['delta'])
         else:
             # Display accumulated layer outputs
             for layer, outputs in layer_outputs.items():