add query bq testing function

app.py

@@ -1,8 +1,24 @@
+import os
+import json
+
 import gradio as gr
+import numpy as np
+from google.oauth2.service_account import Credentials
+from google.cloud import bigquery
 
 from theme import Seafoam
 
 seafoam = Seafoam()
+
+SCOPES = ["https://www.googleapis.com/auth/bigquery"]
+SERVICE_ACCOUNT_INFO = os.getenv("GBQ_TOKEN")
+service_account_info_dict = json.loads(SERVICE_ACCOUNT_INFO)
+
+creds = Credentials.from_service_account_info(
+        service_account_info_dict, scopes=SCOPES
+    )
+client = bigquery.Client(credentials=creds, project=service_account_info_dict['project_id'])
+
 with open('achievement.html', 'r', encoding='utf-8') as file:
     html = file.read()
 
@@ -116,11 +132,26 @@ with gr.Blocks(theme=seafoam, css=css) as demo:
         i = gr.Textbox(value='0', elem_id="test1")
         o = gr.Textbox()
         b = gr.Button(elem_id="test2")
+        df = gr.DataFrame()
 
     def print_text(t):
         return t
         
     b.click(print_text, i, o,)
 
+    def run_query(user_id):
+        QUERY = (
+        'SELECT created_at, points, user_role, joined, developer FROM `datastore_backup.UserData`',
+        f'WHERE user_id = {user_id}'
+        )
+        query_job = client.query(QUERY)
+        query_result = query_job.result()
+        df = query_result.to_dataframe()
+        # Select a subset of columns
+        df = df[["created_at", "points", "user_role", "joined", "developer"]]
+        # Convert numeric columns to standard numpy types
+        df = df.astype({"points": np.int64})
+        return df
+
 if __name__ == "__main__":
     demo.launch()

requirements.txt

@@ -0,0 +1,3 @@
+google-cloud-bigquery[pandas]
+numpy
+pandas