final

.gitignore

@@ -11,4 +11,5 @@
 pyvenv.cfg
 venv
 pip-selfcheck.json
-.env
+.env
+env

app.py

@@ -8,43 +8,37 @@ from langchain.agents.agent_toolkits import ZapierToolkit
 from langchain.agents import AgentType
 from langchain.utilities.zapier import ZapierNLAWrapper
 import json
+from langchain.callbacks import StreamlitCallbackHandler
 
-def main():
-    # Load environment variables
-    load_dotenv(".env")
-
-    # Set necessary environment variables
-    # os.environ["LANGCHAIN_WANDB_TRACING"] = "true"
-    # os.environ["WANDB_PROJECT"] = "Claude"
-    agent = os.getenv("ZAPIER_NLA_API_KEY")
-    api_key = os.getenv("OPENAI_API_KEY")
-    serper_api_key = os.getenv("SERP_API_KEY")
-    anthropic_api_key = os.getenv("ANTHROPIC_API_KEY")
-
-    # Initialize necessary objects
-    llm = ChatOpenAI()
-    zapier = ZapierNLAWrapper()
-    toolkit = ZapierToolkit.from_zapier_nla_wrapper(zapier)
-
-    # Streamlit UI
-    st.title("Zap those Zapzz")
-
-    # Text box for the user to enter their input
-    input_string = st.text_input("Please enter your input for the agent:")
-
-    # Button to execute the agent
-    if st.button("Execute"):
-        result = execute_agent(input_string, llm, toolkit)
-        st.text_area("Output:", value=result, height=None, max_chars=None, key=None)
-
-def execute_agent(input_string, llm, toolkit):
-    # Initialize agent
-    agent = initialize_agent(
-        toolkit.get_tools(), llm, agent=AgentType.ZERO_SHOT_REACT_DESCRIPTION, verbose=True
-    )
-
-    # Execute agent
-    return agent.run(input_string)
-
-if __name__ == "__main__":
-    main()
+# load_dotenv("../../.env")s
+
+agents = os.getenv("ZAPIER_NLA_API_KEY")
+api_key = os.getenv("OPENAI_API_KEY")
+
+if api_key is None:
+    api_key = st.text_input("Enter your OpenAI API key", type="password")
+    os.environ["OPENAI_API_KEY"] = api_key
+
+if agents is None:
+    agents = st.text_input("Enter your Zapier NLA API key", type="password")
+    os.environ["ZAPIER_NLA_API_KEY"] = agents
+# Initialize necessary objects
+llm = ChatOpenAI(streaming=True)
+zapier = ZapierNLAWrapper()
+toolkit = ZapierToolkit.from_zapier_nla_wrapper(zapier)
+
+
+# Streamlit UI
+st.title("Zap those Zapzz")
+
+agent = initialize_agent(
+    toolkit.get_tools(), llm, agent=AgentType.ZERO_SHOT_REACT_DESCRIPTION, verbose=True
+)
+
+
+if prompt := st.chat_input():
+    st.chat_message("user").write(prompt)
+    with st.chat_message("assistant"):
+        st_callback = StreamlitCallbackHandler(st.container())
+        response = agent.run(prompt, callbacks=[st_callback])
+        st.write(response)