Merge pull request #146 from vanderbilt-data-science/115-regenerate-background

app.py

@@ -142,7 +142,10 @@ if authentication_status:
 
 
         with right_col: 
+            regenerate_background = False
             background_info = st.text_area("Background information on original post (This autopopulates for your reference). The Background information generated by OpenAI's GPT-4.", height = 780, value=st.session_state.background_info if 'background_info' in st.session_state else '', key = 'background_info_key')
+            if st.button("Request New Background Information"):
+                regenerate_background = True
 
         with left_col:
             original_post = st.text_area("Paste Original Post Here \n", height=100)
@@ -255,10 +258,43 @@ if authentication_status:
                             model_name='gpt-4', temperature=0.1)
 
                     if chat_mdl is not None:
-                        updated_response = regenerate_custom_response(
-                            chat_mdl, regenerate_prompt, st.session_state.draft_response).content
-                        st.session_state.draft_response = updated_response
+                        updated_response, background_info = regenerate_custom_response(
+                            original_post, chat_mdl, regenerate_prompt, st.session_state.draft_response, regenerate_background, st.session_state.background_text)
+                        st.session_state.draft_response = updated_response.content
                         st.session_state.is_regenerating = False
+                
+
+                    st.rerun()
+
+
+                if regenerate_background:
+                    if st.session_state.api_key:
+                        os.environ['OPENAI_API_KEY'] = st.session_state.api_key
+                        # add condition to check for passphrase to allow use of DSI api key stored in secrets
+                        if (os.environ["OPENAI_API_KEY"] == st.secrets["secret_passphrase"]):
+                            #umang key
+                            os.environ["OPENAI_API_KEY"] = st.secrets["dsi_openai_key"]
+                        elif (os.environ["OPENAI_API_KEY"] == st.secrets["secret_passphrase2"]):
+                            #abbie key
+                            os.environ["OPENAI_API_KEY"] = st.secrets["dsi_openai_key2"]
+                        elif (os.environ["OPENAI_API_KEY"] == st.secrets["secret_passphrase3"]):
+                            #myranda key
+                            os.environ["OPENAI_API_KEY"] = st.secrets["dsi_openai_key3"]
+                        elif (os.environ["OPENAI_API_KEY"] == st.secrets["secret_passphrase4"]):
+                            #jasmine key
+                            os.environ["OPENAI_API_KEY"] = st.secrets["dsi_openai_key4"]
+                        chat_mdl = ChatOpenAI(
+                            model_name='gpt-4', temperature=0.1)
+
+                    if chat_mdl is not None:
+                        updated_response, background_text = regenerate_custom_response(
+                            original_post, chat_mdl, regenerate_prompt, st.session_state.draft_response, regenerate_background, st.session_state.background_text)
+                        st.session_state.draft_response = updated_response.content
+                        st.session_state.background_text = background_info
+                        st.session_state.background_info = background_info # what is the difference between this and background_text?
+                        regenerate_background = False
+                        st.session_state.is_regenerating = False
+                
 
                     st.rerun()
 

free_speech_app/FreeSpeechPromptsResponses.py

@@ -113,12 +113,18 @@ def generate_custom_response(original_post, chat_mdl, principles=None, writing_s
     return draft_response, background_info
 
 # %% ../nbs/free-speech-prompts.ipynb 14
-def regenerate_custom_response(chat_mdl, regenerate_prompt, draft_response):
+def regenerate_custom_response(original_post, chat_mdl, regenerate_prompt, draft_response, regenerate_background=False, prev_background = None):
+    # regenerate background info
+    if regenerate_background:
+        __, background_info = generate_custom_prompt(original_post)
+    else:
+        background_info = prev_background
+
 
     # create customized prompt
-    customized_prompt = f"Please update the original response according to the following request. {regenerate_prompt}. Here is the original response: {draft_response}"
+    customized_prompt = f"Please update the original response according to the following request. {regenerate_prompt}. Here is the original response: {draft_response}. And background information for the original post: {background_info}"
 
     # get response
     updated_response = get_chat_model_response(chat_mdl, customized_prompt)
 
-    return updated_response
+    return updated_response, background_info