Update app.py

app.py

@@ -51,6 +51,12 @@ load_dotenv()
 # langfuse analytics
 from langfuse.callback import CallbackHandler
 
+import multiprocessing
+
+# Define global variables to manage the process and queue
+process = None
+queue = None
+
 # LangFuse API keys and host settings
 os.environ["LANGFUSE_PUBLIC_KEY"] = os.getenv("LANGFUSE_PUBLIC_KEY")
 os.environ["LANGFUSE_SECRET_KEY"] = os.getenv("LANGFUSE_SECRET_KEY")
@@ -550,6 +556,18 @@ iterations = 0
 
 
 def answer_question(user_question, chatbot, audio=None):
+
+    #initialize the process
+    """
+    Starts the agent in a separate process.
+    """
+    global process
+
+    # Create a multiprocessing queue to get the result back from the process
+    queue = multiprocessing.Queue()
+  
+    process.start()
+    
     global iterations
     iterations = 0
     # Ensure the temporary chart directory exists
@@ -675,6 +693,17 @@ def handle_dislike(data: gr.LikeData):
 def update_message(request: gr.Request):
     return f"<h2 style=' font-family: Calibri;'>Welcome, {request.username}</h4>"
 
+def stop_agent():
+    """
+    Stops the agent process if it is still running.
+    """
+    global process
+    if process and process.is_alive():
+        process.terminate()  # Terminate the process forcefully
+        process.join()  # Ensure the process is properly terminated
+        return "Agent execution stopped."
+    return "No active process to stop."
+
 # CSS for styling the buttons and other elements
 css = """
 /* Example of custom button styling */
@@ -748,6 +777,9 @@ with gr.Blocks(css=css, theme=gr.themes.Soft()) as demo:
         with gr.Column(scale=1):
             with gr.Row():
                 button = gr.Button("Submit", elem_id="submit",elem_classes="gr-button")
+                # Button to stop the agent process
+                stop_button = gr.Button("Stop Agent")
+                stop_button.click(stop_agent, [])
 
     button.click(answer_question, [message, chatbot], [chatbot])
     message.submit(answer_question, [message, chatbot], [chatbot])