File size: 1,638 Bytes
46d747a
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
from .models import Room, ChatDetails
import uuid

class DataManage:
    def __init__(self):
        pass

    def DeleteRoom(self, roomid):
        try:
            room = Room.objects.get(RoomID=roomid)
            room.delete()
            return True
        except Room.DoesNotExist:
            return False

    def InsertRoom(self, username,roomname = "New chat"):
        new_room = Room(UserName=username,Name=roomname)
        new_room.save()
        return str(new_room.RoomID)

    def Add_prompt_response(self, roomid, prompt, response):
        try:
            room = Room.objects.get(RoomID=roomid)
            new_chat_detail = ChatDetails(RoomID=room, UserPrompt=prompt, BotResponse=response)
            new_chat_detail.save()
            return True
        except Room.DoesNotExist:
            return False

    def LoadRoomDetail(self, roomid):
        room = Room.objects.get(RoomID=roomid)
        print(room)
        print(room==None)
        chat_details = ChatDetails.objects.filter(RoomID=room.RoomID).order_by('order')
        print(chat_details)
        mes = []
        for x in chat_details:
            mes.append({'role':"user",'content':x.UserPrompt})
            mes.append({'role':"model",'content':x.BotResponse})

        return {'id':str(room.RoomID), 'user':room.UserName,'messages':mes}
    def LoadRooms(self,username):
        #return list of {id,name}
        try:
            rooms = Room.objects.filter(UserName=username)
            room_ids = [{'id':str(room.RoomID),'name':room.Name} for room in rooms]
            return room_ids
        except Room.DoesNotExist:
            return []