Update app.py

app.py

@@ -176,4 +176,24 @@ Sarah
                     st.session_state.result_type = result_type
             else:
                 st.session_state.result = "Please enter an email body or select a sample to analyze."
-                st
+                st.session_state.result_type = ""
+    with col_clear:
+        if st.button("Clear", key="clear"):
+            st.session_state.email_body = ""
+            st.session_state.result = ""
+            st.session_state.result_type = ""
+            st.rerun()
+
+    # Display result with styled box
+    if st.session_state.result:
+        if st.session_state.result_type == "spam":
+            st.markdown(f'<div class="spam-result">{st.session_state.result}</div>', unsafe_allow_html=True)
+        elif st.session_state.result_type == "positive":
+            st.markdown(f'<div class="positive-result">{st.session_state.result}</div>', unsafe_allow_html=True)
+        elif st.session_state.result_type == "negative":
+            st.markdown(f'<div class="negative-result">{st.session_state.result}</div>', unsafe_allow_html=True)
+        else:
+            st.write(st.session_state.result)  # For error messages
+
+if __name__ == "__main__":
+    main()