Update app.py

app.py

@@ -55,8 +55,8 @@ def get_db_connection(conn_url, password=None):
 if 'conn' not in st.session_state:
     st.session_state.conn = get_db_connection(POSTGRE_URL)
 
-if 'cursor' not in st.session_state:
-    st.session_state.cursor = st.session_state.conn.cursor()
+# if 'cursor' not in st.session_state:
+#     st.session_state.cursor = st.session_state.conn.cursor()
 
 if 'memory' not in st.session_state:
     st.session_state['memory'] = ConversationBufferMemory(return_messages=True)
@@ -112,7 +112,7 @@ if prompt := st.chat_input("Ask me anything.."):
     st.session_state.messages.append({"role": "assistant", "content": response})
 
     # Insert data into the table
-    st.session_state.cursor.execute(
+    st.session_state.conn.cursor().execute(
         "INSERT INTO chat_history (input_text, response_text, created_at) VALUES (%s, %s, %s)",
         (prompt, response, datetime.now(timezone.utc) + timedelta(hours=7))
     )