Update app.py

app.py

@@ -67,6 +67,52 @@ if torch.cuda.is_available():
     tokenizer = AutoTokenizer.from_pretrained(model_id)
     tokenizer.use_default_system_prompt = False
 
+def fetch_personalized_data(session_index):
+        # Connect to the database
+        conn = mysql.connector.connect(
+            host="18.153.94.89",
+            user="root",
+            password="N12RXMKtKxRj",
+            database="lionessdb"
+        )
+    
+        # Create a cursor object
+        cursor = conn.cursor()
+    
+        # Replace the placeholders with your actual database and table names
+        core_table = "e5390g37096_core"
+        decisions_table = "e5390g37096_decisions"
+    
+        # Query to fetch relevant data from both tables based on session_index
+        query = f"""
+            SELECT e5390g37096_core.playerNr, 
+                   e5390g37096_core.groupNr, 
+                   e5390g37096_core.subjectNr 
+            FROM e5390g37096_core
+            JOIN e5390g37096_decisions ON 
+                e5390g37096_core.playerNr = e5390g37096_decisions.playerNr
+            WHERE e5390g37096_decisions.session_index = '{session_index}'
+        """
+    
+        try:
+            cursor.execute(query)
+    
+            # Fetch all rows as lists of tuples
+            rows = cursor.fetchall()
+    
+            # Close the database connection
+            conn.close()
+    
+            # 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]
+            print(data)
+            return data
+    
+        except mysql.connector.Error as err:
+            print(f"Error: {err}")
+            return None
+            
 def get_window_url_params():
         return """
             function() {
@@ -180,62 +226,14 @@ examples=[
 
 with gr.Blocks(css="style.css") as demo:
     ## gradio-chatbot-read-query-param   
+    # session_index = 'eb3636167d3a63fbeee32934610e5b2f'
     url_params = gr.JSON({}, visible=False, label="URL Params")
     def get_session_index(url_params):
         session_index = url_params.get('session_index')
-        return session_index
-        
-    session_index = get_session_index(url_params.value)
-    print(session_index)
-
-    def fetch_personalized_data(session_index):
-        # Connect to the database
-        conn = mysql.connector.connect(
-            host="18.153.94.89",
-            user="root",
-            password="N12RXMKtKxRj",
-            database="lionessdb"
-        )
-    
-        # Create a cursor object
-        cursor = conn.cursor()
-    
-        # Replace the placeholders with your actual database and table names
-        core_table = "e5390g37096_core"
-        decisions_table = "e5390g37096_decisions"
-    
-        # Query to fetch relevant data from both tables based on session_index
-        query = f"""
-            SELECT e5390g37096_core.playerNr, 
-                   e5390g37096_core.groupNr, 
-                   e5390g37096_core.subjectNr 
-            FROM e5390g37096_core
-            JOIN e5390g37096_decisions ON 
-                e5390g37096_core.playerNr = e5390g37096_decisions.playerNr
-            WHERE e5390g37096_decisions.session_index = '{session_index}'
-        """
-    
-        try:
-            cursor.execute(query)
-    
-            # Fetch all rows as lists of tuples
-            rows = cursor.fetchall()
-    
-            # Close the database connection
-            conn.close()
-    
-            # 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]
-            print(data)
-            return data
-    
-        except mysql.connector.Error as err:
-            print(f"Error: {err}")
-            return None
-
-    personalized_data = fetch_personalized_data(session_index)
-    print(personalized_data)
+        print(session_index)
+        personalized_data = fetch_personalized_data(session_index)
+        print(personalized_data)
+        return personalized_data
   
     ## trust-game-llama-2-7b-chat
     # app.py