Update app.py

app.py

@@ -15,7 +15,23 @@ def main():
     if st.button("Find Keywords"):
         
         # few-shot learning을 이용한 task_description
-        # task_description = """You are a helpful assistant that generates annotated text for the st-annotated-text library in Python. Highlight the key terms that are most important in the context of the sentence. Your output should be formatted in the following way:
+        task_description = """You are a helpful assistant that generates annotated text for the st-annotated-text library in Python. 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", ""),
+            ". "
+        )"""
+
+        # task_description = """You are a helpful assistant that generates annotated text for the st-annotated-text library in Python. Highlight the key terms that are most important in the context of the sentence, especially focusing on the main subject involved. Your output should be formatted in the following way for text:
         # annotated_text(
         #     "This ",
         #     ("is", ""),
@@ -31,7 +47,7 @@ def main():
         #     ". "
         # )"""
 
-        # task_description = """You are a helpful assistant that generates annotated text for the st-annotated-text library in Python. Highlight the key terms that are most important in the context of the sentence, especially focusing on the main subject involved. Your output should be formatted in the following way for 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 ",
         #     ("is", ""),
@@ -47,6 +63,8 @@ def main():
         #     ". "
         # )"""
 
+        # 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,25 +81,7 @@ def main():
         #     ". "
         # )"""
 
-        # 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 ",
-            ("is", ""),
-            " some ",
-            ("annotated", ""),
-            ("text", ""),
-            " for those of ",
-            ("you", ""),
-            " who ",
-            ("like", ""),
-            " this sort of ",
-            ("thing", ""),
-            ". "
-        )"""
-
-        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}"
+        user_prompt = f"Based on the task description, please annotate the following text by highlighting the key terms central to its topic : {user_text}"
 
         
         messages = [{"role": "system", "content": task_description}, {"role": "user", "content": user_prompt}]