ok

chat/apps.py

@@ -1,9 +1,6 @@
 from django.apps import AppConfig
 import google.generativeai as genai
 
-
-
-
 class MembersConfig(AppConfig):
     default_auto_field = 'django.db.models.BigAutoField'
     name = 'chat'

chat/consumers.py

@@ -1,19 +1,30 @@
 import json
 from . import model_manage2 as md
 from channels.generic.websocket import WebsocketConsumer
-
+from .database_manage import DataManage
 
 class ChatConsumer(WebsocketConsumer):
     def connect(self):
-        self.accept()
-        self.model, self.session = md.init_model("auto")
+        self.user = self.scope["user"]
+        if self.user.is_authenticated:
+            self.accept()
+            self.model, self.session = md.init_model("auto")
+        else:
+            self.close()
         
     def disconnect(self, close_code):
         del self.model, self.session
         pass
     
     def receive(self, text_data):
+        """
+        After socket create with:
+        username: self.user
+        roomid: self.roomid
+        """
         text_data_json = json.loads(text_data)
+        self.database = DataManage()
+        roomid = text_data_json["roomid"]
         message = text_data_json["messages"]
         print(message)
         question = message[-1]['content']
@@ -22,4 +33,5 @@ class ChatConsumer(WebsocketConsumer):
         print("Session history:")
         md.print_history(history_state)
         self.send(text_data=json.dumps({"message": response}))
+        self.database.Add_prompt_response(roomid,question,response)
         

chat/database_manage.py

@@ -0,0 +1,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 []

chat/migrations/0002_chatdetails_room_delete_member_chatdetails_roomid.py

@@ -0,0 +1,41 @@
+# Generated by Django 5.0.4 on 2024-05-21 07:21
+
+import django.db.models.deletion
+import uuid
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('chat', '0001_initial'),
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name='ChatDetails',
+            fields=[
+                ('order', models.AutoField(primary_key=True, serialize=False)),
+                ('UserPrompt', models.TextField()),
+                ('BotResponse', models.TextField()),
+            ],
+            options={
+                'ordering': ['order'],
+            },
+        ),
+        migrations.CreateModel(
+            name='Room',
+            fields=[
+                ('RoomID', models.UUIDField(default=uuid.uuid4, editable=False, primary_key=True, serialize=False)),
+                ('UserName', models.CharField(max_length=50)),
+            ],
+        ),
+        migrations.DeleteModel(
+            name='Member',
+        ),
+        migrations.AddField(
+            model_name='chatdetails',
+            name='RoomID',
+            field=models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, to='chat.room'),
+        ),
+    ]

chat/migrations/0003_room_name_alter_chatdetails_botresponse.py

@@ -0,0 +1,23 @@
+# Generated by Django 5.0.4 on 2024-05-21 12:36
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('chat', '0002_chatdetails_room_delete_member_chatdetails_roomid'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='room',
+            name='Name',
+            field=models.CharField(max_length=50, null=True),
+        ),
+        migrations.AlterField(
+            model_name='chatdetails',
+            name='BotResponse',
+            field=models.TextField(null=True),
+        ),
+    ]

chat/models.py

@@ -0,0 +1,23 @@
+from django.db import models
+from django.contrib.auth.models import AbstractBaseUser, BaseUserManager
+import uuid
+# ChatHistory Model
+class Room(models.Model):
+    RoomID = models.UUIDField(primary_key=True, default=uuid.uuid4, editable=False)
+    UserName = models.CharField(max_length=50)
+    Name = models.CharField(max_length=50,null=True)
+    def __str__(self):
+        return f'Chat {self.RoomID} by {self.UserName}'
+
+# ChatDetails Model
+class ChatDetails(models.Model):
+    RoomID = models.ForeignKey(Room, on_delete=models.CASCADE)
+    order = models.AutoField(primary_key=True)
+    UserPrompt = models.TextField()
+    BotResponse = models.TextField(null=True)
+
+    class Meta:
+        ordering = ['order']    
+
+    def __str__(self):
+        return f'Message {self.order} in Chat {self.RoomID}'

chat/templates/index.html

@@ -6,6 +6,14 @@
     <meta charset="UTF-8">
     <meta name="viewport" content="width=device-width, initial-scale=1.0">
     <script src="https://cdn.tailwindcss.com"></script>
