Merge pull request #109 from vanderbilt-data-science/sources-to-prompt

app.py

@@ -79,11 +79,11 @@ if authentication_status:
         
         principles = st.text_input("My Principles (Paste Principles here)", placeholder = "Enter the main principles of your life you wish this response to uphold", value=user_data["principles"] if user_data and "principles" in user_data else "")
         writing_style = st.text_input("My Writing Style (Paste Examples)", placeholder = "Provide examples of your writing style here", value=user_data["writing_style"] if user_data and "writing_style" in user_data else "")
-        sources = st.text_area("Sources (This autopopulates for your reference)", value=st.session_state.sources if 'sources' in st.session_state else '', key = 'sources_key', height = 100)
+        #sources = st.text_area("Sources (This autopopulates for your reference)", value=st.session_state.sources if 'sources' in st.session_state else '', key = 'sources_key', height = 100)
     
         # Update button
         if st.button("Save Changes"):
-            db.put({"key": username, "principles": principles, "writing_style": writing_style, "sources": sources, "api_key": encrypted_api_key})
+            db.put({"key": username, "principles": principles, "writing_style": writing_style, "api_key": encrypted_api_key})
         
     if page == "OpenAI API Key Setup":
         st.title("OpenAI API Key Setup")
@@ -136,20 +136,20 @@ if authentication_status:
         
                 if chat_mdl is not None:
                     if user_data is None:
-                        draft_response, background_text, sources_text = generate_custom_response(original_post, chat_mdl, "", "")
+                        draft_response, background_text = generate_custom_response(original_post, chat_mdl, "", "")
                         st.session_state.draft_response = draft_response.content
                         st.session_state.background_text = background_text
-                        st.session_state.sources_text = sources_text
+                        #st.session_state.sources_text = sources_text
                         st.session_state.background_info = background_text
-                        st.session_state.sources = sources_text
+                        #st.session_state.sources = sources_text
                         st.rerun()
                     else:
-                        draft_response, background_text, sources_text = generate_custom_response(original_post, chat_mdl, user_data['principles'], user_data['writing_style'])
+                        draft_response, background_text = generate_custom_response(original_post, chat_mdl, user_data['principles'], user_data['writing_style'])
                         st.session_state.draft_response = draft_response.content
                         st.session_state.background_text = background_text
-                        st.session_state.sources_text = sources_text
+                        #st.session_state.sources_text = sources_text
                         st.session_state.background_info = background_text
-                        st.session_state.sources = sources_text
+                        #st.session_state.sources = sources_text
                         st.rerun()
                         
             

free_speech_app/FreeSpeechPromptsResponses.py

@@ -26,8 +26,12 @@ Next, here are some principles I consider particularly important to me:
 \n{principles}\n
 Here are some examples of the style in which I write:
 \n{writing_style}\n
-Finally, here are the sources to consider how to respond to hate speech:
-\n{sources}\n
+\n{writing_style}\n
+Finally, here are some best practices when responding to hate speech:
+1. Keep your response short and to the point
+2. Use either a humorous or empathetic tone - depending on the situation.
+    For instance, an empathetic response could be "This post is very painful for Jewish people to read".
+    A humorous response could be the use of sarcasm to discredit the hate speech.
 Using all the information I have provided, draft an appropriate response to the offensive post in question. Be sure to write as if you are posting on social media with no more than {word_limit} words. Do not acknowledge the original author of the hate speech post."""
 
 try:
@@ -39,13 +43,13 @@ try:
     
     PROMPT_TEMPLATE = PromptTemplate(
         template=PROMPT_TEMPLATE_TEXT,
-        input_variables=["original_post", "background_info", "principles", "writing_style", "sources", "word_limit"])
+        input_variables=["original_post", "background_info", "principles", "writing_style", "word_limit"])
 
 except FileNotFoundError:
     print(DEAFULT_PROMPT_TEMPLATE_TEXT)
     PROMPT_TEMPLATE = PromptTemplate(
         template=DEAFULT_PROMPT_TEMPLATE_TEXT,
-        input_variables=["original_post", "background_info", "principles", "writing_style", "sources", "word_limit"])
+        input_variables=["original_post", "background_info", "principles", "writing_style", "word_limit"])
 
 
 
@@ -76,10 +80,6 @@ def call_chatGPT(query):
 # %% ../nbs/free-speech-prompts.ipynb 12
 def generate_custom_prompt(original_post, principles=None, writing_style=None, word_limit=None):
 
-    # Get database and query retriever
-    
-    sources_db = DeepLake(dataset_path="hub://vanderbilt-dsi/counterspeech-resources", embedding = OpenAIEmbeddings())
-
     # Use defaults in the case of None
     
     if principles is None:
@@ -89,29 +89,28 @@ def generate_custom_prompt(original_post, principles=None, writing_style=None, w
         writing_style="I have no examples of my writing style."
 
     if word_limit is None:
-        word_limit="an infinite amount of"
+        word_limit="100"
 
     retriever_query = original_post
     background_query = "I found this post which I suspect to be hate speech. Can you please explain the background information I need to understand this post? Keep in mind terms/codes commonly used by extremist/hate groups such as Nazis, TERFS, alt-right, etc."
     background_query = background_query + "\n" + original_post 
     background_info = call_chatGPT(background_query)
-    sources = query_retriever(sources_db, retriever_query)
 
     # Fill the prompt
-    filled_prompt = PROMPT_TEMPLATE.format(original_post=original_post, background_info=background_info, principles=principles, writing_style=writing_style, sources=sources, word_limit=word_limit)
+    filled_prompt = PROMPT_TEMPLATE.format(original_post=original_post, background_info=background_info, principles=principles, writing_style=writing_style, word_limit=word_limit)
 
-    return filled_prompt, background_info, sources
+    return filled_prompt, background_info
 
 # %% ../nbs/free-speech-prompts.ipynb 13
 def generate_custom_response(original_post, chat_mdl, principles=None, writing_style=None, word_limit=None):
 
     # create customized prompt
-    customized_prompt, background_info, sources = generate_custom_prompt(original_post, principles, writing_style, word_limit)
+    customized_prompt, background_info = generate_custom_prompt(original_post, principles, writing_style, word_limit)
 
     # get response
     draft_response = get_chat_model_response(chat_mdl, customized_prompt)
 
-    return draft_response, background_info, sources
+    return draft_response, background_info
 
 # %% ../nbs/free-speech-prompts.ipynb 14
 def regenerate_custom_response(chat_mdl, regenerate_prompt, draft_response):