Merge pull request #102 from vanderbilt-data-science/chatgpt-background-info

free_speech_app/FreeSpeechPromptsResponses.py

@@ -1,7 +1,7 @@
 # AUTOGENERATED! DO NOT EDIT! File to edit: ../nbs/free-speech-prompts.ipynb.
 
 # %% auto 0
-__all__ = ['DEAFULT_PROMPT_TEMPLATE_TEXT', 'query_retriever', 'generate_custom_prompt', 'get_chat_model_response',
+__all__ = ['DEAFULT_PROMPT_TEMPLATE_TEXT', 'query_retriever', 'get_chat_model_response', 'call_chatGPT', 'generate_custom_prompt',
            'generate_custom_response', 'regenerate_custom_response']
 
 # %% ../nbs/free-speech-prompts.ipynb 5
@@ -61,11 +61,24 @@ def query_retriever(db, query, num_results = 3):
     return docs_as_one_string
 
 # %% ../nbs/free-speech-prompts.ipynb 10
+def get_chat_model_response(mdl, input_prompt):
+
+    messages = [HumanMessage(content=input_prompt)]
+
+    return mdl(messages)
+
+# %% ../nbs/free-speech-prompts.ipynb 11
+def call_chatGPT(query):
+    chat_model = ChatOpenAI(model_name = 'gpt-4', temperature=0.1)
+    response = get_chat_model_response(chat_model, query)
+
+    return response.content
+
+# %% ../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
-    ####
-    background_db = DeepLake(dataset_path="hub://vanderbilt-dsi/hatespeech-background", embedding = OpenAIEmbeddings())
+    
     sources_db = DeepLake(dataset_path="hub://vanderbilt-dsi/counterspeech-resources", embedding = OpenAIEmbeddings())
 
     # Use defaults in the case of None
@@ -80,7 +93,9 @@ def generate_custom_prompt(original_post, principles=None, writing_style=None, w
         word_limit="an infinite amount of"
 
     retriever_query = original_post
-    background_info = query_retriever(background_db, retriever_query)
+    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
@@ -88,14 +103,7 @@ def generate_custom_prompt(original_post, principles=None, writing_style=None, w
 
     return filled_prompt, background_info, sources
 
-# %% ../nbs/free-speech-prompts.ipynb 11
-def get_chat_model_response(mdl, input_prompt):
-
-    messages = [HumanMessage(content=input_prompt)]
-
-    return mdl(messages)
-
-# %% ../nbs/free-speech-prompts.ipynb 12
+# %% ../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
@@ -106,7 +114,7 @@ def generate_custom_response(original_post, chat_mdl, principles=None, writing_s
 
     return draft_response, background_info, sources
 
-# %% ../nbs/free-speech-prompts.ipynb 13
+# %% ../nbs/free-speech-prompts.ipynb 14
 def regenerate_custom_response(chat_mdl, regenerate_prompt, draft_response):
 
     # create customized prompt