Upload 2 files

app.py

@@ -18,6 +18,28 @@ from pyvis.network import Network
 import pandas as pd
 import matplotlib.pyplot as plt
 
+
+from gqlalchemy import Memgraph
+import os
+
+MEMGRAPH_HOST = '3.76.194.251'
+MEMGRAPH_PORT = 7687
+MEMGRAPH_USERNAME = '[email protected]'
+MEMGRAPH_PASSWORD = os.environ.get("MEMGRAPH_PASSWORD", "<YOUR MEMGRAPH PASSWORD HERE>")
+
+def hello_memgraph():
+    try:
+        connection = Memgraph(MEMGRAPH_HOST, MEMGRAPH_PORT, MEMGRAPH_USERNAME, MEMGRAPH_PASSWORD, encrypted=True)
+        results = connection.execute_and_fetch(
+            'CREATE (n:FirstNode { message: "Hello Memgraph from Python!" }) RETURN n.message AS message'
+        )
+        return f"成功创建节点: {next(results)['message']}"
+    except Exception as e:
+        return f"连接失败: {str(e)}"
+
+
+
+
 # 配置日志
 logging.basicConfig(level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s")
 
@@ -820,4 +842,10 @@ with gr.Blocks(css="""
         import_output = gr.Textbox(label="导入日志")
         import_btn.click(fn=lambda: import_dataset(dataset_path.value), outputs=import_output)
 
+        gr.Markdown("### 测试 Memgraph 数据库连接")
+        memgraph_btn = gr.Button("测试 Memgraph 连接")
+        memgraph_output = gr.Textbox(label="连接测试日志")
+        memgraph_btn.click(fn=hello_memgraph, outputs=memgraph_output)
+
+
 demo.launch(server_name="0.0.0.0", server_port=7860)

requirements.txt

@@ -29,7 +29,9 @@ plotly>=5.10.0
 protobuf>=3.20.0
 accelerate>=0.20.0
 
-
 # 其他依赖
 requests>=2.25.0
-beautifulsoup4>=4.10.0
+beautifulsoup4>=4.10.0
+
+# gqlalchemy
+gqlalchemy>=0.3.1