Update app.py

app.py

@@ -112,7 +112,13 @@ if prompt := st.chat_input("Ask me anything.."):
     st.session_state.messages.append({"role": "assistant", "content": response})
 
     # Insert data into the table
-    cur = st.session_state.conn.cursor()
+    try :
+        cur = st.session_state.conn.cursor()
+    except:
+        get_db_connection.clear()
+        st.session_state.conn = get_db_connection(POSTGRE_URL)
+        cur = st.session_state.conn.cursor()
+        
     cur.execute(
         f"INSERT INTO chat_history (input_text, response_text, created_at) VALUES (%s, %s, %s)",
         (prompt, response, datetime.now(timezone.utc) + timedelta(hours=7))