Add doc loader

app.py

@@ -1,6 +1,25 @@
 import streamlit as st
+from langchain_openai import ChatOpenAI
+from langchain_core.prompts import ChatPromptTemplate
+from langchain_community.document_loaders import WebBaseLoader
 
-st.title("Echo Bot")
+default_url = "https://rocm.docs.amd.com/en/latest/what-is-rocm.html"
+st.title("URL Loader")
+
+url = st.text_input("Provide URL ", default_url)
+if "url_dict" not in st.session_state:
+    st.session_state.url_dict = {}
+if url not in st.session_state.url_dict:
+    loader = WebBaseLoader(url)
+    st.session_state.url_dict[url] = loader.load()
+
+llm = ChatOpenAI(temperature=0.1)
+prompt = ChatPromptTemplate.from_template("""
+    Answer the user's question:
+    Context: {context}
+    Question : {input}
+    """)
+chain = prompt | llm
 
 # Initialize chat history
 if "messages" not in st.session_state:
@@ -12,16 +31,19 @@ for message in st.session_state.messages:
         st.markdown(message["content"])
 
 # React to user input
-if prompt := st.chat_input("What is up?"):
+if prompt := st.chat_input("Ask Question to the URL provided"):
     # Display user message in chat message container
     st.chat_message("user").markdown(prompt)
     # Add user message to chat history
     st.session_state.messages.append({"role": "user", "content": prompt})
 
-    response = f"Echo: {prompt}"
+    response = chain.invoke({
+    "context" : [st.session_state.url_dict[url]],
+    "input" : prompt
+    })
     # Display assistant response in chat message container
     with st.chat_message("assistant"):
-        st.markdown(response)
+        st.markdown(response.content)
     # Add assistant response to chat history
-    st.session_state.messages.append({"role": "assistant", "content": response})
+    st.session_state.messages.append({"role": "assistant", "content": response.content})
 

requirements.txt

@@ -0,0 +1,76 @@
+aiohttp==3.9.5
+aiosignal==1.3.1
+altair==5.3.0
+annotated-types==0.6.0
+anyio==4.3.0
+attrs==23.2.0
+beautifulsoup4==4.12.3
+blinker==1.8.2
+bs4==0.0.2
+cachetools==5.3.3
+certifi==2024.2.2
+charset-normalizer==3.3.2
+click==8.1.7
+dataclasses-json==0.6.6
+distro==1.9.0
+frozenlist==1.4.1
+gitdb==4.0.11
+GitPython==3.1.43
+h11==0.14.0
+httpcore==1.0.5
+httpx==0.27.0
+idna==3.7
+Jinja2==3.1.4
+jsonpatch==1.33
+jsonpointer==2.4
+jsonschema==4.22.0
+jsonschema-specifications==2023.12.1
+langchain==0.2.0
+langchain-community==0.2.0
+langchain-core==0.2.0
+langchain-openai==0.1.7
+langchain-text-splitters==0.2.0
+langsmith==0.1.59
+markdown-it-py==3.0.0
+MarkupSafe==2.1.5
+marshmallow==3.21.2
+mdurl==0.1.2
+multidict==6.0.5
+mypy-extensions==1.0.0
+numpy==1.26.4
+openai==1.30.1
+orjson==3.10.3
+packaging==23.2
+pandas==2.2.2
+pillow==10.3.0
+protobuf==4.25.3
+pyarrow==16.1.0
+pydantic==2.7.1
+pydantic_core==2.18.2
+pydeck==0.9.1
+Pygments==2.18.0
+python-dateutil==2.9.0.post0
+pytz==2024.1
+PyYAML==6.0.1
+referencing==0.35.1
+regex==2024.5.15
+requests==2.31.0
+rich==13.7.1
+rpds-py==0.18.1
+six==1.16.0
+smmap==5.0.1
+sniffio==1.3.1
+soupsieve==2.5
+SQLAlchemy==2.0.30
+streamlit==1.34.0
+tenacity==8.3.0
+tiktoken==0.7.0
+toml==0.10.2
+toolz==0.12.1
+tornado==6.4
+tqdm==4.66.4
+typing-inspect==0.9.0
+typing_extensions==4.11.0
+tzdata==2024.1
+urllib3==2.2.1
+yarl==1.9.4