Update app.py

app.py

@@ -47,6 +47,24 @@ def main():
         #     ". "
         # )"""
 
+        # task_description = """You are a helpful assistant that generates annotated text for the st-annotated-text library in Python. Your task is to highlight the key terms that are central to the overall topic or theme of the sentence. Your output should be formatted in the following way:
+        # annotated_text(
+        #     "This ",
+        #     ("is", ""),
+        #     " some ",
+        #     ("annotated", ""),
+        #     ("text", ""),
+        #     " for those of ",
+        #     ("you", ""),
+        #     " who ",
+        #     ("like", ""),
+        #     " this sort of ",
+        #     ("thing", ""),
+        #     ". "
+        # )"""
+
+        # user_prompt = f"Now, please annotate this text: {user_text}"
+
         task_description = """You are a helpful assistant that generates annotated text for the st-annotated-text library in Python. Your task is to highlight the key terms that are central to the overall topic or theme of the sentence. Your output should be formatted in the following way:
         annotated_text(
             "This ",
@@ -63,12 +81,13 @@ def main():
             ". "
         )"""
 
-        user_prompt = f"Now, please annotate this text: {user_text}"
+        user_prompt = f"Based on the task description, please annotate the following text by highlighting the key terms central to its topic or theme: {user_text}"
+
         
         messages = [{"role": "system", "content": task_description}, {"role": "user", "content": user_prompt}]
         
         response = openai.ChatCompletion.create(
-            model="gpt-4",
+            model="gpt-3.5-turbo-16k",
             messages=messages,
             temperature=0.1,
             max_tokens=1000