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] room_ids.reverse() return room_ids except Room.DoesNotExist: return []