Update app.py

app.py

@@ -113,18 +113,23 @@ if prompt := st.chat_input("Ask me anything.."):
 
     # Insert data into the table
     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()
+        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))
+        )
         
-    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))
-    )
-    
-    # Commit the transaction
-    st.session_state.conn.commit()
-    cur.close()
-        
+        # Commit the transaction
+        st.session_state.conn.commit()
+        cur.close()
+    except Exception as e:
+        print("ERROR!!!\n", str(e))
+        print("User Input :", prompt)
+        print("Chatbot Response :", response)
+