back

app.py

@@ -1,23 +1,159 @@
-from dash import Dash, html
-import dash_cytoscape as cyto
-
-app = Dash(__name__)
-
-app.layout = html.Div([
-    html.P("Dash Cytoscape:"),
-    cyto.Cytoscape(
-        id='cytoscape',
-        elements=[
-            {'data': {'id': 'ca', 'label': 'Canada'}}, 
-            {'data': {'id': 'on', 'label': 'Ontario'}}, 
-            {'data': {'id': 'qc', 'label': 'Quebec'}},
-            {'data': {'source': 'ca', 'target': 'on'}}, 
-            {'data': {'source': 'ca', 'target': 'qc'}}
-        ],
-        layout={'name': 'breadthfirst'},
-        style={'width': '400px', 'height': '500px'}
+import networkx as nx
+import matplotlib.pyplot as plt
+import gradio as gr
+import io
+import base64
+import requests
+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)}
+
+# 创建一个DAG
+def create_knowledge_dag():
+    G = nx.DiGraph()
+    
+    # 添加节点
+    G.add_node("jnc-4-05-1-1", label="真分數、假分數與帶分數的命名及說、讀、聽、寫、做")
+    G.add_node("jnc-4-05-2-1", label="假分數與帶分數的互換")
+    G.add_node("jnc-4-05-3-1", label="同分母分數的大小比較")
+    G.add_node("jnc-4-05-3-2", label="同分母分數的加減")
+    G.add_node("jnc-4-05-3-3", label="分數的整數倍")
+    G.add_node("jnc-4-06-1-1", label="認識等值分數")
+    G.add_node("jnc-4-06-1-2", label="找出等值分數")
+    G.add_node("jnc-4-06-2-1", label="簡單異分母分數的比較")
+    G.add_node("jnc-4-06-2-2", label="簡單異分母分數的加減")
+    G.add_node("jnc-4-06-3-1", label="分數與一位小數的互換")
+    G.add_node("jnc-4-06-3-2", label="分數與二位小數的互換")
+    G.add_node("jnc-4-08-2-1", label="認識分數數線")
+    G.add_node("jnc-4-08-2-2", label="數線的整數、分數、小數")
+    
+    # 添加边
+    G.add_edge("jnc-4-05-1-1", "jnc-4-05-2-1")
+    G.add_edge("jnc-4-05-1-1", "jnc-4-05-3-1")
+    G.add_edge("jnc-4-05-2-1", "jnc-4-05-3-1")
+    G.add_edge("jnc-4-05-1-1", "jnc-4-05-3-2")
+    G.add_edge("jnc-4-05-2-1", "jnc-4-05-3-2")
+    G.add_edge("jnc-4-05-2-1", "jnc-4-05-3-3")
+    G.add_edge("jnc-4-05-3-2", "jnc-4-05-3-3")
+    G.add_edge("jnc-4-05-1-1", "jnc-4-06-1-1")
+    G.add_edge("jnc-4-06-1-1", "jnc-4-06-1-2")
+    G.add_edge("jnc-4-06-1-1", "jnc-4-06-2-1")
+    G.add_edge("jnc-4-06-1-2", "jnc-4-06-2-1")
+    G.add_edge("jnc-4-06-1-1", "jnc-4-06-2-2")
+    G.add_edge("jnc-4-06-1-2", "jnc-4-06-2-2")
+    G.add_edge("jnc-4-06-3-1", "jnc-4-06-3-2")
+    G.add_edge("jnc-4-05-1-1", "jnc-4-08-2-1")
+    G.add_edge("jnc-4-08-2-1", "jnc-4-08-2-2")
+    G.add_edge("jnc-4-05-1-1", "jnc-4-06-3-1")
+    
+    return G
+
+# 绘制DAG并返回图像数据
+def plot_knowledge_dag():
+    G = create_knowledge_dag()
+    
+    plt.figure(figsize=(10, 8))
+    pos = nx.spring_layout(G)
+    nx.draw(G, pos, with_labels=True, node_color='skyblue', node_size=2000, edge_color='gray', linewidths=2, font_size=10, font_weight='bold')
+    labels = nx.get_node_attributes(G, 'label')
+    nx.draw_networkx_labels(G, pos, labels, font_size=8, font_color='black')
+    
+    buf = io.BytesIO()
+    plt.savefig(buf, format='png')
+    buf.seek(0)
+    
+    return base64.b64encode(buf.read()).decode('utf-8')
+
+# 定义Gradio接口
+def show_knowledge_dag(result):
+    if 'error' in result:
+        return result['error']
+    
+    image_data = plot_knowledge_dag()
+    return f'<img src="data:image/png;base64,{image_data}"/>'
+
+with gr.Blocks() as app:
+    gr.Markdown("# Metabase Query and Visualization")
+
+    with gr.Row():
+        username = gr.Textbox(label="Metabase Username", value="purpleice9765@msn.com")
+        password = gr.Textbox(label="Metabase Password", type="password", value=METABASE_PASSWORD)
+
+    with gr.Row():
+        class_code = gr.Textbox(label="Class Code", value="class2")
+        card_id = gr.Textbox(label="Card ID", value="6267")
+        user_id = gr.Textbox(label="User ID", value="stu26")
+
+    result = gr.JSON(label="Query Result")
+    graph_html = gr.HTML()
+
+    query_button = gr.Button("Fetch Data")
+
+    query_button.click(
+        fn=query_metabase,
+        inputs=[username, password, class_code, card_id, user_id],
+        outputs=result
     )
-])
 
+    result.change(
+        fn=show_knowledge_dag,
+        inputs=result,
+        outputs=graph_html
+    )
 
-app.run_server(debug=True)
+app.launch()