Upload AI Search.py

pages/AI Search.py

@@ -0,0 +1,76 @@
+from langchain.agents import ConversationalChatAgent, AgentExecutor
+from langchain.memory import ConversationBufferMemory
+from langchain_community.callbacks import StreamlitCallbackHandler
+from langchain_community.chat_message_histories import StreamlitChatMessageHistory
+from langchain_community.tools import DuckDuckGoSearchRun
+from langchain_core.runnables import RunnableConfig
+from dotenv import load_dotenv
+import os
+from langchain_google_genai import ChatGoogleGenerativeAI
+from datetime import datetime
+
+load_dotenv()
+
+google_gemini_api_key = os.getenv("GOOGLE_API_KEY")
+
+import streamlit as st
+
+st.set_page_config(
+    page_title="J.A.RV.I.S.",
+    page_icon="",
+    layout="centered",
+)
+st.title(":violet[Chat] with AISearch", anchor=False)
+
+if "steps" not in st.session_state:
+    st.session_state.steps = {}
+
+msgs= StreamlitChatMessageHistory()
+memory = ConversationBufferMemory(
+    chat_memory=msgs, return_messages=True, memory_key="chat_history", output_key="output"
+)
+
+# Remove the sidebar button and add a button in the main content area
+if st.button("Reset chat history", key="reset_chat_history"):
+    msgs.clear()
+    st.session_state.steps = {}
+
+avatars = {"human": "user", "ai": "assistant"}
+for idx, msg in enumerate(msgs.messages):
+    with st.chat_message(avatars[msg.type]):
+        # Render intermediate steps if any were saved
+        for step in st.session_state.steps.get(str(idx), []):
+            if step[0].tool == "_Exception":
+                continue
+            with st.status(f"**{step[0].tool}**: {step[0].tool_input}", state="complete"):
+                st.write(step[0].log)
+                st.write(step[1])
+        st.write(msg.content)
+
+from datetime import datetime
+
+if prompt := st.chat_input(placeholder="Who won the Women's U.S. Open in   2018?"):
+    # Get the current date
+    current_date = datetime.now().strftime("%d-%m-%Y")
+    # Modify the prompt to include the current date for the model's search
+    prompt_with_date = f"{prompt} as of {current_date}"
+    # Display the original prompt in the chat without the date
+    st.chat_message("user").write(prompt)
+
+    llm = ChatGoogleGenerativeAI(model="gemini-pro", api_key=google_gemini_api_key, streaming=True, convert_system_message_to_human=True)
+    tools = [DuckDuckGoSearchRun(name="Search")]
+    chat_agent = ConversationalChatAgent.from_llm_and_tools(llm=llm, tools=tools)
+    executor = AgentExecutor.from_agent_and_tools(
+        agent=chat_agent,
+        tools=tools,
+        memory=memory,
+        return_intermediate_steps=True,
+        handle_parsing_errors=True,
+    )
+    with st.chat_message("assistant"):
+        st_cb = StreamlitCallbackHandler(st.container(), expand_new_thoughts=False)
+        cfg = RunnableConfig()
+        cfg["callbacks"] = [st_cb]
+        response = executor.invoke(prompt_with_date, cfg)
+        st.write(response["output"])
+        st.session_state.steps[str(len(msgs.messages) -   1)] = response["intermediate_steps"]