Persist UI results in session state

ui.py

@@ -25,6 +25,17 @@ def render_ui():
         st.session_state.openai_input = ""
     if 'compare_input' not in st.session_state:
         st.session_state.compare_input = ""
+        # Add state for results persistence
+        if 'ingredient_results_html' not in st.session_state:
+            st.session_state.ingredient_results_html = None
+        if 'category_results_html' not in st.session_state:
+            st.session_state.category_results_html = None
+        if 'voyage_results_html' not in st.session_state:
+            st.session_state.voyage_results_html = None
+        if 'openai_results_html' not in st.session_state:
+            st.session_state.openai_results_html = None
+        if 'compare_results_html' not in st.session_state:
+            st.session_state.compare_results_html = None
 
     # Page config is now set in app.py
     st.title("Product Categorization Tool")
@@ -79,6 +90,7 @@ def render_ui():
                 # Results section
                 st.subheader("Results")
                 results_placeholder_ingredient = st.container() # Use container for results area
+                # --- Results Display Logic (Ingredient Tab) ---
                 if find_ingredients_btn:
                     if st.session_state.ingredient_input: # Check state value
                         with st.spinner("Finding similar ingredients..."):
@@ -89,12 +101,17 @@ def render_ui():
                                 top_n,
                                 confidence
                             )
-                        results_placeholder_ingredient.markdown(results_html, unsafe_allow_html=True)
+                        st.session_state.ingredient_results_html = results_html # Store results
+                        results_placeholder_ingredient.markdown(results_html, unsafe_allow_html=True) # Display immediately
                     else:
+                        st.session_state.ingredient_results_html = None # Clear results if input is empty
                         results_placeholder_ingredient.warning("Please enter product names.")
+                # Display stored results if button wasn't clicked but results exist
+                elif 'ingredient_results_html' in st.session_state and st.session_state.ingredient_results_html:
+                     results_placeholder_ingredient.markdown(st.session_state.ingredient_results_html, unsafe_allow_html=True)
                 else:
-                    # Keep the placeholder visible even when no button is clicked yet
-                    results_placeholder_ingredient.write("Results will appear here.")
+                     # Initial state or after clearing
+                     results_placeholder_ingredient.write("Results will appear here.")
 
     # --- Category Matching Tab ---
     with tab_category:
@@ -132,6 +149,7 @@ def render_ui():
             with st.container(border=True):
                 st.subheader("Results")
                 results_placeholder_category = st.container() # Use container
+                # --- Results Display Logic (Category Tab) ---
                 if match_categories_btn:
                     if st.session_state.category_input:
                         with st.spinner("Matching to categories..."):
@@ -142,11 +160,17 @@ def render_ui():
                                 category_top_n,
                                 category_confidence
                             )
-                        results_placeholder_category.markdown(results_html, unsafe_allow_html=True)
+                        st.session_state.category_results_html = results_html # Store results
+                        results_placeholder_category.markdown(results_html, unsafe_allow_html=True) # Display immediately
                     else:
+                        st.session_state.category_results_html = None # Clear results if input is empty
                         results_placeholder_category.warning("Please enter product names.")
+                # Display stored results if button wasn't clicked but results exist
+                elif 'category_results_html' in st.session_state and st.session_state.category_results_html:
+                     results_placeholder_category.markdown(st.session_state.category_results_html, unsafe_allow_html=True)
                 else:
-                    results_placeholder_category.write("Results will appear here.")
+                     # Initial state or after clearing
+                     results_placeholder_category.write("Results will appear here.")
 
     # --- Common function for Reranking Tabs ---
     def create_reranking_ui(tab, tab_key_prefix, tab_name, backend_function, default_match="categories"):
@@ -195,11 +219,15 @@ def render_ui():
                 with st.container(border=True):
                     st.subheader("Results")
                     results_placeholder_rerank = st.container() # Use container
+                    # --- Results Display Logic (Reranking Tabs) ---
+                    results_key = f"{tab_key_prefix}_results_html"
+                    input_key = f"{tab_key_prefix}_input"
+
                     if tab_match_btn:
-                        if st.session_state[f"{tab_key_prefix}_input"]:
+                        if st.session_state[input_key]:
                             with st.spinner(f"Matching using {tab_name}..."):
                                 results_html = backend_function(
-                                    st.session_state[f"{tab_key_prefix}_input"],
+                                    st.session_state[input_key],
                                     False,
                                     tab_expansion,
                                     tab_emb_top_n,
@@ -207,11 +235,17 @@ def render_ui():
                                     tab_confidence,
                                     tab_match_type
                                 )
-                            results_placeholder_rerank.markdown(results_html, unsafe_allow_html=True)
+                            st.session_state[results_key] = results_html # Store results
+                            results_placeholder_rerank.markdown(results_html, unsafe_allow_html=True) # Display immediately
                         else:
+                            st.session_state[results_key] = None # Clear results if input is empty
                             results_placeholder_rerank.warning("Please enter product names.")
+                    # Display stored results if button wasn't clicked but results exist
+                    elif results_key in st.session_state and st.session_state[results_key]:
+                         results_placeholder_rerank.markdown(st.session_state[results_key], unsafe_allow_html=True)
                     else:
-                        results_placeholder_rerank.write("Results will appear here.")
+                         # Initial state or after clearing
+                         results_placeholder_rerank.write("Results will appear here.")
 
     # Create the reranking tabs
     create_reranking_ui(tab_voyage, "voyage", "Voyage AI Reranking", categorize_products_with_voyage_reranking, "categories")
@@ -276,6 +310,7 @@ def render_ui():
             with st.container(border=True):
                 st.subheader("Comparison Results")
                 results_placeholder_compare = st.container() # Use container
+                # --- Results Display Logic (Compare Tab) ---
                 if compare_btn:
                     if st.session_state.compare_input:
                         with st.spinner("Comparing methods..."):
@@ -287,11 +322,17 @@ def render_ui():
                                 compare_match_type,
                                 compare_expansion
                             )
-                        results_placeholder_compare.markdown(results_html, unsafe_allow_html=True)
+                        st.session_state.compare_results_html = results_html # Store results
+                        results_placeholder_compare.markdown(results_html, unsafe_allow_html=True) # Display immediately
                     else:
+                        st.session_state.compare_results_html = None # Clear results if input is empty
                         results_placeholder_compare.warning("Please enter product names.")
+                # Display stored results if button wasn't clicked but results exist
+                elif 'compare_results_html' in st.session_state and st.session_state.compare_results_html:
+                     results_placeholder_compare.markdown(st.session_state.compare_results_html, unsafe_allow_html=True)
                 else:
-                    results_placeholder_compare.write("Results will appear here.")
+                     # Initial state or after clearing
+                     results_placeholder_compare.write("Results will appear here.")
 
     st.markdown("---")
     st.markdown("Powered by Voyage AI embeddings • Built with Streamlit")