the last samurai

chat/database_manage.py

@@ -36,8 +36,19 @@ class DataManage:
         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 LoadHistoryRoomForModel(self, roomid):
+        room = Room.objects.get(RoomID=roomid)
+        print(f"{room.RoomID} đã gọi chi tiết đoạn chat theo pattern context []")
+        chat_details = ChatDetails.objects.filter(RoomID=room.RoomID).order_by('order')
+        mes = []
+        for x in chat_details:
+            text = [{"text":str(x.UserPrompt)}]
+            mes.append({'role':"user",'parts':text})
+            text = [{"text":str(x.BotResponse)}]
+            mes.append({'role':"model",'parts':text})
+        return mes
     def LoadRooms(self,username):
         #return list of {id,name}
         try:

chat/templates/index.html

@@ -78,7 +78,7 @@
                 <!-- head -->
                 <div class="flex flex-row justify-between items-center p-4 select-none">
                     <h1 class="text-2xl font-medium text-center">
-                        TinyChat
+                        <a :href="`/chat`">Arxiv Chatbot</a>
                     </h1>
                     <div>
                         <svg width="24" height="24" viewBox="0 0 24 24" fill="none" xmlns="http://www.w3.org/2000/svg"
@@ -217,6 +217,7 @@
                             },
                         ],
                         isLoading: false,
+                        isFirst:true,
                         message: '',
                         messages: roomdata.messages,
                         // cs:roomdata,
@@ -244,8 +245,9 @@
                     }
                     // this.connect()
                     this.ques = params.get('ques');
-                    if (this.ques != null) {
+                    if (this.ques != null && this.isFirst==true) {
                         this.message = this.ques;
+                        this.isFirst = false;
                         this.sendMessageAPI();
                     }
                 },

chat/views.py

@@ -51,23 +51,25 @@ def deletechat(request):
 
 @method_decorator(csrf_exempt, name='dispatch')
 class MessageView(View):
-    def __init__(self) -> None:
-        super().__init__()
-        self.model = None
-        self.session = None
-        self.database = None
-
     def post(self, request):
-        if not self.model:
-            self.model, self.session = md.init_model("auto")
         text_data_json = json.loads(request.body)
-        self.database = DataManage()
-        print(text_data_json)
         roomid = text_data_json["roomid"]
         message = text_data_json["messages"]
+        self.database = DataManage()
+        history = self.database.LoadHistoryRoomForModel(roomid)
+        model, session = md.init_model("auto",history=history)
+        # if 'session' not in request.session:
+        #     history = self.database.LoadHistoryRoomForModel(roomid)
+        #     model, session = md.init_model("auto",history=history)
+        #     # request.session['model'] = model
+        #     request.session['session'] = session
+        # else:
+        #     # model = request.session['model']
+        #     session = request.session['session']
+        print(text_data_json)
         print(message)
         question = message[-1]['content']
-        response, history_state = md.full_chain_history_question(question, self.session, mode="auto")
+        response, history_state = md.full_chain_history_question(question, session, mode="auto")
         # print("First answer: ",response)
         print("Session history:")
         md.print_history(history_state)

chatbot_django/settings.py

@@ -15,6 +15,10 @@ from pathlib import Path
 # Build paths inside the project like this: BASE_DIR / 'subdir'.
 BASE_DIR = Path(__file__).resolve().parent.parent
 
+SESSION_ENGINE = 'django.contrib.sessions.backends.db' 
+
+# settings.py
+SESSION_EXPIRE_AT_BROWSER_CLOSE = True  # Cấu hình session hết hạn khi trình duyệt đóng
 
 # Quick-start development settings - unsuitable for production
 # See https://docs.djangoproject.com/en/5.0/howto/deployment/checklist/
@@ -26,7 +30,7 @@ LOGIN_URL = '/users/login/'
 
 
 # SECURITY WARNING: don't run with debug turned on in production!
-DEBUG = False
+DEBUG = True
 
 ALLOWED_HOSTS = ['artteiv-arxiv-chatbot.hf.space','arxivbot-arxiv-chatbot.hf.space','localhost','127.0.0.1']