modifying the prediction view

app.py

@@ -464,8 +464,12 @@ if st.button('Generate Predictions'):
                     # st.write(f"·Number of Characters: {int(Character_Count)}, Target Rate: {round(round(selected_variable_number, 3)*100, 3)}", "%")
                 st.write("\n")
                 df_modelpred=pd.DataFrame(list(zip(chars, sel_var_values)), columns=["Number of Characters", "Target Rate"])
-                st.checkbox("Use container width", value=False, key="use_container_width")
-                st.dataframe(df_modelpred.style.highlight_max(axis=0), use_container_width=st.session_state.use_container_width)
+                # st.checkbox("Use container width", value=False, key="use_container_width")
+                # st.dataframe(df_modelpred.style.highlight_max(axis=0), use_container_width=st.session_state.use_container_width)
+                df_modelpred.sort_values(by='Target Rate',inplace = True)
+                def color_coding(row):
+                    return ['background-color:yellow'] * len(row) if row.col2 ==df_modelpred['Target Rate'][0]  else ['background-color:yellow'] * len(row)
+                st.dataframe(df_modelpred.style.apply(color_coding, axis=1))
 
                 if len(chars) > 1:
                     #fig = plt.figure()