Update app.py

app.py

@@ -3,27 +3,49 @@ import time
 import random
 import json
 import mysql.connector
-import os
-import csv
-from datetime import datetime
-from huggingface_hub import Repository, hf_hub_download
-
-get_window_url_params = """
-    function() {
-        const params = new URLSearchParams(window.location.search);
-        const url_params = Object.fromEntries(params);
-        return url_params;
-        }
-    """
 
-get_window_session_index = """ 
-    function() {
-        const urlParams = new URLSearchParams(window.location.search);
-        const session_index = urlParams.get('session_index'); 
-        return 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 [list(row) for row in rows]  # Convert each row to a list
+
+    except mysql.connector.Error as err:
+        print(f"Error: {err}")
+        return None
+
 with gr.Blocks() as demo:
     gr.Markdown("""## Gradio send queryparam to chatbot 
     type `read query`
@@ -44,63 +66,32 @@ with gr.Blocks() as demo:
             here your URL params:
             {json.dumps(url_params, indent=4)}
             """
-            # 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(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"
-        
-            # 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 [list(row) for row in rows]  # Convert each row to a list
-        
-            except mysql.connector.Error as err:
-                print(f"Error: {err}")
-                return None
+            # Fetch personalized data
+            personalized_data = fetch_personalized_data(session_index)
+            if personalized_data:
+                bot_message = "Personalized data fetched successfully."
+                return [[bot_message, personalized_data]]
+            else:
+                bot_message = "Error fetching personalized data."
+                return [[bot_message, None]]
 
         elif "print data" in history[-1][0]:
             personalized_data = fetch_personalized_data()
             print(personalized_data)
             bot_message = "Data printed."
+            return [[bot_message, None]]
         else:
             bot_message = random.choice(["Yes", "No"])
         history[-1][1] = bot_message
         time.sleep(1)
-        return history         
+        return history
 
     msg.submit(user, inputs=[msg, url_params, chatbot], outputs=[msg, chatbot], queue=False).then(
         fn=bot, inputs=[chatbot, url_params], outputs=[chatbot]
     )
-    
+
     clear.click(lambda: None, None, chatbot, queue=False)
-    
+
     demo.load(
         fn=lambda x: x,
         inputs=[url_params],
@@ -108,5 +99,5 @@ with gr.Blocks() as demo:
         _js=get_window_url_params,
         queue=False
     )
-    
-demo.launch()
+
+demo.launch()