Update app.py

app.py

@@ -93,6 +93,13 @@ def handle_input(user_input, state):
         logger.info("User attempted to interact without uploading a file.")
         return state['chat_history'], state
     elif state['mode'] == 'waiting_for_query':
+        if user_input.lower() == "exit":
+            logger.info("User entered 'exit'. Ending session.")
+            state['mode'] = 'exited'
+            state['chat_history'].append(("User", "exit"))
+            state['chat_history'].append(("Bot", "Session ended. You can download the log file."))
+            return state['chat_history'], state
+        
         query = user_input
         state['current_query'] = query
         state['feedback_count'] = 0
@@ -109,6 +116,13 @@ def handle_input(user_input, state):
         # Log the query and initial answer here:
         logger.info(f"Query: {query}, Answer: {answer}")
     elif state['mode'] == 'waiting_for_feedback':
+        if user_input.lower() == "exit":
+            logger.info("User entered 'exit'. Ending session.")
+            state['mode'] = 'exited'
+            state['chat_history'].append(("User", "exit"))
+            state['chat_history'].append(("Bot", "Session ended. You can download the log file."))
+            return state['chat_history'], state
+        
         feedback = user_input.lower()
         state['chat_history'].append(("User", feedback))
         logger.info(f"Feedback: {feedback}")
@@ -135,6 +149,9 @@ def handle_input(user_input, state):
             state['feedback_count'] += 1
             state['chat_history'].append(("Bot", f"Updated answer: {adjusted_answer}\nPlease provide feedback: good, too vague, not helpful."))
             logger.info(f"Adjusted answer: {adjusted_answer}")
+    elif state['mode'] == 'exited':
+        state['chat_history'].append(("Bot", "Session is over. Please download the log."))
+        logger.info("User interacted after exiting.")
     return state['chat_history'], state
 
 # Initial state