Update app.py

app.py

@@ -66,6 +66,38 @@ if torch.cuda.is_available():
     tokenizer = AutoTokenizer.from_pretrained(model_id)
     tokenizer.use_default_system_prompt = False
 
+def fetch_personalized_data(session_index):
+    try:
+        # Connect to the database
+        with mysql.connector.connect(
+            host="18.153.94.89",
+            user="root",
+            password="N12RXMKtKxRj",
+            database="lionessdb"
+        ) as conn:
+            # Create a cursor object
+            with conn.cursor() as cursor:
+                # Query to fetch relevant data from both tables based on session_index
+                query = """
+                    SELECT e5390g37096_core.playerNr, 
+                           e5390g37096_core.groupNrStart, 
+                           e5390g37096_core.subjectNr, 
+                           e5390g37096_core.onPage
+                    FROM e5390g37096_core
+                    JOIN e5390g37096_decisions ON 
+                        e5390g37096_core.playerNr = e5390g37096_decisions.playerNr
+                    WHERE e5390g37096_decisions.session_index = %s
+                """
+                cursor.execute(query, (session_index,))
+                # Fetch data row by row
+                data = [{'playerNr': row[0], 'groupNrStart': row[1], 'subjectNr': row[2], 'onPage': row[3]} for row in cursor]
+                print(data)
+                return data
+    except mysql.connector.Error as err:
+        print(f"Error: {err}")
+        return None
+        
+'''
 def fetch_personalized_data(session_index):
         # Connect to the database
         conn = mysql.connector.connect(
@@ -85,8 +117,9 @@ def fetch_personalized_data(session_index):
         # Query to fetch relevant data from both tables based on session_index
         query = f"""
             SELECT e5390g37096_core.playerNr, 
-                   e5390g37096_core.groupNr, 
-                   e5390g37096_core.subjectNr 
+                   e5390g37096_core.groupNrStart, 
+                   e5390g37096_core.subjectNr, 
+                   e5390g37096_core.onPage
             FROM e5390g37096_core
             JOIN e5390g37096_decisions ON 
                 e5390g37096_core.playerNr = e5390g37096_decisions.playerNr
@@ -104,14 +137,15 @@ def fetch_personalized_data(session_index):
     
             # return [[str(row[0]), str(row[1]), str(row[2])] for row in rows]  # Convert each row to a list
             # Convert the rows to a list of dictionaries
-            data = [{'playerNr': row[0], 'groupNr': row[1], 'subjectNr': row[2]} for row in rows]
+            data = [{'playerNr': row[0], 'groupNrStart': row[1], 'subjectNr': row[2], 'onPage': row[3]} for row in rows]
             print(data)
             return data
     
         except mysql.connector.Error as err:
             print(f"Error: {err}")
             return None
-            
+'''
+
 def get_window_url_params():
         return """
             function() {
@@ -259,8 +293,10 @@ with gr.Blocks(css="style.css") as demo:
         #BINST, EINST = "[INST]", "[/INST]"
         BSYS, ESYS = "<<SYS>>\n", "\n<</SYS>>\n\n"
 
-        DEFAULT_SYSTEM_PROMPT = f""" You are an intelligent and fair game guide in a 2-player trust game, assisting players in making decisions to win. 
-        Answer in a consistent style. Answer per question should be maximum 2 sentences long. The players are called The Investor and The Dealer and keep their role throughout the whole game. 
+        DEFAULT_SYSTEM_PROMPT = f""" You are an intelligent and fair game guide in a 2-player trust game.
+        You are assisting players in making decisions to win. 
+        Answer in a consistent style. Each of your answers should be maximum 2 sentences long. 
+        The players are called The Investor and The Dealer and keep their role throughout the whole game. 
         Both start with 10€ in round 1. The game consists of 3 rounds. In round 1, The Investor invests between 0€ and 10€. 
         This amount is tripled automatically, and The Dealer can then distribute the tripled amount. After that, round 1 is over. 
         Both go into the next round with their current asset: The Investor with 10€ minus what he invested plus what he received back from The Dealer.