feat: return chat history id

app.py

@@ -79,7 +79,7 @@ def on_click_callback():
             )
         )
         st.session_state.token_count += cb.total_tokens
-        insert_chat_history(conn, human_prompt, llm_response["answer"])
+        history_id = insert_chat_history(conn, human_prompt, llm_response["answer"])
 
 
 load_css()

chat_history.py

@@ -15,3 +15,10 @@ def insert_chat_history(conn: sqlalchemy.engine.Connection, query: str, answer:
             ],
         )
         conn.commit()
+
+        result = conn.execute(
+            text("SELECT id FROM chat_history ORDER BY id DESC LIMIT 1;")
+        )
+        last_row_id = result.fetchone()[0]
+        conn.commit()
+    return last_row_id