Update app.py

app.py

@@ -596,7 +596,7 @@ def handle_query(user_question, chatbot, audio=None):
 
     """
     Function to handle the processing of user input with `AgentExecutor.invoke()`.
-    """
+    
     global current_event, stop_event
 
     # Clear previous stop event and current_event
@@ -639,7 +639,47 @@ def handle_query(user_question, chatbot, audio=None):
             print(f"Error occurred: {e}")
             chatbot.append((user_question, "Sorry, we encountered an error while processing your request. Please try after some time."))
             return gr.update(value=chatbot)
+"""
+    global current_event, stop_event
 
+    # Clear the stop event to ensure we start fresh
+    stop_event.clear()
+
+    # Check if a task is already running
+    if current_event and not current_event.done():
+        chatbot.append(("", "A query is already being processed. Please stop it before starting a new one."))
+        return gr.update(value=chatbot)
+
+    # Start the processing in a new thread
+    current_event = executor.submit(answer_question_thread, user_question, chatbot)
+
+    # Periodically check if current_event is done
+    try:
+        while not current_event.done():
+            if stop_event.is_set():
+                # Attempt to cancel the current_event
+                if current_event.cancel():
+                    chatbot.append((user_question, "Processing was stopped by the user."))
+                else:
+                    chatbot.append((user_question, "Unable to stop processing. Please try again."))
+                
+                # Return the updated chatbot and stop further processing
+                return gr.update(value=chatbot)
+
+            time.sleep(1)  # Wait for 1 second before checking again
+
+        # Task completed successfully; fetch the result
+        try:
+            user_question1, response_text1 = current_event.result()
+            chatbot.append((user_question1, response_text1))
+            return gr.update(value=chatbot)
+        except Exception as e:
+            print(f"Error occurred: {e}")
+            chatbot.append((user_question, "An error occurred while processing your request. Please try again later."))
+            return gr.update(value=chatbot)
+    finally:
+        # Clean up after processing
+        current_event = None
 
 def stop_processing(chatbot):
     """