Update app.py

app.py

@@ -7,6 +7,7 @@ from langchain_huggingface import HuggingFacePipeline
 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
 
 # Email configuration
 SENDER_EMAIL = "[email protected]"  # Replace with your email
@@ -86,7 +87,11 @@ prompt = PromptTemplate.from_template(
 
 agent = create_react_agent(local_llm, tools, prompt)
 agent_executor = AgentExecutor.from_agent_and_tools(
-    agent=agent, tools=tools, verbose=True, memory=ConversationBufferMemory(memory_key="chat_history", return_messages=True)
+    agent=agent, 
+    tools=tools, 
+    verbose=True, 
+    memory=ConversationBufferMemory(memory_key="chat_history", return_messages=True),
+    handle_parsing_errors=True
 )
 
 # Streamlit interface
@@ -108,9 +113,19 @@ if prompt := st.chat_input("Your message"):
 
     with st.chat_message("assistant"):
         with st.spinner("Thinking..."):
-            response = agent_executor.run(prompt)
-            st.markdown(response)
-            st.session_state.messages.append({"role": "assistant", "content": response})
+            try:
+                response = agent_executor({"input": prompt})
+                if isinstance(response, dict) and "output" in response:
+                    assistant_response = response["output"]
+                elif isinstance(response, (AgentAction, AgentFinish)):
+                    assistant_response = response.return_values.get("output", str(response))
+                else:
+                    assistant_response = str(response)
+                
+                st.markdown(assistant_response)
+                st.session_state.messages.append({"role": "assistant", "content": assistant_response})
+            except Exception as e:
+                st.error(f"An error occurred: {str(e)}")
 
 st.sidebar.title("About")
 st.sidebar.info("This is an interactive CyberSecurity Program Meeting Scheduler. Chat with the AI to schedule your meeting. It will collect your information and send a meeting invitation via email.")