enable sidebar after predictions are made

ProtHGT_app.py

@@ -42,9 +42,16 @@ if 'previous_inputs' not in st.session_state:
 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
+def reset_prediction_state():
+    st.session_state.generating_predictions = False
+    st.session_state.submitted = False
+    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
 
 with st.expander("🚀 Upcoming Features"):
     st.info("""
@@ -249,7 +256,7 @@ with st.sidebar:
         if st.button("Generate Predictions", 
                     disabled=button_disabled, 
                     key="generate_predictions",
-                    on_click=set_generating_predictions):
+                    on_click=set_generating_predictions(True)):
             pass
 
         # Create a tuple of current inputs to track changes
@@ -312,7 +319,7 @@ if st.session_state.submitted:
 
             st.session_state.predictions_df = predictions_df
         
-            st.session_state.generating_predictions = False
+            set_generating_predictions(False)
 
         # Display and filter predictions
         st.success("Predictions generated successfully!")