init aichat

Dockerfile

@@ -0,0 +1,6 @@
+FROM ghcr.io/open-webui/open-webui:main
+
+COPY sync_data.sh sync_data.sh
+
+RUN chmod -R 777 ./data && \
+    sed -i "1r sync_data.sh" ./start.sh

README.md

@@ -1,10 +1,9 @@
 ---
 title: AiChat
-emoji: 📉
-colorFrom: gray
+emoji: 😻
+colorFrom: purple
 colorTo: red
 sdk: docker
 pinned: false
----
-
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+app_port: 8080
+---

sync_data.sh

@@ -0,0 +1,144 @@
+#!/bin/bash
+
+# 检查必要的WebDAV环境变量
+if [ -z "$WEBDAV_URL" ] || [ -z "$WEBDAV_USERNAME" ] || [ -z "$WEBDAV_PASSWORD" ]; then
+    echo "缺少必要的WebDAV环境变量: WEBDAV_URL, WEBDAV_USERNAME 或 WEBDAV_PASSWORD"
+    exit 1
+fi
+
+# 去除WEBDAV_URL末尾的斜杠（如果有的话）
+WEBDAV_URL=$(echo "$WEBDAV_URL" | sed 's/\/$//')
+
+# 创建数据目录
+mkdir -p ./data/webdav
+
+check_webdav_webui_latest_exists() {
+    http_code=$(curl --user "$WEBDAV_USERNAME:$WEBDAV_PASSWORD" --head --silent --write-out "%{http_code}" "$WEBDAV_URL/OpenWebUI/webui_latest.db" -o /dev/null)
+    if [ "$http_code" -eq 200 ]; then
+        return 0  # 文件存在
+    else
+        return 1  # 文件不存在
+    fi
+}
+
+# 从WebDAV拉取最新的webui_latest.db文件
+download_latest_db() {
+    echo "检查云端是否存在webui_latest.db文件..."
+
+    if check_webdav_webui_latest_exists; then
+        echo "云端存在webui_latest.db文件，开始拉取..."
+        curl -o ./data/webdav/webui_latest.db \
+            --user "$WEBDAV_USERNAME:$WEBDAV_PASSWORD" \
+            "$WEBDAV_URL/OpenWebUI/webui_latest.db" && {
+            echo "最新的webui_latest.db文件拉取成功"
+            cp ./data/webdav/webui_latest.db ./data/webui.db
+        } || {
+            echo "拉取webui_latest.db文件失败"
+        }
+    else
+        echo "云端不存在webui_latest.db文件，跳过拉取"
+    fi
+}
+
+# 比较两个文件的hash值
+compare_hashes() {
+    local file1=$1
+    local file2=$2
+    local hash1
+    local hash2
+
+    # 检查文件是否存在
+    if [ ! -f "$file1" ] || [ ! -f "$file2" ]; then
+        echo "文件 $file1 或 $file2 不存在，无法比较哈希值"
+        return 1
+    fi
+
+    # 计算文件的hash值
+    hash1=$(sha256sum "$file1" | awk '{print $1}')
+    hash2=$(sha256sum "$file2" | awk '{print $1}')
+
+    # 比较hash值
+    if [ "$hash1" == "$hash2" ]; then
+        return 0 # 哈希值相同
+    else
+        return 1 # 哈希值不同
+    fi
+}
+
+# 上传备份文件到WebDAV
+upload_to_webdav() {
+    echo "检查是否需要上传文件到 WebDAV..."
+
+    # 如果 ./data/webui.db 文件不存在，跳过上传
+    if [ ! -f ./data/webui.db ]; then
+        echo "数据库未初始化，跳过上传"
+        return
+    fi
+
+    # 检查云端是否存在webui_latest.db文件
+    if check_webdav_webui_latest_exists; then
+        # 比较本地和云端的webui.db文件hash值
+        if compare_hashes ./data/webui.db ./data/webdav/webui_latest.db; then
+            echo "本地和云端的webui.db文件哈希值一致，跳过上传"
+            return
+        fi
+        echo "本地和云端的webui.db文件哈希值不同，准备上传..."
+    else
+        echo "云端不存在webui_latest.db文件，直接上传文件"
+    fi
+
+    # 上传最新的webui_latest.db到 WebDAV
+    cp ./data/webui.db ./data/webdav/webui_latest.db
+    curl -T ./data/webdav/webui_latest.db --user "$WEBDAV_USERNAME:$WEBDAV_PASSWORD" "$WEBDAV_URL/OpenWebUI/webui_latest.db" && {
+        echo "最新的webui_latest.db上传成功"
+    } || {
+        echo "上传webui_latest.db失败"
+    }
+
+    # 上传以日期命名的备份文件
+    FILENAME="webui_$(date +'%m_%d').db"
+    curl -T ./data/webdav/webui_latest.db --user "$WEBDAV_USERNAME:$WEBDAV_PASSWORD" "$WEBDAV_URL/OpenWebUI/$FILENAME" && {
+        echo "备份文件 $FILENAME 上传成功"
+    } || {
+        echo "上传备份文件 $FILENAME 失败"
+    }
+
+    # 获取当前 WebDAV 目录中的所有备份文件，按时间排序
+    echo "检查云端备份文件..."
+    backup_files=$(curl -s --user "$WEBDAV_USERNAME:$WEBDAV_PASSWORD" "$WEBDAV_URL/OpenWebUI/" | grep -o 'webui_.*\.db' | grep -v 'webui_latest.db' | sort)
+
+    # 如果备份文件超过 5 个，则删除最旧的一个
+    backup_count=$(echo "$backup_files" | wc -l)
+    if [ "$backup_count" -gt 5 ]; then
+        oldest_file=$(echo "$backup_files" | head -n 1)
+        echo "云端备份文件超过5个，删除最旧的备份文件: $oldest_file"
+
+        curl -X DELETE --user "$WEBDAV_USERNAME:$WEBDAV_PASSWORD" "$WEBDAV_URL/OpenWebUI/$oldest_file" && {
+            echo "删除最旧备份文件成功"
+        } || {
+            echo "删除最旧备份文件失败"
+        }
+    fi
+}
+
+# 定义同步函数
+sync_data() {
+    while true; do
+        echo "正在开始同步"
+
+        # 上传文件到WebDAV
+        upload_to_webdav
+
+        # 等待时间间隔
+        SYNC_INTERVAL=${SYNC_INTERVAL:-7200}
+        echo "当前时间 $(date '+%Y-%m-%d %H:%M:%S')"
+        echo "等待 ${SYNC_INTERVAL} 秒后进行下一次同步..."
+        sleep $SYNC_INTERVAL
+    done
+}
+
+# 初始拉取最新的webui_latest.db文件
+download_latest_db
+
+# 后台启动同步进程
+sync_data &