Update ui.py

snowflakedb · Jul 4, 2024 · 5acd9df · 5acd9df
1 parent fb8c26c
commit 5acd9df
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/hybrid-tables/python/src/ui.py b/hybrid-tables/python/src/ui.py
@@ -17,8 +17,10 @@ def run(self):
         st.dataframe(self.session.table('internal.dictionary').to_pandas(), use_container_width=True)
 
     def add(self, key: str, value: str):
-        result = self.session.call('core.add_key_value', key, value)
-        st.write(json.loads(result))
+        result = json.loads(self.session.call('core.add_key_value', key, value))
+        if "SQLCODE" in result:
+            error = f"Primary key violation on Hybrid Table. **{key}** already exists." if result["SQLCODE"] == 200001 else result["SQLERRM"]
+            st.error(error, icon="🚨")
 
 if __name__ == '__main__':
     ui = UI(Session.builder.getOrCreate())