Improving the UI

app.py

@@ -45,17 +45,11 @@ def main():
 
 
 def perform_training(jammer_type, channel_switching_cost):
-    st.sidebar.write("======================")
-    st.sidebar.write('Training Starting')
     train(jammer_type, channel_switching_cost)
 
 
 def perform_testing(jammer_type, channel_switching_cost):
-    st.sidebar.write("======================")
-    st.sidebar.write('Testing Starting')
     test(jammer_type, channel_switching_cost)
-    st.sidebar.write("Testing completed")
-    st.sidebar.write("======================")
 
 
 if __name__ == "__main__":

trainer.py

@@ -81,8 +81,8 @@ def train(jammer_type, channel_switching_cost):
 
     # Display the Streamlit figure using streamlit.pyplot
     st.set_option('deprecation.showPyplotGlobalUse', False)
-    st.sidebar.subheader("Training Graph")
-    st.sidebar.pyplot(fig)
+    st.subheader("Training Graph")
+    st.pyplot(fig)
 
     # Save the figure
     plot_name = f'./data/train_rewards_{jammer_type}_csc_{channel_switching_cost}.png'