Update app.py

app.py

@@ -8,7 +8,7 @@ import csv
 from datetime import datetime
 from huggingface_hub import Repository, hf_hub_download
 
-def fetch_personalized_data():
+'''def fetch_personalized_data():
     # Connect to the database
     conn = mysql.connector.connect(
         host="18.153.94.89",
@@ -59,7 +59,7 @@ if personalized_data:
         print(row)  # Print each row
 else:
     print("No data found.")
-
+'''
 
 get_window_url_params = """
     function() {
@@ -98,63 +98,42 @@ with gr.Blocks() as demo:
             """
             # Connect to the database
             conn = mysql.connector.connect(
-            host="18.153.94.89",
-            user="root",
-            password="N12RXMKtKxRj",
-            database="lionessdb"
+                host="18.153.94.89",
+                user="root",
+                password="N12RXMKtKxRj",
+                database="lionessdb"
             )
-    
+        
             # Create a cursor object
-            cursor = conn.cursor()
-            
+            cursor = conn.cursor(dictionary=True)  # Use dictionary cursor for easier data manipulation
+        
             # Replace the placeholders with your actual database and table names
             core_table = "e5390g37096_core"
             decisions_table = "e5390g37096_decisions"
-            session_index = session_index
-            #print(session_index)
-            playerNr = "1"
-            
+        
             # Query to fetch relevant data from both tables based on session_index
             query = f"""
-                SELECT 
-                    {core_table}.playerNr, 
-                    {core_table}.subjectNr, 
-                    {core_table}.onPage,
-                    {decisions_table}.transfer1,
-                    {decisions_table}.tripledAmount1,
-                    {decisions_table}.keptForSelf1,
-                    {decisions_table}.returned1,
-                    {decisions_table}.newCreditRound2,
-                    {decisions_table}.transfer2,
-                    {decisions_table}.tripledAmount2,
-                    {decisions_table}.keptForSelf2,
-                    {decisions_table}.returned2,
-                    {decisions_table}.resultsWithoutAI
-                FROM {core_table}
-                JOIN {decisions_table} ON 
-                    {core_table}.playerNr = {decisions_table}.playerNr AND
-                    {core_table}.groupNr = {decisions_table}.groupNr AND
-                    {core_table}.subjectNr = {decisions_table}.subjectNr
-                WHERE {decisions_table}.session_index = '{session_index}'
+                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}'
             """
-            cursor.execute(query)
-            
-            # Fetch all rows and convert to a list of dictionaries
-            rows = cursor.fetchall()
-            result = []
-            for row in rows:
-                d = {}
-                for i, col in enumerate(cursor.description):
-                    d[col[0]] = row[i]
-                result.append(d)
         
-            # Convert the list of dictionaries to JSON
-            personalized_result = json.dumps(result)
-            print(personalized_result)
-
-            # Close the database connection
-            conn.close() 
-            
+            try:
+                cursor.execute(query)
+                # Fetch all rows as dictionaries
+                rows = cursor.fetchall()
+                print(rows)
+                # Close the database connection
+                conn.close()
+                return rows  # Return the fetched rows
+            except mysql.connector.Error as err:
+                print(f"Error: {err}")
+                return None
+                
         elif "print data" in history[-1][0]:
             personalized_data = fetch_personalized_data()
             print(personalized_data)