Update app.py

app.py

@@ -8,6 +8,7 @@ from langchain.agents import create_react_agent, AgentExecutor, Tool
 from langchain.prompts import PromptTemplate
 from langchain.memory import ConversationBufferMemory
 from langchain.schema import AgentAction, AgentFinish
+import concurrent.futures
 import time
 
 # Email configuration
@@ -74,6 +75,14 @@ agent_executor = AgentExecutor.from_agent_and_tools(
     early_stopping_method="generate"  # Stop if the agent starts looping
 )
 
+def run_agent_with_timeout(executor, input_data, timeout):
+    with concurrent.futures.ThreadPoolExecutor() as pool:
+        future = pool.submit(executor, input_data)
+        try:
+            return future.result(timeout=timeout)
+        except concurrent.futures.TimeoutError:
+            raise TimeoutError("The operation timed out.")
+
 st.title("CyberSecurity Program Meeting Scheduler")
 st.write("Chat with the AI to schedule your meeting. The AI will ask for your name, email, and preferred meeting date.")
 
@@ -93,7 +102,7 @@ if prompt := st.chat_input("Your message"):
         with st.spinner("Thinking..."):
             try:
                 start_time = time.time()
-                response = agent_executor({"input": prompt}, timeout=10)  # 10-second timeout
+                response = run_agent_with_timeout(agent_executor, {"input": prompt}, timeout=10)
                 if isinstance(response, dict) and "output" in response:
                     assistant_response = response["output"]
                 elif isinstance(response, (AgentAction, AgentFinish)):