update ProtHGT_app.py

ProtHGT_app.py

@@ -48,7 +48,6 @@ def reset_prediction_state():
     st.session_state.predictions_df = None
     st.session_state.previous_inputs = None
 
-
 def set_generating_predictions():
     st.session_state.generating_predictions = True
     st.session_state.submitted = True
@@ -323,13 +322,14 @@ if st.session_state.submitted:
             # Reset the flags to release the sidebar
             st.session_state.generating_predictions = False
             st.session_state.submitted = False
+            st.rerun()
 
         # Display and filter predictions
         st.success("Predictions generated successfully!")
         st.markdown("### Filter and View Predictions")
         
         # Create filters
-        col1, col2, col3, col4 = st.columns(4)  # Changed to 4 columns
+        col1, col2, col3, col4 = st.columns(4)
         
         with col1:
             # Protein filter