Add message polling for new responses from the assistant

app.py

@@ -2,6 +2,7 @@ import logging
 import os
 import sqlite3
 import threading
+import time
 import urllib.parse
 from http.server import BaseHTTPRequestHandler, HTTPServer
 
@@ -131,6 +132,23 @@ if prompt := st.chat_input("Type your message..."):
         client.messages.create(
             body=prompt, from_=TWILIO_PHONE_NUMBER, to=KORA_PHONE_NUMBER
         )
+
+        # Periodically check for new messages from the assistant
+        def check_for_new_messages():
+            while True:
+                try:
+                    with db_lock:
+                        latest_message = c.execute(
+                            "SELECT sender, text FROM messages ORDER BY timestamp DESC LIMIT 1"
+                        ).fetchone()
+                        if latest_message and latest_message[0] == "assistant":
+                            st.chat_message("assistant").write(latest_message[1])
+                            break
+                except sqlite3.Error as e:
+                    logging.error(f"Error checking for new messages: {e}")
+                time.sleep(5)
+
+        check_for_new_messages()
     except sqlite3.Error as e:
         st.error(f"Error saving message to database: {e}")
     except Exception as e: