Update app.py

app.py

@@ -66,6 +66,8 @@ if torch.cuda.is_available():
     tokenizer = AutoTokenizer.from_pretrained(model_id)
     tokenizer.use_default_system_prompt = False
 
+import mysql.connector
+
 def fetch_personalized_data(session_index):
     try:
         # Connect to the database
@@ -103,10 +105,8 @@ def fetch_personalized_data(session_index):
                     FROM e5390g37096_core
                     JOIN e5390g37096_decisions ON 
                         e5390g37096_core.playerNr = e5390g37096_decisions.playerNr
-                    WHERE e5390g37096_decisions.session_index = '{session_index}'
-
+                    WHERE e5390g37096_decisions.session_index = %s
                     UNION ALL
-
                     SELECT e5390g37096_core.playerNr, 
                            e5390g37096_core.groupNrStart, 
                            e5390g37096_core.subjectNr, 
@@ -136,8 +136,8 @@ def fetch_personalized_data(session_index):
                         FROM e5390g37096_core
                         JOIN e5390g37096_decisions 
                             ON e5390g37096_core.playerNr = e5390g37096_decisions.playerNr
-                        WHERE e5390g37096_decisions.session_index = '{session_index}'
-                    ) AND e5390g37096_decisions.session_index != '{session_index}'
+                        WHERE e5390g37096_decisions.session_index = %s
+                    ) AND e5390g37096_decisions.session_index != %s
                 """
                 cursor.execute(query, (session_index, session_index, session_index))
                 # Fetch data row by row
@@ -169,7 +169,7 @@ def fetch_personalized_data(session_index):
     except mysql.connector.Error as err:
         print(f"Error: {err}")
         return None
-        
+
 
 def get_window_url_params():
         return """