+    <!-- <script>
+        // const roomdata = JSON.parse('{{ roomdata|safe }}');
+        var roomdata = JSON.parse('{{ roomdata|safe }}');
+        console.log(roomdata)
+        function call() {
+            return roomdata;
+        }
+    </script> -->
     <style>
         body {
             margin: 0;
@@ -13,16 +21,19 @@
             background-color: #2A2A2A;
             color: white;
         }
+
         .container {
             display: flex;
             height: 100vh;
         }
+
         .left-column {
             width: 25%;
             background-color: #3A3A3A;
             padding: 20px;
             overflow-y: auto;
         }
+
         .right-column {
             width: 75%;
             background-color: #2E2E2E;
@@ -34,230 +45,253 @@
 <script>
     function textAreaAdjust(element) {
         element.style.height = "1px";
-        element.style.height = (25+element.scrollHeight)+"px";
+        element.style.height = (25 + element.scrollHeight) + "px";
     }
 </script>
+
 <body>
-    <div class="container">
-    <div class="left-column">
-        <h2 class="text-xl font-medium mb-4">Lịch sử đoạn chat</h2>
-        <ul class="list-none">
-            <li class="p-2 bg-[#4A4A4A] mb-2 rounded cursor-pointer">Đoạn chat 1</li>
-            <li class="p-2 bg-[#4A4A4A] mb-2 rounded cursor-pointer">Đoạn chat 2</li>
-            <li class="p-2 bg-[#4A4A4A] mb-2 rounded cursor-pointer">Đoạn chat 3</li>
-            <!-- Thêm các đoạn chat khác tại đây -->
-        </ul>
-    </div>
-    
-    <div id="app" class="right-column flex flex-col md:mx-auto mx-0 bg-[#2E2E2E] w-full h-screen">
+    <div id="app" class="container">
+        <div class="left-column">
+            <h2 class="text-xl font-medium mb-4">Lịch sử đoạn chat</h2>
+            <ul class="list-none">
+                {% for x in rooms %}
+                    <li class="p-2 bg-[#4A4A4A] mb-2 rounded flex justify-between items-center">
+                        <a :href="`/chat?id={{x.id}}`" class="cursor-pointer">{{ x.name }}</a>
+                        <button @click="deleteChat(index)" class="bg-red-500 text-white px-2 py-1 rounded">
+                            Xoá
+                        </button>
+                    </li>
+                {% endfor %}
+            </ul>
+        </div>
 
-        <div class="flex flex-col h-full">
-            <!-- head -->
-            <div class="flex flex-row justify-between items-center p-4 select-none">
-                <h1 class="text-2xl font-medium text-center">
-                    TinyChat
-                </h1>
-                <div>
-                    <svg width="24" height="24" viewBox="0 0 24 24" fill="none" xmlns="http://www.w3.org/2000/svg"
-                        class="cursor-pointer" :class="{'text-white':!showConfig,'text-[#919191]':showConfig}"
-                        @click="showConfig=!showConfig">
-                        <path
-                            d="M11.6439 3C10.9352 3 10.2794 3.37508 9.92001 3.98596L9.49643 4.70605C8.96183 5.61487 7.98937 6.17632 6.935 6.18489L6.09966 6.19168C5.39095 6.19744 4.73822 6.57783 4.38385 7.19161L4.02775 7.80841C3.67338 8.42219 3.67031 9.17767 4.01968 9.7943L4.4315 10.5212C4.95126 11.4386 4.95126 12.5615 4.4315 13.4788L4.01968 14.2057C3.67031 14.8224 3.67338 15.5778 4.02775 16.1916L4.38385 16.8084C4.73822 17.4222 5.39095 17.8026 6.09965 17.8083L6.93501 17.8151C7.98938 17.8237 8.96184 18.3851 9.49644 19.294L9.92001 20.014C10.2794 20.6249 10.9352 21 11.6439 21H12.3561C13.0648 21 13.7206 20.6249 14.08 20.014L14.5035 19.294C15.0381 18.3851 16.0106 17.8237 17.065 17.8151L17.9004 17.8083C18.6091 17.8026 19.2618 17.4222 19.6162 16.8084L19.9723 16.1916C20.3267 15.5778 20.3298 14.8224 19.9804 14.2057L19.5686 13.4788C19.0488 12.5615 19.0488 11.4386 19.5686 10.5212L19.9804 9.7943C20.3298 9.17767 20.3267 8.42219 19.9723 7.80841L19.6162 7.19161C19.2618 6.57783 18.6091 6.19744 17.9004 6.19168L17.065 6.18489C16.0106 6.17632 15.0382 5.61487 14.5036 4.70605L14.08 3.98596C13.7206 3.37508 13.0648 3 12.3561 3H11.6439Z"
-                            stroke="white" stroke-width="2" stroke-linejoin="round" />
-                        <path
-                            d="M12 14.5C13.3807 14.5 14.5 13.3807 14.5 12C14.5 10.6193 13.3807 9.5 12 9.5C10.6193 9.5 9.5 10.6193 9.5 12C9.5 13.3807 10.6193 14.5 12 14.5Z"
-                            stroke="white" stroke-width="2" />
-                    </svg>
+        <div class="right-column flex flex-col md:mx-auto mx-0 bg-[#2E2E2E] w-full h-screen">
+
+            <div class="flex flex-col h-full">
+                <!-- head -->
+                <div class="flex flex-row justify-between items-center p-4 select-none">
+                    <h1 class="text-2xl font-medium text-center">
+                        TinyChat
+                    </h1>
+                    <div>
+                        <svg width="24" height="24" viewBox="0 0 24 24" fill="none" xmlns="http://www.w3.org/2000/svg"
+                            class="cursor-pointer" :class="{'text-white':!showConfig,'text-[#919191]':showConfig}"
+                            @click="showConfig=!showConfig">
+                            <path
+                                d="M11.6439 3C10.9352 3 10.2794 3.37508 9.92001 3.98596L9.49643 4.70605C8.96183 5.61487 7.98937 6.17632 6.935 6.18489L6.09966 6.19168C5.39095 6.19744 4.73822 6.57783 4.38385 7.19161L4.02775 7.80841C3.67338 8.42219 3.67031 9.17767 4.01968 9.7943L4.4315 10.5212C4.95126 11.4386 4.95126 12.5615 4.4315 13.4788L4.01968 14.2057C3.67031 14.8224 3.67338 15.5778 4.02775 16.1916L4.38385 16.8084C4.73822 17.4222 5.39095 17.8026 6.09965 17.8083L6.93501 17.8151C7.98938 17.8237 8.96184 18.3851 9.49644 19.294L9.92001 20.014C10.2794 20.6249 10.9352 21 11.6439 21H12.3561C13.0648 21 13.7206 20.6249 14.08 20.014L14.5035 19.294C15.0381 18.3851 16.0106 17.8237 17.065 17.8151L17.9004 17.8083C18.6091 17.8026 19.2618 17.4222 19.6162 16.8084L19.9723 16.1916C20.3267 15.5778 20.3298 14.8224 19.9804 14.2057L19.5686 13.4788C19.0488 12.5615 19.0488 11.4386 19.5686 10.5212L19.9804 9.7943C20.3298 9.17767 20.3267 8.42219 19.9723 7.80841L19.6162 7.19161C19.2618 6.57783 18.6091 6.19744 17.9004 6.19168L17.065 6.18489C16.0106 6.17632 15.0382 5.61487 14.5036 4.70605L14.08 3.98596C13.7206 3.37508 13.0648 3 12.3561 3H11.6439Z"
+                                stroke="white" stroke-width="2" stroke-linejoin="round" />
+                            <path
+                                d="M12 14.5C13.3807 14.5 14.5 13.3807 14.5 12C14.5 10.6193 13.3807 9.5 12 9.5C10.6193 9.5 9.5 10.6193 9.5 12C9.5 13.3807 10.6193 14.5 12 14.5Z"
+                                stroke="white" stroke-width="2" />
+                        </svg>
 
+                    </div>
                 </div>
-            </div>
-            <!-- config -->
-            <div v-show="showConfig" class="flex flex-col gap-4 p-4 select-none">
-                <!-- render configFields -->
-                <div v-for="field in configFields" :key="field.name" class="flex flex-col gap-2">
-                    <label class="text-sm text-gray-400">
-                        {{ field.label }}
-                    </label>
-                    <input v-if="field.type=='text'" type="text"
-                        class="bg-transparent border border-[#919191] rounded-lg p-2" v-model="config[field.name]" />
-                    <input v-if="field.type=='number'" type="number"
-                        class="bg-transparent border border-[#919191] rounded-lg p-2" v-model="config[field.name]" />
+                <!-- config -->
+                <div v-show="showConfig" class="flex flex-col gap-4 p-4 select-none">
+                    <!-- render configFields -->
+                    <div v-for="field in configFields" :key="field.name" class="flex flex-col gap-2">
+                        <label class="text-sm text-gray-400">
+                            {{ field.label }}
+                        </label>
+                        <input v-if="field.type=='text'" type="text"
+                            class="bg-transparent border border-[#919191] rounded-lg p-2"
+                            v-model="config[field.name]" />
+                        <input v-if="field.type=='number'" type="number"
+                            class="bg-transparent border border-[#919191] rounded-lg p-2"
+                            v-model="config[field.name]" />
+                    </div>
                 </div>
-            </div>
-            <!-- messages -->
-            <div class="flex flex-col overflow-y-auto scroll-smooth" id="messages">
-                <div v-for="(message,index) in messages" :key="message.id"
-                    class="flex flex-col odd:bg-[#F7F7F7]/10 group">
-                    <div class="p-4 flex flex-col justify-between gap-4">
-                        <div class="flex flex-row gap-4 capitalize">
-                            <div class="text-gray-400 text-sm" >
-                                <textarea
-                                    class="text-gray-400 text-sm size-full outline-0 bg-transparent border-none text-white rounded-lg resize-none"
-                                    :rows="message.role" v-model="message.role"></textarea>
-                            </div>
-                            <div class="flex flex-col w-full">
-                                <textarea
-                                onkeyup="textAreaAdjust(this)" style="overflow:hidden"
-                                    class="size-full outline-0 bg-transparent border-none text-white rounded-lg resize-y"
-                                    :rows="message.content.split('\n').length" v-model="message.content"></textarea>
-                                <div class="mt-4" v-if="index==messages.length-1 && message.role=='assistant'">
-                                    <svg stroke="currentColor" fill="none" stroke-width="1.5" viewBox="0 0 24 24"
-                                        stroke-linecap="round" stroke-linejoin="round" height="1em" width="1em"
-                                        xmlns="http://www.w3.org/2000/svg" v-if="index==messages.length-1"
-                                        class="group-hover:block cursor-pointer h-3 w-3" @click="refresh"
-                                        :class="{'animate-spin':isLoading}">
-                                        >
-                                        >
-                                        >
-                                        <polyline points="1 4 1 10 7 10"></polyline>
-                                        <polyline points="23 20 23 14 17 14"></polyline>
-                                        <path d="M20.49 9A9 9 0 0 0 5.64 5.64L1 10m22 4l-4.64 4.36A9 9 0 0 1 3.51 15">
-                                        </path>
+                <!-- messages -->
+                <div class="flex flex-col overflow-y-auto scroll-smooth" id="messages">
+                    <div v-for="(message,index) in messages" :key="message.id"
+                        class="flex flex-col odd:bg-[#F7F7F7]/10 group">
+                        <div class="p-4 flex flex-col justify-between gap-4">
+                            <div class="flex flex-row gap-4 capitalize">
+                                <div class="text-gray-400 text-sm">
+                                    <textarea
+                                        class="text-gray-400 text-sm size-full outline-0 bg-transparent border-none text-white rounded-lg resize-none"
+                                        :rows="message.role" v-model="message.role"></textarea>
+                                </div>
+                                <div class="flex flex-col w-full">
+                                    <textarea onkeyup="textAreaAdjust(this)" style="overflow:hidden"
+                                        class="size-full outline-0 bg-transparent border-none text-white rounded-lg resize-y"
+                                        :rows="message.content.split('\n').length" v-model="message.content"></textarea>
+                                    <div class="mt-4" v-if="index==messages.length-1 && message.role=='assistant'">
+                                        <svg stroke="currentColor" fill="none" stroke-width="1.5" viewBox="0 0 24 24"
+                                            stroke-linecap="round" stroke-linejoin="round" height="1em" width="1em"
+                                            xmlns="http://www.w3.org/2000/svg" v-if="index==messages.length-1"
+                                            class="group-hover:block cursor-pointer h-3 w-3" @click="refresh"
+                                            :class="{'animate-spin':isLoading}">
+                                            >
+                                            >
+                                            >
+                                            <polyline points="1 4 1 10 7 10"></polyline>
+                                            <polyline points="23 20 23 14 17 14"></polyline>
+                                            <path
+                                                d="M20.49 9A9 9 0 0 0 5.64 5.64L1 10m22 4l-4.64 4.36A9 9 0 0 1 3.51 15">
+                                            </path>
 
-                                    </svg>
+                                        </svg>
+                                    </div>
                                 </div>
                             </div>
                         </div>
                     </div>
                 </div>
             </div>
-        </div>
-        <!-- send message -->
-        <div class="flex flex-row justify-between items-center p-4 sticky bottom-0 bg-[#2E2E2E]" v-if="!showConfig">
-            <div class="w-full flex flex-row border border-1 border-[#919191] rounded-xl px-4 py-6 items-center">
-                <textarea
-                    class="w-full h-full outline-0 bg-transparent border-none text-white resize-none text-md max-h-[100px]"
-                    v-on:keyup.enter="sendMessage" placeholder="Type a message" :rows="message.split('\n').length"
-                    v-model="message"></textarea>
-                <div class="flex flex-col justify-end gap-4 text-[#2E2E2E] rounded-md p-1"
-                    :class="{'bg-white':message!='','bg-white/10':message==''}">
-                    <button @click="sendMessage">
-                        <svg class="w-6 h-6" fill="none" stroke="currentColor" viewBox="0 0 24 24"
-                            xmlns="http://www.w3.org/2000/svg">
-                            <path stroke-linecap="round" stroke-linejoin="round" stroke-width="2"
-                                d="M12 19l9 2-9-18-9 18 9-2zm0 0v-8" />
-                        </svg>
-                    </button>
+            <!-- send message -->
+            <div class="flex flex-row justify-between items-center p-4 sticky bottom-0 bg-[#2E2E2E]" v-if="!showConfig">
+                <div class="w-full flex flex-row border border-1 border-[#919191] rounded-xl px-4 py-6 items-center">
+                    <textarea
+                        class="w-full h-full outline-0 bg-transparent border-none text-white resize-none text-md max-h-[100px]"
+                        v-on:keyup.enter="sendMessage" placeholder="Type a message" :rows="message.split('\n').length"
+                        v-model="message"></textarea>
+                    <div class="flex flex-col justify-end gap-4 text-[#2E2E2E] rounded-md p-1"
+                        :class="{'bg-white':message!='','bg-white/10':message==''}">
+                        <button @click="sendMessage">
+                            <svg class="w-6 h-6" fill="none" stroke="currentColor" viewBox="0 0 24 24"
+                                xmlns="http://www.w3.org/2000/svg">
+                                <path stroke-linecap="round" stroke-linejoin="round" stroke-width="2"
+                                    d="M12 19l9 2-9-18-9 18 9-2zm0 0v-8" />
+                            </svg>
+                        </button>
+                    </div>
                 </div>
             </div>
-        </div>
-
-    </div>
-    <script type="module">
-        import { createApp } from 'https://unpkg.com/vue@3/dist/vue.esm-browser.js'
-        import { SSE } from "https://cdn.jsdelivr.net/npm/sse.js@2.1.0/lib/sse.min.js"
 
-        createApp({
-            data() {
-                return {
-                    showConfig: false,
-                    configFields: [
-                        {
-                            name: "endPoint",
-                            type: "text",
-                            label: "End Point"
-                        },
-                        {
-                            name: "apiKey",
-                            type: "text",
-                            label: "API Key"
-                        },
-                        {
-                            name: "model",
-                            type: "text",
-                            label: "Model"
-                        },
-                        {
-                            name: "temperature",
-                            type: "number",
-                            label: "Temperature"
-                        },
-                        {
-                            name: "presence_penalty",
-                            type: "number",
-                            label: "Presence Penalty"
-                        },
-                        {
-                            name: "top_p",
-                            type: "number",
-                            label: "Top P"
-                        },
-                        {
-                            name: "frequency_penalty",
-                            type: "number",
-                            label: "Frequency Penalty"
+        </div>
+        
+        <script type="module">
+            import { createApp } from 'https://unpkg.com/vue@3/dist/vue.esm-browser.js'
+            // import { SSE } from "https://cdn.jsdelivr.net/npm/sse.js@2.1.0/lib/sse.min.js"
+            
+            var roomdata = {{ roomdata|safe }};
+            
+            createApp({
+                data() {
+                    return {
+                        showConfig: false,
+                        configFields: [
+                            {
+                                name: "endPoint",
+                                type: "text",
+                                label: "End Point"
+                            },
+                            {
+                                name: "apiKey",
+                                type: "text",
+                                label: "API Key"
+                            },
+                            {
+                                name: "model",
+                                type: "text",
+                                label: "Model"
+                            },
+                            {
+                                name: "temperature",
+                                type: "number",
+                                label: "Temperature"
+                            },
+                            {
+                                name: "presence_penalty",
+                                type: "number",
+                                label: "Presence Penalty"
+                            },
+                            {
+                                name: "top_p",
+                                type: "number",
+                                label: "Top P"
+                            },
+                            {
+                                name: "frequency_penalty",
+                                type: "number",
+                                label: "Frequency Penalty"
+                            },
+                        ],
+                        isLoading: false,
+                        message: '',
+                        messages: roomdata.messages,
+                        // cs:roomdata,
+                        chatSocket: null,
+                        config: {
+                            endPoint: "",
+                            apiKey: "",
+                            model: "gpt-3.5-turbo",
+                            temperature: 1,
+                            presence_penalty: 0,
+                            top_p: 1,
+                            frequency_penalty: 0,
+                            stream: true
                         },
-                    ],
-                    isLoading: false,
-                    message: '',
-                    messages: [],
-                    
-                    chatSocket : new WebSocket(
-                    'ws://'
-                    + window.location.host
-                    + '/ws/chat'
-                    ),
-                    config: {
-                        endPoint: "",
-                        apiKey: "",
-                        model: "gpt-3.5-turbo",
-                        temperature: 1,
-                        presence_penalty: 0,
-                        top_p: 1,
-                        frequency_penalty: 0,
-                        stream: true
+                        // roomdata: roomdata
                     }
-                }
 
-            },
-            created() {
-                this.connect();
-            },
-            methods: {
-                refresh() {
-                    this.send(true);
                 },
-                sendMessage() {
-                    this.messages.push({
-                        role: "user",
-                        content: this.message
-                    });
-                    this.chatSocket.send(JSON.stringify({
-                        'message':this.message,
-                        'messages': this.messages,
-                    }));
-                    console.log(JSON.stringify({
-                        'messages': this.messages
-                    }))
-                    this.message = "";
+                created() {
+                    // const roomdata = JSON.parse('roomdata');
+                    const params = new URLSearchParams(window.location.search);
+                    this.id = params.get('id');//chat id
+                    if (this.id == null) {
+                        return
+                    }
+                    this.connect()
                 },
-                connect() {
-                    this.chatSocket.onopen = () => {
-                        this.connectionStatus = 'Connected';
-                        console.log('WebSocket connected');
-                    };
-
-                    this.chatSocket.onmessage = (event) => {
-                        var mes = JSON.parse(event.data)
+                methods: {
+                    refresh() {
+                        this.send(true);
+                    },
+                    sendMessage() {
+                        if (this.id == null) {
+                            window.location.href = '/chat/newchat?ques=' + this.message;
+                            return;
+                        }
                         this.messages.push({
-                            role: "model",
-                            content: mes.message
+                            role: "user",
+                            content: this.message
                         });
-                        console.log('Message received: ', event.data);
-                    };
+                        this.chatSocket.send(JSON.stringify({
+                            'roomid':this.id,
+                            'message': this.message,
+                            'messages': this.messages,
+                        }));
+                        console.log(JSON.stringify({
+                            'messages': this.messages
+                        }))
+                        this.message = "";
+                    },
+                    
+                    connect() {
+                        this.chatSocket = new WebSocket(
+                            'ws://' + window.location.host + '/ws/chat'
+                        );
 
-                    this.chatSocket.onclose = () => {
-                        this.connectionStatus = 'Disconnected';
-                        console.log('WebSocket disconnected');
-                    };
+                        this.chatSocket.onopen = () => {
+                            this.connectionStatus = 'Connected';
+                            console.log('WebSocket connected');
+                        };
 
-                    this.chatSocket.onerror = (error) => {
-                        console.error('WebSocket Error: ', error);
-                    };
-                }
-            },
-        }).mount('#app')
-    </script>
+                        this.chatSocket.onmessage = (event) => {
+                            var mes = JSON.parse(event.data)
+                            this.messages.push({
+                                role: "model",
+                                content: mes.message
+                            });
+                            console.log('Message received: ', event.data);
+                        };
+
+                        this.chatSocket.onclose = () => {
+                            this.connectionStatus = 'Disconnected';
+                            console.log('WebSocket disconnected');
+                        };
+
+                        this.chatSocket.onerror = (error) => {
+                            console.error('WebSocket Error: ', error);
+                        };
+                    }
+                },
+            }).mount('#app')
+        </script>
     </div>
 </body>
 

chat/urls.py

@@ -1,7 +1,10 @@
 from django.urls import path
 from . import views
 
+app_name = 'chat'
+
 urlpatterns = [
     path('', views.index, name='index'),
+    path('newchat/',views.newchat,name = 'newchat'),
     path("<str:room_name>/", views.room, name="room"),
 ]

chat/views.py

@@ -1,9 +1,39 @@
 # chat/views.py
-from django.shortcuts import render
-
-
+from django.shortcuts import render,redirect
+from django.http import HttpResponse
+from django.contrib.auth.decorators import login_required
+from .database_manage import DataManage
+import uuid
+import json
+@login_required
 def index(request):
-    return render(request, "index.html")
+    roomid = request.GET.get('id')
+    data = DataManage()
+    rooms = data.LoadRooms(username=request.user.username)
+    if roomid:
+        # try:
+            roomdata = data.LoadRoomDetail(roomid=roomid)
+            if roomdata==None:
+                raise("Room not found")
+            context = {
+                'rooms': rooms,
+                'roomdata': json.dumps(roomdata),
+            }
+            return render(request, 'index.html',context=context)
+        # except Exception as e:
+        #     print(e)
+        #     return HttpResponse("Room not found", status=404)
+    else:
+        roomdata = {'messages':[]}
+        return render(request, "index.html",{"rooms":rooms,'roomdata':roomdata})
+
+@login_required
+def newchat(request):
+    data = DataManage()
+    user = request.user.username
+    ques = request.GET.get('ques')
+    room = data.InsertRoom(user,ques[:23])
+    return redirect(f'/chat?id={room}&ques={ques}')
 
 
 def room(request, room_name):

chatbot_django/settings.py

@@ -22,6 +22,9 @@ BASE_DIR = Path(__file__).resolve().parent.parent
 # SECURITY WARNING: keep the secret key used in production secret!
 SECRET_KEY = 'django-insecure-l39x6jgx075p9bw8^px-c=pys%&$bzdo_sz_oabf#5bv@^isj9'
 
+LOGIN_URL = '/users/login/'
+
+
 # SECURITY WARNING: don't run with debug turned on in production!
 DEBUG = True
 

chatbot_django/urls.py

@@ -16,9 +16,14 @@ Including another URLconf
 """
 from django.contrib import admin
 from django.urls import path,include
+from django.shortcuts import redirect
+
+def redirect_to_chat(request):
+    return redirect('chat:index')
 
 urlpatterns = [
     path('chat/',include('chat.urls')),
     path('admin/', admin.site.urls),
-    path('users/', include('users.urls'))
+    path('users/', include('users.urls')),
+    path('',redirect_to_chat)
 ]

test.py

@@ -0,0 +1,3 @@
+a = [(1,2),(4,5),(3,5)]
+
+dic = [{"id1":x[0]},{"id":x[1]} for x in a]

users/views.py

@@ -8,14 +8,14 @@ def register_view(request):
         form = UserCreationForm(request.POST) 
         if form.is_valid(): 
             login(request, form.save())
-            return redirect("chat:room")
+            return redirect("chat:index")
     else:
         form = UserCreationForm()
     return render(request, "users/register.html", { "form": form })
 
 def login_view(request):
     if request.user.is_authenticated:
-        return redirect("../chat/room.html")
+        return redirect("chat:index")
     if request.method == "POST": 
         form = AuthenticationForm(data=request.POST)
         if form.is_valid(): 
@@ -25,15 +25,14 @@ def login_view(request):
             if 'next' in request.POST:
                 return redirect(request.POST.get('next'))
             else:
-                return redirect("chat.views.room")
+                return redirect("chat:index")
     else: 
         form = AuthenticationForm()
     return render(request, "users/login.html", { "form": form })
 
 def logout_view(request):
-    if request.method == "POST": 
-        logout(request) 
-        return redirect("chat:room")
+    logout(request) 
+    return redirect("chat:index")
     
 def index_view(request):
     return render(request, "users/index.html")