Update app.py

app.py

@@ -1,4 +1,4 @@
-# Changed now everyting from session_index to PROLIFIC_PID 
+# Changed now everyting from session_index to externalID 
 
 # Original code from https://huggingface.co/spaces/huggingface-projects/llama-2-7b-chat
 # Modified for trust game purposes
@@ -87,7 +87,7 @@ if torch.cuda.is_available():
     tokenizer.use_default_system_prompt = False
 
 
-def fetch_personalized_data(PROLIFIC_PID):
+def fetch_personalized_data(externalID):
     try:
         # Connect to the database
         with mysql.connector.connect(
@@ -98,14 +98,14 @@ def fetch_personalized_data(PROLIFIC_PID):
         ) as conn:
             # Create a cursor object
             with conn.cursor() as cursor:
-                # Query to fetch relevant data from both tables based on PROLIFIC_PID      
+                # Query to fetch relevant data from both tables based on externalID = PROLIFIC_PID      
                 query = """
                     SELECT e5390g37623_core.playerNr, 
                            e5390g37623_core.groupNrStart, 
                            e5390g37623_core.subjectNr, 
                            e5390g37623_core.onPage, 
                            e5390g37623_core.role,
-                           e5390g37623_session.PROLIFIC_PID, 
+                           e5390g37623_session.externalID, 
                            e5390g37623_decisions.transfer1,
                            e5390g37623_decisions.tripledAmount1,
                            e5390g37623_decisions.keptForSelf1,
@@ -126,14 +126,14 @@ def fetch_personalized_data(PROLIFIC_PID):
                          e5390g37623_core.playerNr = e5390g37623_session.playerNr
                     JOIN e5390g37623_decisions ON 
                          e5390g37623_core.playerNr = e5390g37623_decisions.playerNr
-                    WHERE e5390g37623_session.PROLIFIC_PID = %s
+                    WHERE e5390g37623_session.externalID = %s
                     UNION ALL
                     SELECT e5390g37623_core.playerNr, 
                            e5390g37623_core.groupNrStart, 
                            e5390g37623_core.subjectNr, 
                            e5390g37623_core.onPage, 
                            e5390g37623_core.role,
-                           e5390g37623_session.PROLIFIC_PID, 
+                           e5390g37623_session.externalID, 
                            e5390g37623_decisions.transfer1,
                            e5390g37623_decisions.tripledAmount1,
                            e5390g37623_decisions.keptForSelf1,
@@ -159,10 +159,10 @@ def fetch_personalized_data(PROLIFIC_PID):
                         FROM e5390g37623_core
                         JOIN e5390g37623_session 
                             ON e5390g37623_core.playerNr = e5390g37623_session.playerNr
-                        WHERE e5390g37623_session.PROLIFIC_PID = %s
-                    ) AND e5390g37623_session.PROLIFIC_PID != %s
+                        WHERE e5390g37623_session.externalID = %s
+                    ) AND e5390g37623_session.externalID != %s
                 """
-                cursor.execute(query,(PROLIFIC_PID, PROLIFIC_PID, PROLIFIC_PID))
+                cursor.execute(query,(externalID, externalID, externalID))
                 # Fetch data row by row
                 data = [{
                     'playerNr': row[0], 
@@ -170,7 +170,7 @@ def fetch_personalized_data(PROLIFIC_PID):
                     'subjectNr': row[2], 
                     'onPage': row[3],
                     'role': row[4],  # Include the role here
-                    'PROLIFIC_PID': row[5],
+                    'externalID': row[5],
                     'transfer1': row[6],
                     'tripledAmount1': row[7],
                     'keptForSelf1': row[8],
@@ -281,18 +281,18 @@ def generate(
     #print(params)
     
     # Assuming params = request.query_params is the dictionary containing the query parameters    
-    # Extract the value of the 'PROLIFIC_PID' parameter
-    PROLIFIC_PID = params.get('PROLIFIC_PID')
+    # Extract the value of the 'externalID' parameter = PROLIFIC_PID   
+    PROLIFIC_PID = params.get('externalID')
     
-    # Check if PROLIFIC_PID_value is None or contains a value
+    # Check if PROLIFIC_PID value is None or contains a value
     if PROLIFIC_PID is not None:
-        print("Session index:", PROLIFIC_PID)
+        print("PROLIFIC_PID:", PROLIFIC_PID)
     else:
         PROLIFIC_PID = 'no_PROLIFIC_PID'
-        print("Session index not found or has no value.")
+        print("PROLIFIC_PID not found or has no value.")
 
     # Fetch personalized data
-    personalized_data = fetch_personalized_data(PROLIFIC_PID)
+    personalized_data = fetch_personalized_data(externalID)
     
     # Construct the input prompt using the functions from the system_prompt_config module
     input_prompt = construct_input_prompt(chat_history, message, personalized_data)
@@ -359,8 +359,8 @@ def generate(
     # Iterate over each dictionary in the list
     if personalized_data:
         for entry in personalized_data:
-            # Check if the PROLIFIC_PID matches the value in PROLIFIC_PID variable
-            if entry['PROLIFIC_PID'] == PROLIFIC_PID:
+            # Check if the externalID matches the value in externalID variable
+            if entry['externalID'] == PROLIFIC_PID:
                 onPage = entry.get('onPage', "no_onPage")  # Retrieve onPage value
                 playerNr = entry.get('playerNr', "no_playerNr")  # Retrieve playerNr value
                 groupNrStart = entry.get('groupNrStart', "groupNrStart")  # Retrieve groupNrStart value