Update app.py

app.py

@@ -571,22 +571,23 @@ def handle_query(user_question, chatbot, audio=None):
     print(user_question)
     # Start the processing in a new thread
     future = executor.submit(answer_question_test, user_question, chatbot)
-    print(future)
-    import time
-    time.sleep(10)
-    # Check if the process is done or cancelled
-    if future.done():
-        if future.cancelled():
-            return "Processing was cancelled."
+    
+    # Periodically check if future is done
+    while not future.done():
+        if stop_event.is_set():
+            future.cancel()
+            return [("Processing was cancelled.", "bot")]
+        time.sleep(1)  # Wait for 1 second before checking again
+
+    if future.cancelled():
+        return [("Processing was cancelled.", "")]
+    else:
         try:
             user_question1, response_text1 = future.result()  # Get the result of the completed future
             chatbot.append((user_question1, response_text1))
             return gr.update(value=chatbot)
-            
         except Exception as e:
-            return f"Error occurred: {e}"
-    else:
-        return "Processing your request. You can stop it if needed."
+            return [(f"Error occurred: {e}", "bot")]
 
 
 def stop_processing(chatbot):