bug fixing for df_recom_extra

app.py

@@ -42,6 +42,21 @@ def create_image(width, height, rgb_color=(0, 0, 0)):
     image[:] = color
     return image
 
+def add_text(image, cta_txt):
+  font = cv2.FONT_HERSHEY_SIMPLEX    
+  # fontScale
+  fontScale = 1
+  # Blue color in BGR
+  color = (0, 0, 255)
+  
+  # Line thickness of 2 px
+  h=img.shape[0]/2
+  w=img.shape[1]/2
+  thickness = 2
+  image = cv2.putText(image, cta_txt, (h,w), font, 
+                   fontScale, color, thickness, cv2.LINE_AA)
+  return image
+
 def add_bg_from_url():
     st.markdown(
         f"""
@@ -439,7 +454,12 @@ def select_cta_button(ccolor, text):
     #st.write(st.session_state)
     buttons_out=[]
     for x in np.arange(len(st.session_state.ccolor)):
-        ctab=st.button("Call_To_Action text: "+str(st.session_state.text[x])+"; color: "+str(st.session_state.ccolor[x]), key = x)  
+        color_rgb=get_rgb(recom_cta)
+        color_img=create_image(100,30,color_rgb)
+        color_img=add_text(color_img,"Call_To_Action text: "+str(st.session_state.text[x]))
+        st.image(color_img, channels='BGR')
+        # ctab=st.button("Call_To_Action text: "+str(st.session_state.text[x])+"; color: "+str(st.session_state.ccolor[x]), key = x)
+        ctab=st.button("Select This CTA button to optimize", key = x) 
         res=[]
         res.append(x)
         val={}
@@ -663,6 +683,8 @@ def get_predictions(selected_variable, selected_industry, selected_campaign,
 #                     df_recom_opt_rank = df_recom_opt.nlargest(3, [selected_variable])
 
                     print(f"\nTo get a higher {selected_variable}, the model recommends the following options: ")
+                    st.markdown('##### To get a higher {}, the model recommends the following options:'.format(selected_variable), unsafe_allow_html=True)
+
                     if len(df_recom_opt_rank_out) < 2:
                         # st.markdown('##### Youve already achieved the highest {} with the current Call-To-Action Colors!'.format(selected_variable), unsafe_allow_html=True)
 #                         print("You've already achieved the highest", selected_variable, 
@@ -676,7 +698,7 @@ def get_predictions(selected_variable, selected_industry, selected_campaign,
                             color_img=create_image(100,30,color_rgb)
                             st.image(color_img, channels='BGR')
                             # st.markdown('##### recom_cta is: <span style="color:yellow">{}</span>'.format(recom_cta), unsafe_allow_html=True)
-                            st.markdown('##### target_rate for recom_cta is: <span style="color:yellow">{}</span>'.format(round(target_rate*100, 2)), unsafe_allow_html=True)
+                            st.markdown('##### target_rate for above recommended CTA button Color is: <span style="color:yellow">{}</span>'.format(round(target_rate*100, 2)), unsafe_allow_html=True)
 
 
                             print(f"  {(color('  ', fore='#ffffff', back=recom_cta))}  \x1b[1m{round(target_rate*100, 2)}%\x1b[22m")
@@ -689,7 +711,7 @@ def get_predictions(selected_variable, selected_industry, selected_campaign,
                             color_img=create_image(100,30,color_rgb)
                             st.image(color_img, channels='BGR')
                             # st.markdown('##### recom_cta is: {}'.format(recom_cta), unsafe_allow_html=True)
-                            st.markdown('##### target_rate for recom_cta is: <span style="color:yellow">{}</span>'.format(round(target_rate*100, 2)), unsafe_allow_html=True)
+                            st.markdown('##### target_rate for above recommended CTA button Color is: <span style="color:yellow">{}</span>'.format(round(target_rate*100, 2)), unsafe_allow_html=True)
                             print(f"  {(color('  ', fore='#ffffff', back=recom_cta))}  \x1b[1m{round(target_rate*100, 2)}%\x1b[22m")
 
                 elif selected_cta == 'Text':