Hugging Face's logo

Spaces:
qa1145
/
astrbbbb
Paused

App Files Community
astrbbbb
File size: 4,457 Bytes
8ede856
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
import { ref, computed } from 'vue';
import axios from 'axios';
import { useRouter } from 'vue-router';

export interface Conversation {
    cid: string;
    title: string;
    updated_at: number;
}

export function useConversations(chatboxMode: boolean = false) {
    const router = useRouter();
    const conversations = ref<Conversation[]>([]);
    const selectedConversations = ref<string[]>([]);
    const currCid = ref('');
    const pendingCid = ref<string | null>(null);

    // 编辑标题相关
    const editTitleDialog = ref(false);
    const editingTitle = ref('');
    const editingCid = ref('');

    const getCurrentConversation = computed(() => {
        if (!currCid.value) return null;
        return conversations.value.find(c => c.cid === currCid.value);
    });

    async function getConversations() {
        try {
            const response = await axios.get('/api/chat/conversations');
            conversations.value = response.data.data;

            // 处理待加载的会话
            if (pendingCid.value) {
                const conversation = conversations.value.find(c => c.cid === pendingCid.value);
                if (conversation) {
                    selectedConversations.value = [pendingCid.value];
                    pendingCid.value = null;
                }
            } else if (!currCid.value && conversations.value.length > 0) {
                // 默认选择第一个会话
                const firstConversation = conversations.value[0];
                selectedConversations.value = [firstConversation.cid];
            }
        } catch (err: any) {
            if (err.response?.status === 401) {
                router.push('/auth/login?redirect=/chatbox');
            }
            console.error(err);
        }
    }

    async function newConversation() {
        try {
            const response = await axios.get('/api/chat/new_conversation');
            const cid = response.data.data.conversation_id;
            currCid.value = cid;

            // 更新 URL
            const basePath = chatboxMode ? '/chatbox' : '/chat';
            router.push(`${basePath}/${cid}`);
            
            await getConversations();
            return cid;
        } catch (err) {
            console.error(err);
            throw err;
        }
    }

    async function deleteConversation(cid: string) {
        try {
            await axios.get('/api/chat/delete_conversation?conversation_id=' + cid);
            await getConversations();
            currCid.value = '';
            selectedConversations.value = [];
        } catch (err) {
            console.error(err);
        }
    }

    function showEditTitleDialog(cid: string, title: string) {
        editingCid.value = cid;
        editingTitle.value = title || '';
        editTitleDialog.value = true;
    }

    async function saveTitle() {
        if (!editingCid.value) return;

        const trimmedTitle = editingTitle.value.trim();
        try {
            await axios.post('/api/chat/rename_conversation', {
                conversation_id: editingCid.value,
                title: trimmedTitle
            });

            // 更新本地会话标题
            const conversation = conversations.value.find(c => c.cid === editingCid.value);
            if (conversation) {
                conversation.title = trimmedTitle;
            }
            editTitleDialog.value = false;
        } catch (err) {
            console.error('重命名对话失败:', err);
        }
    }

    function updateConversationTitle(cid: string, title: string) {
        const conversation = conversations.value.find(c => c.cid === cid);
        if (conversation) {
            conversation.title = title;
        }
    }

    function newChat(closeMobileSidebar?: () => void) {
        currCid.value = '';
        selectedConversations.value = [];
        
        const basePath = chatboxMode ? '/chatbox' : '/chat';
        router.push(basePath);
        
        if (closeMobileSidebar) {
            closeMobileSidebar();
        }
    }

    return {
        conversations,
        selectedConversations,
        currCid,
        pendingCid,
        editTitleDialog,
        editingTitle,
        editingCid,
        getCurrentConversation,
        getConversations,
        newConversation,
        deleteConversation,
        showEditTitleDialog,
        saveTitle,
        updateConversationTitle,
        newChat
    };
}