init

app.py

@@ -0,0 +1,74 @@
+import requests
+import gradio as gr
+import json
+import os
+
+# Env Vars
+METABASE_USERNAME = os.getenv('METABASE_USERNAME')
+METABASE_PASSWORD = os.getenv('METABASE_PASSWORD')
+
+def query_metabase(username, password, class_code, card_id, user_id):
+    try:
+        # 获取会话令牌
+        session_response = requests.post(
+            'https://metabase.cloud.junyiacademy.org/api/session',
+            headers={'Content-Type': 'application/json'},
+            json={'username': username, 'password': password}
+        )
+        session_response.raise_for_status()
+        session_token = session_response.json()['id']
+        print(f"Session token: {session_token}")
+
+        # 打印请求信息
+        request_payload = {
+            "parameters": [
+                {
+                    "type": "category",
+                    "target": ["variable", ["template-tag", "class_code"]],
+                    "value": class_code
+                },
+                {
+                    "type": "category",
+                    "target": ["variable", ["template-tag", "user_id"]],
+                    "value": user_id
+                }
+            ]
+        }
+        print(f"Request payload: {json.dumps(request_payload, indent=2)}")
+
+        # 使用提供的 card_id 查询 Metabase 卡片
+        query_response = requests.post(
+            f'https://metabase.cloud.junyiacademy.org/api/card/{card_id}/query/json',
+            headers={
+                'Content-Type': 'application/json',
+                'X-Metabase-Session': session_token
+            },
+            json=request_payload
+        )
+
+        # 打印响应状态码和内容
+        print(f"Response status code: {query_response.status_code}")
+        print(f"Response content: {query_response.text}")
+
+        query_response.raise_for_status()
+        return query_response.json()
+    except requests.RequestException as e:
+        print(f"Failed to query Metabase card: {str(e)}")
+        return {"error": f"Failed to query Metabase card: {str(e)}"}
+    except Exception as e:
+        print(f"Error: {str(e)}")
+        return {"error": str(e)}
+
+iface = gr.Interface(
+    fn=query_metabase,
+    inputs=[
+        gr.Textbox(label="Metabase Username", value=METABASE_USERNAME),
+        gr.Textbox(label="Metabase Password", type="password", value=METABASE_PASSWORD),
+        gr.Textbox(label="Class Code", value="class2"),
+        gr.Textbox(label="Card ID", value="6267"),
+        gr.Textbox(label="User ID", value="stu26")
+    ],
+    outputs="json"
+)
+
+iface.launch()

requirements.txt

@@ -0,0 +1,4 @@
+fastapi
+uvicorn
+requests
+gradio