new function to disable sidebar when generating predictions

ProtHGT_app.py

@@ -38,9 +38,14 @@ if 'submitted' not in st.session_state:
     st.session_state.submitted = False
 if 'previous_inputs' not in st.session_state:
     st.session_state.previous_inputs = None
+# Initialize session state variables
 if 'generating_predictions' not in st.session_state:
     st.session_state.generating_predictions = False
 
+def set_generating_predictions():
+    st.session_state.generating_predictions = True
+    st.session_state.submitted = True
+
 with st.expander("🚀 Upcoming Features"):
     st.info("""
     We are actively working on enhancing ProtHGT application with new capabilities:
@@ -231,6 +236,14 @@ with st.sidebar:
     st.warning("⚠️ Due to memory and computational constraints, the maximum number of proteins that can be processed at once is limited to 100 proteins. For larger datasets, please consider running the model locally using our GitHub repository.")
 
     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):
+            pass
+
         # Create a tuple of current inputs to track changes
         current_inputs = (tuple(selected_proteins), selected_go_category)
         
@@ -246,8 +259,8 @@ with st.sidebar:
             st.session_state.submitted = True
 
 if st.session_state.submitted:
-    st.session_state.generating_predictions = True
     with st.spinner("Generating predictions..."):
+
         # Generate predictions only if not already in session state
         if st.session_state.predictions_df is None:
 
@@ -296,7 +309,8 @@ if st.session_state.submitted:
 
             st.session_state.predictions_df = predictions_df
         
-        st.session_state.generating_predictions = False
+            st.session_state.generating_predictions = False
+            
         # Display and filter predictions
         st.success("Predictions generated successfully!")
         st.markdown("### Filter and View Predictions")