adding gen ai support

app.py

@@ -505,10 +505,14 @@ if st.button('Generate Predictions'):
                                 prefrence_variables,
                                 index=1
                             )
-                    if(preference):
-                        ai_generated_email=generate_example_email_with_context(email_body, campaign, industry, target, sorted_chars_out, preference)
-                        st.markdown('##### Here is the recommended Generated Email for you:')
-                        st.markdown('####### {}:'.format(ai_generated_email),unsafe_allow_html=True)
+                    # if(preference):
+                    if st.button('Generate AI Recommended Email'):
+                        if(preference is None):
+                            st.error('Please upload a email (HTML format)')
+                        else:
+                            ai_generated_email=generate_example_email_with_context(email_body, campaign, industry, target, sorted_chars_out, preference)
+                            st.markdown('##### Here is the recommended Generated Email for you:')
+                            st.markdown('####### {}:'.format(ai_generated_email),unsafe_allow_html=True)
 
 
                 # chars_out = dict(zip(chars, sel_var_values))

utils.py

@@ -2,6 +2,8 @@ import openai
 from io import BytesIO
 from config import config
 
+openai.api_key = config.OPEN_API_KEY
+
 
 def ask_chat_gpt(prompt, model=config.OPENAI_MODEL_TYPE, temp=0, max_tokens=500):
     response = openai.Completion.create(