Update database.py

database.py

@@ -91,15 +91,32 @@ def end_session(session_id, end_time):
         _run(connection, "UPDATE session SET end_time = ? WHERE id = ?", (end_time, session_id))
 
 
-def add_interaction(user_id, input_text, detected_language, translated_text, nlu_summary, timestamp):
-    with get_connection() as connection:
-        return _insert_and_get_id(
-            connection,
-            """INSERT INTO interaction
-               (user_id, input_text, detected_language, translated_text, nlu_summary, timestamp)
-               VALUES (?, ?, ?, ?, ?, ?)""",
-            (user_id, input_text, detected_language, translated_text, nlu_summary, timestamp),
-        )
+def add_interaction(user_id, input_text, detected_language, translated_text,
+                    nlu_summary, timestamp):
+    p = _placeholder()
+
+    sql = (
+        f"INSERT INTO interaction "
+        f"(user_id, input_text, detected_language, translated_text, nlu_summary, timestamp) "
+        f"VALUES ({p}, {p}, {p}, {p}, {p}, {p})"
+    )
+
+    params = (
+        user_id,
+        input_text,
+        detected_language,
+        translated_text,
+        nlu_summary,
+        timestamp
+    )
+
+    with _connection() as conn:
+        if DATABASE_URL:
+            cursor = _run(conn, sql + " RETURNING id", params)
+            return cursor.fetchone()["id"]
+        else:
+            cursor = _run(conn, sql, params)
+            return cursor.lastrowid
 
 
 def get_interactions_for_user(user_id):