fix sidebar enable/disable

ProtHGT_app.py

@@ -48,10 +48,10 @@ def reset_prediction_state():
     st.session_state.predictions_df = None
     st.session_state.previous_inputs = None
 
-    
-def set_generating_predictions(flag):
-    st.session_state.generating_predictions = flag
-    st.session_state.submitted = flag
+
+def set_generating_predictions():
+    st.session_state.generating_predictions = True
+    st.session_state.submitted = True
 
 with st.expander("🚀 Upcoming Features"):
     st.info("""
@@ -253,10 +253,11 @@ with st.sidebar:
     if selected_proteins and selected_go_category:
 
         button_disabled = st.session_state.submitted
+
         if st.button("Generate Predictions", 
                     disabled=button_disabled, 
                     key="generate_predictions",
-                    on_click=set_generating_predictions(True)):
+                    on_click=set_generating_predictions):
             pass
 
         # Create a tuple of current inputs to track changes
@@ -319,7 +320,9 @@ if st.session_state.submitted:
 
             st.session_state.predictions_df = predictions_df
         
-            set_generating_predictions(False)
+            # Reset the flags to release the sidebar
+            st.session_state.generating_predictions = False
+            st.session_state.submitted = False
 
         # Display and filter predictions
         st.success("Predictions generated successfully!")