Update app.py

app.py

@@ -1,41 +1,38 @@
-import os
-
-import anthropic
 import streamlit as st
 
-st.title("Claude 3 by Streamlit")
+from langchain.agents import initialize_agent, AgentType
+from langchain.callbacks import StreamlitCallbackHandler
+from langchain.chat_models import ChatOpenAI
+from langchain.tools import DuckDuckGoSearchRun
+
+openai_api_key = os.getenv("OPENAI_API_KEY")
 
-ai_model = os.environ.get("AI_MODEL")
-api_key = os.environ.get("ANTHROPIC_API_KEY")
+st.title("🔎 PromptingAI.io - Chat with search")
 
-client = anthropic.Anthropic(
-    api_key=api_key,
-)
-if "ai_model" not in st.session_state:
-    st.session_state["ai_model"] = ai_model
 
 if "messages" not in st.session_state:
-    st.session_state.messages = []
+    st.session_state["messages"] = [
+        {"role": "assistant", "content": "Hi, I'm a chatbot who can search the web. How can I help you?"}
+    ]
 
-for message in st.session_state.messages:
-    with st.chat_message(message["role"]):
-        st.markdown(message["content"])
+for msg in st.session_state.messages:
+    st.chat_message(msg["role"]).write(msg["content"])
 
-if prompt := st.chat_input("What is up?"):
+if prompt := st.chat_input(placeholder="What is PromptingAI.io"):
     st.session_state.messages.append({"role": "user", "content": prompt})
-    with st.chat_message("user"):
-        st.markdown(prompt)
+    st.chat_message("user").write(prompt)
+
+    if not openai_api_key:
+        st.info("Please add your OpenAI API key to continue.")
+        st.stop()
 
+    llm = ChatOpenAI(model_name="gpt-3.5-turbo", openai_api_key=openai_api_key, streaming=True)
+    search = DuckDuckGoSearchRun(name="Search")
+    search_agent = initialize_agent(
+        [search], llm, agent=AgentType.ZERO_SHOT_REACT_DESCRIPTION, handle_parsing_errors=True
+    )
     with st.chat_message("assistant"):
-        message_placeholder = st.empty()
-        full_response = ""
-        with client.messages.stream(
-            max_tokens=1024,
-            messages=[{"role": m["role"], "content": m["content"]} for m in st.session_state.messages],
-            model=st.session_state["ai_model"],
-        ) as stream:
-            for text in stream.text_stream:
-                full_response += str(text) if text is not None else ""
-                message_placeholder.markdown(full_response + "▌")
-        message_placeholder.markdown(full_response)
-    st.session_state.messages.append({"role": "assistant", "content": full_response})
+        st_cb = StreamlitCallbackHandler(st.container(), expand_new_thoughts=False)
+        response = search_agent.run(st.session_state.messages, callbacks=[st_cb])
+        st.session_state.messages.append({"role": "assistant", "content": response})
+        st.write(response)