update: Enhance session state management for search results and reactions in the evaluation interface

app.py

@@ -32,11 +32,15 @@ def save_reactions_to_dataset(user_type, query, results):
 def main():
     st.title("Semantic Text Retrieval Evaluation Interface")
 
-    # Initialize session state for reactions and results
+    # Initialize session state variables
+    if "search_performed" not in st.session_state:
+        st.session_state.search_performed = False
+    if "top_results" not in st.session_state:
+        st.session_state.top_results = []
     if "reactions" not in st.session_state:
         st.session_state.reactions = {}
-    if "results" not in st.session_state:
-        st.session_state.results = []
+    if "results_saved" not in st.session_state:
+        st.session_state.results_saved = False
 
     # Select device
     device = "cuda" if t.cuda.is_available() else "cpu"
@@ -74,41 +78,49 @@ def main():
             # Get top 3 results
             top_results_dot_product = t.topk(dot_scores, k=3)
 
+            # Store the top results indices in session_state
+            st.session_state.top_results = top_results_dot_product[1].tolist()
+            st.session_state.search_performed = True
+
+        # Display results and collect reactions
+        if st.session_state.search_performed and not st.session_state.results_saved:
             st.subheader("Query Results")
             st.write(f"Query: {query}")
 
-            # Clear previous results in session state
-            st.session_state.results = []
-
-            for _, idx in zip(top_results_dot_product[0], top_results_dot_product[1]):
+            for idx in st.session_state.top_results:
                 text = df.iloc[int(idx)]["ext"]
                 st.write(f"**Text:** {text}")
 
-                # Initialize reaction in session state if not already present
-                if f"reaction_{idx}" not in st.session_state.reactions:
-                    st.session_state.reactions[f"reaction_{idx}"] = "🤷"  # Default reaction
+                key = f"reaction_{idx}"
+                if key not in st.session_state.reactions:
+                    st.session_state.reactions[key] = "🤷"
 
-                # Display reaction options and update session state
                 reaction = st.radio(
                     label=f"Rate this result (Result {idx}):",
                     options=["👎", "🤷", "👍"],
-                    index=["👎", "🤷", "👍"].index(st.session_state.reactions[f"reaction_{idx}"]),
-                    key=f"reaction_{idx}",
+                    index=["👎", "🤷", "👍"].index(st.session_state.reactions[key]),
+                    key=key,
                     horizontal=True
                 )
-                st.session_state.reactions[f"reaction_{idx}"] = reaction  # Save selected reaction in session state
-
-                # Append reaction for this result
-                st.session_state.results.append({"text": text, "reaction": st.session_state.reactions[f"reaction_{idx}"]})
+                st.session_state.reactions[key] = reaction
 
-            # Save all reactions when the button is clicked
+            # Save reactions when the button is clicked
             if st.button("Save Reactions"):
-                save_reactions_to_dataset(user_type, query, st.session_state.results)
+                # Collect the results to save
+                results = []
+                for idx in st.session_state.top_results:
+                    key = f"reaction_{idx}"
+                    results.append({
+                        "text": df.iloc[int(idx)]["ext"],
+                        "reaction": st.session_state.reactions[key]
+                    })
+                save_reactions_to_dataset(user_type, query, results)
                 st.success("Reactions saved successfully!")
 
-                # Clear reactions and results from session state after saving
+                # Reset flags
+                st.session_state.search_performed = False
+                st.session_state.results_saved = True
                 st.session_state.reactions = {}
-                st.session_state.results = []
 
     except Exception as e:
         st.error(f"Failed to load database: {str(e)}")