add personality

app.py

@@ -53,9 +53,12 @@ SYSTEM_INSTRUCTION_NON_PERSONALIZATION_FILE = os.path.join(
     STATIC_FILE, "txt/system_instruction_non_personalization.txt"
 )
 SYSTEM_INSTRUCTION_PERSONALITY_FILE = os.path.join(STATIC_FILE, "txt/system_instruction_personality.txt")
-SYSTEM_INSTRUECTION_PREFERENCE_ELICITATION_FILE = os.path.join(
+SYSTEM_INSTRUCTION_PREFERENCE_ELICITATION_FILE = os.path.join(
     STATIC_FILE, "txt/system_instruction_preference_elicitation.txt"
 )
+SYSTEM_INSTRUCTION_PREFERENCE_ELICITATION_PERSONALITY_FILE = os.path.join(
+    STATIC_FILE, "txt/system_instruction_preference_elicitation_personality.txt"
+)
 SUMMARIZATION_PROMPT_FILE = os.path.join(STATIC_FILE, "txt/system_summarization_user_preference_elicitation.txt")
 PERSONALITY_EXT_FILE = os.path.join(STATIC_FILE, "txt/personality_ext.txt")
 PERSONALITY_INT_FILE = os.path.join(STATIC_FILE, "txt/personality_int.txt")
@@ -116,7 +119,10 @@ FINAL_EVALUATION = load_static_content(FINAL_EVALUATION_FILE)
 SYSTEM_INSTRUCTION_PERSONALIZATION = load_static_content(SYSTEM_INSTRUCTION_PERSONALIZATION_FILE)
 SYSTEM_INSTRUCTION_NON_PERSONALIZATION = load_static_content(SYSTEM_INSTRUCTION_NON_PERSONALIZATION_FILE)
 SYSTEM_INSTRUCTION_PERSONALITY = load_static_content(SYSTEM_INSTRUCTION_PERSONALITY_FILE)
-SYSTEM_INSTRUECTION_PREFERENCE_ELICITATION = load_static_content(SYSTEM_INSTRUECTION_PREFERENCE_ELICITATION_FILE)
+SYSTEM_INSTRUCTION_PREFERENCE_ELICITATION = load_static_content(SYSTEM_INSTRUCTION_PREFERENCE_ELICITATION_FILE)
+SYSTEM_INSTRUCTION_PREFERENCE_ELICITATION_PERSONALITY = load_static_content(
+    SYSTEM_INSTRUCTION_PREFERENCE_ELICITATION_PERSONALITY_FILE
+)
 SUMMARIZATION_PROMPT = load_static_content(SUMMARIZATION_PROMPT_FILE)
 PERSONALITY_EXT = load_static_content(PERSONALITY_EXT_FILE)
 PERSONALITY_INT = load_static_content(PERSONALITY_INT_FILE)
@@ -584,7 +590,9 @@ def create_demo():
             [tab_session, reason, likelihood, confidence, familiarity, evaluation_send_button, textbox],
         )
 
-    def click_control_preference_stage(tabs, user_id, user_preference_elicitation_session):
+    def click_control_preference_stage(
+        tabs, user_id, user_preference_elicitation_session, system_description_user_elicitation
+    ):
         (
             start_conversation,
             msg_button,
@@ -604,7 +612,7 @@ def create_demo():
                 user_id,
                 user_preference_elicitation_data,
                 history,
-                SYSTEM_INSTRUECTION_PREFERENCE_ELICITATION,
+                system_description_user_elicitation,
                 user_elicitation=True,
             ),
             [user_id, user_preference_elicitation_session, elicitation_chatbot],
@@ -616,7 +624,7 @@ def create_demo():
                 tab_data,
                 message,
                 tab_data["history"],
-                SYSTEM_INSTRUECTION_PREFERENCE_ELICITATION,
+                system_description_user_elicitation,
                 user_elicitation=True,
             ),
             [user_id, user_preference_elicitation_session, msg, elicitation_chatbot],
@@ -627,7 +635,7 @@ def create_demo():
                 user_id,
                 tab_data,
                 tab_data["history"],
-                SYSTEM_INSTRUECTION_PREFERENCE_ELICITATION,
+                system_description_user_elicitation,
                 user_elicitation=True,
             ),
             [user_id, user_preference_elicitation_session, elicitation_chatbot],
@@ -1117,6 +1125,14 @@ def create_demo():
         elif experiment_id == "3":
             return SYSTEM_INSTRUCTION_PERSONALITY.format(personality=PERSONALITY_INT)
 
+    def get_user_preference_elicitation(experiment_id):
+        if experiment_id == "0" or experiment_id == "1":
+            return SYSTEM_INSTRUCTION_PREFERENCE_ELICITATION
+        elif experiment_id == "2":
+            return SYSTEM_INSTRUCTION_PREFERENCE_ELICITATION_PERSONALITY.format(personality=PERSONALITY_EXT)
+        elif experiment_id == "3":
+            return SYSTEM_INSTRUCTION_PREFERENCE_ELICITATION_PERSONALITY.format(personality=PERSONALITY_INT)
+
     def get_stock_related_context(narrative_id, user_in_narrative_id):
         raw_context_list = build_raw_context_list(context_info_list[int(narrative_id)])
         stock_context_list = build_context(context_info_list[int(narrative_id)])
@@ -1129,6 +1145,9 @@ def create_demo():
         user_id, user_in_narrative_id, narrative_id, experiment_id = set_user_id(request)
         # System instruction without prompt
         system_description_without_context = get_inst_without_context(experiment_id)
+        # user_preference_elicitation
+        system_description_user_elicitation = get_user_preference_elicitation(experiment_id)
+        print(system_description_user_elicitation)
         # Stock related context
         raw_context_list, stock_context_list = get_stock_related_context(narrative_id, user_in_narrative_id)
         # User Narrative
@@ -1158,6 +1177,7 @@ def create_demo():
             narrative_id,
             experiment_id,
             system_description_without_context,
+            system_description_user_elicitation,
             raw_context_list,
             stock_context_list,
             user_narrative,
@@ -1190,6 +1210,7 @@ def create_demo():
         narrative_id = gr.State()
         experiment_id = gr.State()
         system_description_without_context = gr.State()
+        system_description_user_elicitation = gr.State()
         # Context data
         raw_context_list = gr.State()
         stock_context_list = gr.State()
@@ -1229,7 +1250,10 @@ def create_demo():
             user_preference_elicitation_tab = tab_creation_preference_stage()
             user_narrative = user_preference_elicitation_tab["user_narrative"]
             click_control_preference_stage(
-                user_preference_elicitation_tab, user_id, user_preference_elicitation_session
+                user_preference_elicitation_tab,
+                user_id,
+                user_preference_elicitation_session,
+                system_description_user_elicitation,
             )
         with gr.Tab("Financial Decision Stage") as financial_decision:
             # Experiment Tag
@@ -1309,6 +1333,7 @@ def create_demo():
                 narrative_id,
                 experiment_id,
                 system_description_without_context,
+                system_description_user_elicitation,
                 raw_context_list,
                 stock_context_list,
                 user_narrative,