Update app.py

app.py

@@ -284,8 +284,9 @@ def Page():
                 if messages.value !=[]: response(messages.value[-1]["content"])
     
             result = solara.lab.use_task(result, dependencies=[user_message_count])  # type: ignore
-            for item in messages.value:
-                with ChatBox(style={"background-color":"transparent!important","flex-grow":"1","position": "fixed", "bottom": "10rem", "width": "50%"}):
+            
+            with ChatBox(style={"background-color":"transparent!important","flex-grow":"1","position": "fixed", "bottom": "10rem", "width": "50%"}):
+                for item in messages.value:
                     with ChatMessage(
                         user=item["role"] == "user",
                         name="StreamBot" if item["role"] == "assistant" else "User",