Update app.py

app.py

@@ -108,6 +108,8 @@ def main():
     
     # Check the model to use
     option = st.selectbox("Select a model to use:", models_available)
+    if option and "prediction" in st.session_state:
+        del st.session_state.prediction
 
     # Load the selected trained model
     if option in ("BERT-based model", "DistilBERT-based model (BERT light)"):
@@ -133,27 +135,6 @@ def main():
         if text.strip() == "":
             st.error("Please enter some text")
         else:
-            # # Add a progress bar
-            # progress_bar = st.progress(0)
-
-            # # Add a placeholder for the progress message
-            # status_text = st.empty()
-
-            # # Simulate a long-running process
-            # for i in range(100):
-            #     # Update the progress bar every 0.02 seconds
-            #     time.sleep(0.05)
-            #     progress_bar.progress(i + 1)
-                
-            #     if i % 2 == 0:
-            #         magic = "✨"
-            #     else:
-            #         magic = ""
-            #     # Update the progress message
-            #     status_text.write(f"Work in progress {i + 1}%... Wait for the magic 🪄🔮{magic}")
-            # # Clear the progress bar and status message
-            # progress_bar.empty()
-            # status_text.empty()
             with st.spinner("Wait for the magic 🪄🔮"):
                 # Use model
                 if option in ("Naive Bayes", "Logistic Regression"):
@@ -171,18 +152,12 @@ def main():
             st.session_state["text"] = text
             
             # Print result
-            # st.write(f"<span style='font-size: 24px;'>I think this text is: {prediction}</span>", 
-            #         unsafe_allow_html=True)
             st.markdown(f"I think this text is: **:{st.session_state['color_pred']}[{st.session_state['prediction']}]** (Prediction probability: {st.session_state['predict_proba'] * 100}%)")
-                   # help=f"I estimate that its probability is {st.session_state['predict_proba'] * 100}%")
 
     elif "prediction" in st.session_state:
-        # Display the stored result if available
-        # st.write(f"<span style='font-size: 24px;'>I think this text is: {st.session_state['prediction']}</span>", unsafe_allow_html=True)
-        
+        # Display the stored result if available        
         st.markdown(f"I think this text is: **:{st.session_state['color_pred']}[{st.session_state['prediction']}]** (Prediction probability: {st.session_state['predict_proba'] * 100}%)")
-                   #help=f"I estimate that its probability is {st.session_state['predict_proba'] * 100}%")
-        #**:blue[colored]**
+
     if st.button("Model Explanation"):
         # Check if there's text in the session state
         if "text" in st.session_state: