From 13e3e5dcc5e2fed79d0c09f46ee42897753d2497 Mon Sep 17 00:00:00 2001
From: lzp <zpli@aidlab.hk>
Date: Mon, 2 Mar 2026 13:51:14 +0800
Subject: [PATCH] fix

---
 .../FlowCanvas/components/cards/index.vue     |  1 +
 .../FlowCanvas/components/header-tools.vue    |  8 +++-
 .../components/tools/upload-file.vue          | 13 ++++++-
 .../Canvas/FlowCanvas/flow-canvas.vue         | 37 +++++++++++++++----
 .../Canvas/FlowCanvas/manager/NodeManager.ts  | 10 ++---
 .../Canvas/FlowCanvas/manager/StateManager.ts |  2 +-
 .../Canvas/FlowCanvas/tools/index.d.ts        |  8 ++++
 src/ignoredWarning.ts                         |  2 +-
 8 files changed, 62 insertions(+), 19 deletions(-)

diff --git a/src/components/Canvas/FlowCanvas/components/cards/index.vue b/src/components/Canvas/FlowCanvas/components/cards/index.vue
index 8986d18..9319afb 100644
--- a/src/components/Canvas/FlowCanvas/components/cards/index.vue
+++ b/src/components/Canvas/FlowCanvas/components/cards/index.vue
@@ -107,6 +107,7 @@
 	const componentRef = ref(null)
 	const onGenerateClick = () => {
 		const data = { ...componentRef.value.data }
+		if (attrs.node.data) attrs.node.data.data = data
 		nodeManager.createResultNode({
 			data: {
 				tier: currentComponent.value.tier,
diff --git a/src/components/Canvas/FlowCanvas/components/header-tools.vue b/src/components/Canvas/FlowCanvas/components/header-tools.vue
index 97eb499..51dc905 100644
--- a/src/components/Canvas/FlowCanvas/components/header-tools.vue
+++ b/src/components/Canvas/FlowCanvas/components/header-tools.vue
@@ -10,6 +10,10 @@
 			<span class="icon"><svg-icon name="export" size="11" /></span>
 			<span class="text">Export</span>
 		</button>
+		<button class="import" @click="emit('import')">
+			<span class="icon"><svg-icon name="import" size="11" /></span>
+			<span class="text">Import</span>
+		</button>
 	</div>
 </template>
 
@@ -19,7 +23,7 @@
 		zoom: { default: 1, type: Number },
 		step: { default: 0.1, type: Number }
 	})
-	const emit = defineEmits(['mouse', 'hand', 't', 'undo', 'redo', 'export'])
+	const emit = defineEmits(['mouse', 'hand', 't', 'undo', 'redo', 'export', 'import'])
 </script>
 
 <style lang="less" scoped>
@@ -60,7 +64,7 @@
 				background-color: #dfdfdf;
 			}
 		}
-		> .export {
+		> button {
 			width: 10rem;
 			height: 3rem;
 			border-radius: 0.4rem;
diff --git a/src/components/Canvas/FlowCanvas/components/tools/upload-file.vue b/src/components/Canvas/FlowCanvas/components/tools/upload-file.vue
index 5ac6a04..b4396e0 100644
--- a/src/components/Canvas/FlowCanvas/components/tools/upload-file.vue
+++ b/src/components/Canvas/FlowCanvas/components/tools/upload-file.vue
@@ -18,13 +18,22 @@
 	import { reactive, computed } from 'vue'
 	const emit = defineEmits(['update:modelValue', 'change'])
 	const props = defineProps({
-		modelValue: { type: [File, null] },
+		modelValue: { type: [File, Object, String, null] },
 		tip: { type: String, default: 'Upload your files' }
 	})
 	const data = reactive({
 		file: null
 	})
-	const url = computed(() => (props.modelValue ? URL.createObjectURL(props.modelValue) : ''))
+	const url = computed(() => {
+		const type = props.modelValue?.constructor
+		var str = ''
+		if (type === File) {
+			str = URL.createObjectURL(props.modelValue as File)
+		} else if (type === String) {
+			str = props.modelValue as string
+		}
+		return str
+	})
 	const onChange = (v) => {
 		emit('update:modelValue', v)
 		emit('change', v)
diff --git a/src/components/Canvas/FlowCanvas/flow-canvas.vue b/src/components/Canvas/FlowCanvas/flow-canvas.vue
index 32030a7..2c8ef01 100644
--- a/src/components/Canvas/FlowCanvas/flow-canvas.vue
+++ b/src/components/Canvas/FlowCanvas/flow-canvas.vue
@@ -14,7 +14,7 @@
 			<template v-for="v in nodeTypes" :key="v" #[`node-${v}`]="nodeProps">
 				<node :type="v" :stateManager="stateManager" :node="nodeProps">
 					<component
-						:is="nodeProps.data.component"
+						:is="components[nodeProps.data.component]"
 						:node="nodeProps"
 						:data="nodeProps.data.data"
 						v-bind="nodeProps.data"
@@ -23,7 +23,7 @@
 			</template>
 		</VueFlow>
 	</div>
-	<header-tools @export="exportFlow" />
+	<header-tools @export="exportFlow" @import="importFlow" />
 	<zoom
 		:zoom="stateManager.zoom.value"
 		:step="0.1"
@@ -34,13 +34,20 @@
 
 <script setup lang="ts">
 	import { VueFlow, useVueFlow } from '@vue-flow/core'
+	import { computed, ref, markRaw, onMounted, reactive, nextTick, provide } from 'vue'
 	import { useLayout } from '@/utils/treeDiagram'
-	import { NODE_TYPE } from './tools/index.d'
+	import { NODE_TYPE, NODE_COMPONENT } from './tools/index.d'
 	// 组件
 	import headerTools from './components/header-tools.vue'
 	import zoom from '../components/zoom.vue'
+	// 节点
 	import node from './components/node.vue'
-	import { computed, ref, markRaw, onMounted, reactive, nextTick, provide } from 'vue'
+	import resultImage from './components/result/result-image.vue'
+	import card from './components/cards/index.vue'
+	const components = {
+		[NODE_COMPONENT.RESULT_IMAGE]: resultImage,
+		[NODE_COMPONENT.CARD]: card
+	}
 
 	// 管理器
 	import { StateManager } from './manager/StateManager'
@@ -83,7 +90,7 @@
 	const { layout } = useLayout()
 	const index = ref(0)
 	async function layoutGraph(direction) {
-		if (index.value > 0) return
+		// if (index.value > 0) return
 		index.value++
 		setTimeout(() => {
 			stateManager.nodes.value = layout(
@@ -99,8 +106,24 @@
 
 	const exportFlow = () => {
 		// flowManager.exportFlow()
-		const json = JSON.parse(JSON.stringify(stateManager.nodes.value))
-		console.log(json)
+		const str = JSON.stringify(stateManager.nodes.value)
+		const json = JSON.parse(str)
+		localStorage.setItem('flow_json', str)
+	}
+	const importFlow = async () => {
+		try {
+			stateManager.nodes.value = []
+			await nextTick()
+			const json = JSON.parse(localStorage.getItem('flow_json') || '[]')
+			stateManager.nodes.value = json
+			setTimeout(() => {
+				nextTick(() => {
+					fitView({ maxZoom: 1 })
+				})
+			}, 0)
+		} catch (error) {
+			console.log(error)
+		}
 	}
 
 	onMounted(() => {
diff --git a/src/components/Canvas/FlowCanvas/manager/NodeManager.ts b/src/components/Canvas/FlowCanvas/manager/NodeManager.ts
index e363ab0..541b235 100644
--- a/src/components/Canvas/FlowCanvas/manager/NodeManager.ts
+++ b/src/components/Canvas/FlowCanvas/manager/NodeManager.ts
@@ -1,7 +1,5 @@
 import { createId } from '../../tools/tools'
-import card from '../components/cards/index.vue'
-import resultImage from '../components/result/result-image.vue'
-import { NODE_DATATYPE } from '../tools/index.d'
+import { NODE_DATATYPE, NODE_COMPONENT } from '../tools/index.d'
 interface NodeOptions {
 	id?: string
 	position?: { x: number, y: number }
@@ -51,7 +49,7 @@ export class NodeManager {
 	createResultNode(options?: NodeOptions) {
 		const options_ = {
 			...(options ? options : {}),
-			component: resultImage,
+			component: NODE_COMPONENT.RESULT_IMAGE,
 			data: {
 				tier: 0,
 				type: NODE_DATATYPE.RESULT_IMAGE,
@@ -65,7 +63,7 @@ export class NodeManager {
 	createCardsSelect(options?: NodeOptions) {
 		const options_ = {
 			...(options ? options : {}),
-			component: card,
+			component: NODE_COMPONENT.CARD,
 			positionY: 50,
 			data: {
 				tier: 1,
@@ -79,7 +77,7 @@ export class NodeManager {
 	createCardNode(options?: NodeOptions) {
 		const options_ = {
 			...(options ? options : {}),
-			component: card,
+			component: NODE_COMPONENT.CARD,
 		}
 		return this.createNode(options_)
 	}
diff --git a/src/components/Canvas/FlowCanvas/manager/StateManager.ts b/src/components/Canvas/FlowCanvas/manager/StateManager.ts
index 601243f..7795e4c 100644
--- a/src/components/Canvas/FlowCanvas/manager/StateManager.ts
+++ b/src/components/Canvas/FlowCanvas/manager/StateManager.ts
@@ -56,7 +56,7 @@ export class StateManager {
 						id: `el-${id}-${target}`,
 						source: id,
 						target: target,
-						type: 'output'
+						type: 'default'
 					})
 				}
 			})
diff --git a/src/components/Canvas/FlowCanvas/tools/index.d.ts b/src/components/Canvas/FlowCanvas/tools/index.d.ts
index 047b28c..ac7d26f 100644
--- a/src/components/Canvas/FlowCanvas/tools/index.d.ts
+++ b/src/components/Canvas/FlowCanvas/tools/index.d.ts
@@ -1,3 +1,11 @@
+/**  
+ * 组件
+*/
+export const NODE_COMPONENT = {
+	RESULT_IMAGE: 'result-image',
+	CARD: 'card',
+}
+
 /**
  * 节点类型
  */
diff --git a/src/ignoredWarning.ts b/src/ignoredWarning.ts
index aa98087..60d4a0c 100644
--- a/src/ignoredWarning.ts
+++ b/src/ignoredWarning.ts
@@ -1,6 +1,6 @@
 // 定义要忽略的警告关键词列表
 const ignoredWarnings = [
-	'`markRaw` or using `shallowRef` instead of `ref`',
+	// '`markRaw` or using `shallowRef` instead of `ref`',
 ]
 
 /** 忽略组件响应式警告 */