code with stock familiarity in the decision making process

app.py

@@ -54,7 +54,6 @@ SYSTEM_INSTRUECTION_PREFERENCE_ELICITATION_FILE = os.path.join(
 SUMMARIZATION_PROMPT_FILE = os.path.join(STATIC_FILE, "txt/system_summarization_user_preference_elicitation.txt")
 
 uuid_this_session = str(uuid.uuid4())
-feedback_file_stock_familiarity = Path("user_feedback/") / f"stock_familiarity_{uuid_this_session}.json"
 feedback_file_interaction = Path("user_feedback/") / f"interaction_{uuid_this_session}.json"
 feedback_file_summarization = Path("user_feedback/") / f"summarization_{uuid_this_session}.json"
 feedback_file_round_evaluation = Path("user_feedback/") / f"round_evaluation_{uuid_this_session}.json"
@@ -180,29 +179,32 @@ def get_task_instruction_for_user(context):
     user_narrative = user_narrative.replace("\n", "<br>")
     html_user_narrative = markdown.markdown(user_narrative)
     general_instruction = GENERAL_INSTRUCTION
-    stock_instruction = f"""
+    round_instruction = f"""
 <div style="background-color: #f9f9f9; padding: 20px; border-radius: 10px; box-shadow: 0 4px 8px rgba(0, 0, 0, 0.1); margin-bottom: 20px; max-height: 780px; overflow-y: auto; overflow-x: hidden;">
     <!-- Stock Information (Bold label, Normal ticker name) -->
-    <h4 style="color: #2c3e50; text-align: center; margin-bottom: 20px; font-size: 20px; font-weight: 600;">
-        Stock Information
-    </h4>
+    <h2 style="color: #2c3e50; text-align: center; margin-bottom: 20px; font-size: 20px; font-weight: 600;">
+        Round Info
+    </h2>
+    <div style="text-align: left; font-size: 20px; font-weight: bold; margin-bottom: 20px;">
+        Stock
+    </div>
     <div style="text-align: left; font-weight: normal; font-size: 16px; margin-bottom: 20px;">
-        <span style="font-weight: bold;">This Round's Stock:</span>
+        <span style="font-weight: bold;">
+            This Round's Stock:
+        </span>
         {ticker_name}
     </div>
-</div>"""
-    user_narrative_instruction = f"""
-<div style="background-color: #f9f9f9; padding: 20px; border-radius: 10px; box-shadow: 0 4px 8px rgba(0, 0, 0, 0.1); margin-bottom: 20px; max-height: 780px; overflow-y: auto; overflow-x: hidden;">
+
     <!-- User Narrative (Bold label, Normal narrative) -->
-    <h4 style="color: #2c3e50; text-align: center; margin-bottom: 20px; font-size: 20px; font-weight: 600;">
+    <div style="text-align: left; font-size: 20px; font-weight: bold; margin-bottom: 20px;">
         User Narrative
-    </h4>
+    </div>
     <div style="text-align: left; font-weight: normal; font-size: 16px; margin-bottom: 20px;">
         {html_user_narrative}
     </div>
 </div>"""
 
-    return general_instruction, stock_instruction, user_narrative_instruction
+    return general_instruction, round_instruction
 
 
 def display_system_instruction_with_html(
@@ -278,35 +280,15 @@ def create_demo():
         english_order = ["1", "2", "3", "4", "5"]
         with gr.Tab(f"{english_order[order]}-1:Discuss"):
             general_instruction = gr.HTML(label="General Instruction")
-            with gr.Row():
-                with gr.Column():
-                    stock_instruction = gr.HTML(label="Candidate Stock")
-                with gr.Column():
-                    stock_familiarity = gr.Radio(
-                        [1, 2, 3, 4, 5, 6, 7],
-                        label="What is your level of familiarity with the candidate stock? (1 = Not Familiar, 7 = Very Familiar)",
-                    )
-
-            with gr.Row():
-                with gr.Column():
-                    with gr.Row():
-                        stock_familiarity_explain = gr.HTML(
-                            """<div style="background-color: #f8d7da; color: #721c24; padding: 15px; border: 1px solid #f5c6cb; border-radius: 5px; margin-bottom: 20px;">
-                    Please select your familiarity with the stock
-                    </div>"""
-                        )
-                with gr.Column():
-                    with gr.Row():
-                        stock_familiarity_button = gr.Button(value="Submit Stock Familiarity")
             with gr.Row():
                 with gr.Column():
                     with gr.Row():
-                        user_narrative_instruction = gr.HTML(label="User Narrative")
+                        round_instruction = gr.HTML(label="Round Instruction")
                 with gr.Column():
                     with gr.Row():
                         chatbot = gr.Chatbot(height=600)
                     with gr.Row():
-                        start_conversation = gr.Button(value="Start Conversation", interactive=False)
+                        start_conversation = gr.Button(value="Start Conversation")
                     with gr.Row():
                         msg = gr.Textbox(scale=1, label="User Input")
                     with gr.Row():
@@ -356,11 +338,7 @@ def create_demo():
             "confidence": confidence,
             "evaluation_send_button": evaluation_send_button,
             "general_instruction": general_instruction,
-            "stock_instruction": stock_instruction,
-            "stock_familiarity_explain": stock_familiarity_explain,
-            "stock_familiarity_button": stock_familiarity_button,
-            "stock_familiarity": stock_familiarity,
-            "user_narrative_instruction": user_narrative_instruction,
+            "round_instruction": round_instruction,
             "textbox": textbox,
         }
 
@@ -482,9 +460,6 @@ def create_demo():
             confidence,
             evaluation_send_button,
             textbox,
-            stock_familiarity,
-            stock_familiarity_button,
-            stock_familiarity_explain,
         ) = (
             tabs["comp"],
             tabs["system_instruction_context"],
@@ -499,19 +474,8 @@ def create_demo():
             tabs["confidence"],
             tabs["evaluation_send_button"],
             tabs["textbox"],
-            tabs["stock_familiarity"],
-            tabs["stock_familiarity_button"],
-            tabs["stock_familiarity_explain"],
         )
         system_instruction = ""
-        stock_familiarity_button.click(
-            lambda user_id, comp, stock_familiarity, stock_familiarity_button, start_conversation: respond_familiarity(
-                user_id, comp, stock_familiarity, stock_familiarity_button, start_conversation
-            ),
-            [user_id, comp, stock_familiarity, stock_familiarity_button, start_conversation],
-            [stock_familiarity_explain, stock_familiarity_button, start_conversation],
-        )
-
         start_conversation.click(
             lambda user_id, tab_session, history, comp, user_preference_elicitation_session, system_description_without_context, system_instruction_context: respond_start_conversation(
                 user_id,
@@ -864,31 +828,6 @@ def create_demo():
 
         return tab_data, "", history
 
-    def respond_familiarity(user_id, tab_name, stock_familiarity, stock_familiarity_button, start_conversation):
-        if stock_familiarity is None:
-            return (
-                """<div style="background-color: #f8d7da; color: #721c24; padding: 15px; border: 1px solid #f5c6cb; border-radius: 5px; margin-bottom: 20px;">
-                    <strong>Please select your familiarity with the stock.</strong>
-                    </div>""",
-                stock_familiarity_button,
-                start_conversation,
-            )
-        else:
-            save_feedback(
-                user_id,
-                uuid_this_session,
-                "familiarity",
-                {"stock": tab_name, "familiarity": stock_familiarity},
-                feedback_file_stock_familiarity,
-            )
-            return (
-                """<div style="background-color: #d4edda; color: #155724; padding: 15px; border: 1px solid #c3e6cb; border-radius: 5px; margin-bottom: 20px;">
-                        Thank you for submitting your familiarity rating. Please proceed by reading the user narrative, and then press ‘Start Conversation’ to begin the discussion.
-                    </div>""",
-                gr.Button(value="Familiarity Received", interactive=False),
-                gr.Button(value="Start Conversation", interactive=True),
-            )
-
     def respond_start_conversation(
         user_id,
         tab_data,
@@ -1214,10 +1153,8 @@ def create_demo():
     def get_context(index, raw_context_list, stock_context_list):
         comp = raw_context_list[index]["short_name"]
         context = stock_context_list[index]
-        general_instruction, stock_instruction, user_narrative_instruction = get_task_instruction_for_user(
-            raw_context_list[index]
-        )
-        return comp, context, general_instruction, stock_instruction, user_narrative_instruction
+        general_instruction, round_instruction = get_task_instruction_for_user(raw_context_list[index])
+        return comp, context, general_instruction, round_instruction
 
     def set_user_id(request: gr.Request):
         user_id = request.username
@@ -1244,34 +1181,18 @@ def create_demo():
         # User Narrative
         user_narrative = get_user_narrative_from_raw(raw_context_list[0]["user_narrative"])
         # Tab Context
-        (
-            first_comp,
-            first_context,
-            first_general_instruction,
-            first_stock_instruction,
-            first_user_narrative_instruction,
-        ) = get_context(0, raw_context_list, stock_context_list)
-        (
-            second_comp,
-            second_context,
-            second_general_instruction,
-            second_stock_instruction,
-            second_user_narrative_instruction,
-        ) = get_context(1, raw_context_list, stock_context_list)
-        (
-            third_comp,
-            third_context,
-            third_general_instruction,
-            third_stock_instruction,
-            thired_user_narrative_instruction,
-        ) = get_context(2, raw_context_list, stock_context_list)
-        (
-            fourth_comp,
-            fourth_context,
-            fourth_general_instruction,
-            fourth_stock_instruction,
-            fourth_user_narrative_instruction,
-        ) = get_context(3, raw_context_list, stock_context_list)
+        first_comp, first_context, first_general_instruction, first_round_instruction = get_context(
+            0, raw_context_list, stock_context_list
+        )
+        second_comp, second_context, second_general_instruction, second_round_instruction = get_context(
+            1, raw_context_list, stock_context_list
+        )
+        third_comp, third_context, third_general_instruction, third_round_instruction = get_context(
+            2, raw_context_list, stock_context_list
+        )
+        fourth_comp, fourth_context, fourth_general_instruction, fourth_round_instruction = get_context(
+            3, raw_context_list, stock_context_list
+        )
         # Final Evaluation
         ranking_first_comp = gr.Dropdown(choices=[1, 2, 3, 4, 5], label=first_comp)
         ranking_second_comp = gr.Dropdown(choices=[1, 2, 3, 4, 5], label=second_comp)
@@ -1288,23 +1209,19 @@ def create_demo():
             first_comp,
             first_context,
             first_general_instruction,
-            first_stock_instruction,
-            first_user_narrative_instruction,
+            first_round_instruction,
             second_comp,
             second_context,
             second_general_instruction,
-            second_stock_instruction,
-            second_user_narrative_instruction,
+            second_round_instruction,
             third_comp,
             third_context,
             third_general_instruction,
-            third_stock_instruction,
-            third_user_narrative_instruction,
+            third_round_instruction,
             fourth_comp,
             fourth_context,
             fourth_general_instruction,
-            fourth_stock_instruction,
-            fourth_user_narrative_instruction,
+            fourth_round_instruction,
             ranking_first_comp,
             ranking_second_comp,
             ranking_third_comp,
@@ -1360,10 +1277,9 @@ def create_demo():
         with gr.Tab("Financial Decision Stage") as financial_decision:
             # Experiment Tag
             first_tab = tab_creation_exploration_stage(0, first_comp, first_context)
-            first_general_instruction, first_stock_instruction, first_user_narrative_instruction = (
+            first_general_instruction, first_round_instruction = (
                 first_tab["general_instruction"],
-                first_tab["stock_instruction"],
-                first_tab["user_narrative_instruction"],
+                first_tab["round_instruction"],
             )
             click_control_exploration_stage(
                 first_tab,
@@ -1373,10 +1289,9 @@ def create_demo():
                 system_description_without_context,
             )
             second_tab = tab_creation_exploration_stage(1, second_comp, second_context)
-            second_general_instruction, second_stock_instruction, second_user_narrative_instruction = (
+            second_general_instruction, second_round_instruction = (
                 second_tab["general_instruction"],
-                second_tab["stock_instruction"],
-                second_tab["user_narrative_instruction"],
+                second_tab["round_instruction"],
             )
             click_control_exploration_stage(
                 second_tab,
@@ -1386,10 +1301,9 @@ def create_demo():
                 system_description_without_context,
             )
             third_tab = tab_creation_exploration_stage(2, third_comp, third_context)
-            third_general_instruction, third_stock_instruction, third_user_narrative_instruction = (
+            third_general_instruction, third_round_instruction = (
                 third_tab["general_instruction"],
-                third_tab["stock_instruction"],
-                third_tab["user_narrative_instruction"],
+                third_tab["round_instruction"],
             )
             click_control_exploration_stage(
                 third_tab,
@@ -1399,10 +1313,9 @@ def create_demo():
                 system_description_without_context,
             )
             fourth_tab = tab_creation_exploration_stage(3, fourth_comp, fourth_context)
-            fourth_general_instruction, fourth_stock_instruction, fourth_user_narrative_instruction = (
+            fourth_general_instruction, fourth_round_instruction = (
                 fourth_tab["general_instruction"],
-                fourth_tab["stock_instruction"],
-                fourth_tab["user_narrative_instruction"],
+                fourth_tab["round_instruction"],
             )
             click_control_exploration_stage(
                 fourth_tab,
@@ -1444,23 +1357,19 @@ def create_demo():
                 first_comp,
                 first_context,
                 first_general_instruction,
-                first_stock_instruction,
-                first_user_narrative_instruction,
+                first_round_instruction,
                 second_comp,
                 second_context,
                 second_general_instruction,
-                second_stock_instruction,
-                second_user_narrative_instruction,
+                second_round_instruction,
                 third_comp,
                 third_context,
                 third_general_instruction,
-                third_stock_instruction,
-                third_user_narrative_instruction,
+                third_round_instruction,
                 fourth_comp,
                 fourth_context,
                 fourth_general_instruction,
-                fourth_stock_instruction,
-                fourth_user_narrative_instruction,
+                fourth_round_instruction,
                 ranking_first_comp,
                 ranking_second_comp,
                 ranking_third_comp,