Update app.py

app.py

@@ -18,7 +18,7 @@ def main():
     if st.button("Send"):
         messages = chat.run(prompt)
 
-        for message in messages:
+        for message in prompt:
             if message['source'] == 'user':
                 user_msg = cont.chat_message("user")
                 user_msg.write(message['message'])