Change ChatGPT prompt

Author: Praytic

__debug__.py

@@ -8,7 +8,7 @@ def get_json(self):
         return self.json_data
 
 def main():
-    data = {'gem_url': 'https://www.shouselaw.com/ca/labor/wrongful-termination/warn-act/', 'new_page_id': '610fd5b3-78a6-4119-96a4-dcc1ed3d1f3a'}
+    data = {'gem_url': 'https://www.balancepro.org/resources/articles/five-steps-to-smart-tax-management/', 'new_page_id': 'c7a5bebc-a76c-4f26-a7ae-3c8ddddd41ea'}
     describe_url(MockWebhook(data))
 
 if __name__ == '__main__':

main.py

@@ -51,11 +51,12 @@ def describe_url(webhook):
     categories = [option["name"] for option in options]
 
     # Generate the prompt
-    prompt = f"Describe this webpage ({gem_url}) using only the following tags: {', '.join(categories)}"
+    prompt = f"Choose the most relevant tags for this webpage {gem_url} from the following list: {', '.join(categories)}. You are not allowed to use any other tags."
     print("ChatGPT prompt:", prompt)
 
     # Get the response from the ChatGPT API
     chatgpt_answer = get_chatgpt_response(prompt)
+    print("ChatGPT answer:", chatgpt_answer)
 
     # Select Notion categories that were selected by ChatGPT
     tokens = chatgpt_answer.lower().rstrip(" ,.!/").split(", ")
@@ -67,7 +68,7 @@ def describe_url(webhook):
     request_payload = { "properties": { "Category": request_payload } }
 
     # Update the Notion database with the ChatGPT answer
-    print("Notion request payload:", pick(request_payload, "properties"))
+    print("Notion request payload for updated page:", pick(request_payload, "properties"))
     notion.pages.update(
         new_page_id,
         **request_payload