Update app.py

app.py

@@ -207,6 +207,37 @@ def get_default_system_prompt(personalized_data):
     print(DEFAULT_SYSTEM_PROMPT)
     return DEFAULT_SYSTEM_PROMPT
 
+def map_onPage(onPage):
+    # Define the mapping of onPage values to onPage_filename and onPage_prompt
+    onPage_mapping = {
+        "stage407906.php": ("stage 1", "Welcome"),
+        "stage407908.php": ("stage 2", "Trust Game Instructions 1/3"),
+        "stage407909.php": ("stage 3", "Trust Game Instructions 2/3"),
+        "stage407915.php": ("stage 4", "Trust Game Instructions 3/3"),
+        "stage407923.php": ("stage 5", "Lobby with AI"),
+        "stage407924.php": ("stage 6", "Round 1: Investor’s turn with AI"),
+        "stage407925.php": ("stage 7", "Round 1: Dealer’s turn with AI"),
+        "stage407926.php": ("stage 8", "Round 2: Investor’s turn with AI"),
+        "stage407927.php": ("stage 9", "Round 2: Investor’s turn with AI"),
+        "stage407928.php": ("stage 10", "Results with AI after 2 rounds"),
+        "stage407929.php": ("stage 11", "Round 3: Investor’s turn with AI"),
+        "stage407930.php": ("stage 12", "Round 3: Dealer’s turn with AI"),
+        "stage407931.php": ("stage 13", "Overall Questionnaire"),
+        "stage407932.php": ("stage 14", "Results with AI after 3 rounds"),
+        "stage407933.php": ("stage 15", "Redirect to Prolific - Dropout no compensation"),
+        "stage407934.php": ("stage 16", "Not used yet: Redirect to Prolific - Win"),
+        "stage407935.php": ("stage 17", "Not used yet: Redirect to Prolific - Completion"),
+    }
+
+    # Check if onPage is in the mapping
+    if onPage in onPage_mapping:
+        onPage_filename, onPage_prompt = onPage_mapping[onPage]
+    else:
+        # If onPage is not in the mapping, set onPage_filename and onPage_prompt to "unknown"
+        onPage_filename, onPage_prompt = "unknown", "unknown"
+
+    return onPage_filename, onPage_prompt
+
 ## trust-game-llama-2-7b-chat
 # app.py
 def construct_input_prompt(chat_history, message, personalized_data):
@@ -330,45 +361,40 @@ def generate(
     
     # Print the onPage value
     print("onPage for session_index =", session_index, ":", onPage)
+    onPage_filename, onPage_prompt = map_onPage(onPage)
+    print("onPage_filename:", onPage_filename)
+    print("onPage_prompt:", onPage_prompt)
 
     # Save chat history to .csv file on HuggingFace Hub 
     # Generate filename with bot id and session id
-    filename = f"{session_index}_{onPage}_{DATA_FILENAME}"
+    filename = f"{session_index}_{DATA_FILENAME}"
     data_file = os.path.join(DATA_DIRECTORY, filename)
     
     # Generate timestamp
     timestamp = datetime.datetime.now()
-
+    
     # Check if the file already exists
     if os.path.exists(data_file):
         # If file exists, load existing data
         existing_data = pd.read_csv(data_file)
         
-        # Add timestamp, message and a readable sentence column 
+        # Add timestamp column
         conversation_df = pd.DataFrame(conversation)
-        #conversation_df = pd.DataFrame()
-        conversation_df['message'] = message
-        conversation_df['input_prompt'] = input_prompt
         conversation_df['readable_sentence'] = readable_sentence
         conversation_df['timestamp'] = timestamp
-
+        
         # Append new conversation to existing data
         updated_data = pd.concat([existing_data, conversation_df], ignore_index=True)
         updated_data.to_csv(data_file, index=False)
     else:
         # If file doesn't exist, create new file with conversation data
         conversation_df = pd.DataFrame(conversation)
-        #conversation_df = pd.DataFrame()
-        conversation_df['message'] = message
-        conversation_df['input_prompt'] = input_prompt
         conversation_df['readable_sentence'] = readable_sentence
         conversation_df['timestamp'] = timestamp
         conversation_df.to_csv(data_file, index=False)
 
     print("Updating .csv")
-
-timestamp = datetime.datetime.now()
-repo.push_to_hub(blocking=False, commit_message=f"Updating data at {timestamp}")
+    repo.push_to_hub(blocking=False, commit_message=f"Updating data at {timestamp}")
 
 chat_interface = gr.ChatInterface(
 fn=generate,