Update app.py

app.py

@@ -62,7 +62,7 @@ class ChatbotAPP:
         self.service = self.get_drive_service()  # Google Drive service instance
         self.app = self.create_app()  # Gradio app instance
         self.chat_log_name = ""  # Filename for the chat log
-        self.start_time=datetime.now() #save the chat session start
+        self.start_time=None #save the chat session start
 
     # Method to create Google Drive service instance
     def get_drive_service(self):
@@ -139,21 +139,28 @@ class ChatbotAPP:
             print(f"Updated existing file with ID: {updated_file.get('id')}")
 
     def generate(self,prompt, history):    # Method to generate a response to the user's input
-
-        current_time = datetime.now()
-
-        if (current_time - self.start_time) > timedelta (seconds = 20):
-            temp_history = []
-            temp_history.append(("\n\n\n\n\n","10 minutes has been elapsed. Barry died due to not getting treatment!"))
-            return temp_history
             
         #if not len(Name) == 0 and not len(Occupation) == 0 and not len(Ethnicity) == 0 and not len(Gender) == 0 and not len(Age) == 0 and not len(YearsOfExp):
         if self.agreed:
+            
+            current_time = datetime.now()
+
+            if (current_time - self.start_time) > timedelta (minutes = 2):
+                temp_history = []
+                temp_history.append(("""
+                                     .
+                                     .
+                                     .
+                                     .
+                                     .
+                                     .""","10 minutes has been elapsed. Barry died due to not getting treatment!"))
+                return temp_history
 
             user_msg_timestamp = datetime.now().strftime("%H:%M:%S") #capture timestamp the user message is recieved. 
             
             firstmsg =""
             if self.isFirstRun:
+                seld.start_time = datetime.now()
                 self.context = self.initContext
                 self.isFirstRun = False
                 firstmsg = prompt