initial commit

app.py

@@ -0,0 +1,12 @@
+import streamlit as st
+
+pg = st.navigation([
+    st.Page("pages/simple_page.py"), 
+    st.Page("pages/formatted_page.py"),
+    st.Page("pages/history_page.py"), 
+    st.Page("pages/rag_page.py"),
+    st.Page("pages/filtered_rag_page.py"),
+])
+pg.run()
+
+

pages/filtered_rag_page.py

@@ -0,0 +1,9 @@
+from streamlit.runtime.scriptrunner import get_script_run_ctx
+from pages.page_base import chat_interface
+
+
+chat_title = "Filtered RAG Chat App"
+url = "[YOUR FILTERED RAG URL]"
+page_hash = get_script_run_ctx().page_script_hash
+
+chat_interface(chat_title, page_hash, url)

pages/formatted_page.py

@@ -0,0 +1,9 @@
+from streamlit.runtime.scriptrunner import get_script_run_ctx
+from pages.page_base import chat_interface
+
+
+chat_title = "Formatted Chat App"
+url = "[YOUR FORMATTED CHAT URL]"
+page_hash = get_script_run_ctx().page_script_hash
+
+chat_interface(chat_title, page_hash, url)

pages/history_page.py

@@ -0,0 +1,9 @@
+from streamlit.runtime.scriptrunner import get_script_run_ctx
+from pages.page_base import chat_interface
+
+
+chat_title = "History Chat App"
+url = "[YOUR HISTORY CHAT URL]"
+page_hash = get_script_run_ctx().page_script_hash
+
+chat_interface(chat_title, page_hash, url)

pages/page_base.py

@@ -0,0 +1,48 @@
+import streamlit as st
+from langserve.client import RemoteRunnable
+
+def get_response(user_input, url, username):
+    response_placeholder = st.empty()
+    full_response = ""
+    chain = RemoteRunnable(url)
+    stream = chain.stream(input={'question': user_input, 'username': username})
+    for chunk in stream:
+        full_response += chunk
+        response_placeholder.markdown(full_response)
+
+    return full_response
+
+def chat_interface(chat_title, page_hash ,url):
+    st.title(chat_title)
+
+    # Add username input at the top of the page
+    username = st.text_input("Enter your username:", key="username_input", value="Guest")
+
+    # Initialize page-specific chat history
+    if "chat_histories" not in st.session_state:
+        st.session_state.chat_histories = {}
+    
+    if page_hash not in st.session_state.chat_histories:
+        st.session_state.chat_histories[page_hash] = []
+
+    # Display chat messages from history for the current page
+    for message in st.session_state.chat_histories[page_hash]:
+        with st.chat_message(message["role"]):
+            st.markdown(message["content"])
+
+    # React to user input
+    if prompt := st.chat_input("What is your message?"):
+        # Display user message in chat message container
+        st.chat_message("user").markdown(prompt)
+        # Add user message to chat history
+        st.session_state.chat_histories[page_hash].append({"role": "user", "content": prompt})
+
+        # Get streaming response
+        with st.chat_message("assistant"):
+            full_response = get_response(prompt, url, username)
+
+        # Add assistant response to chat history
+        st.session_state.chat_histories[page_hash].append({"role": "assistant", "content": full_response})
+
+if __name__ == "__main__":
+    chat_interface()

pages/rag_page.py

@@ -0,0 +1,9 @@
+from streamlit.runtime.scriptrunner import get_script_run_ctx
+from pages.page_base import chat_interface
+
+
+chat_title = "RAG Chat App"
+url = "[YOUR RAG CHAT URL]"
+page_hash = get_script_run_ctx().page_script_hash
+
+chat_interface(chat_title, page_hash, url)

pages/simple_page.py

@@ -0,0 +1,12 @@
+# import streamlit as st
+# import time
+# from langserve import RemoteRunnable
+from streamlit.runtime.scriptrunner import get_script_run_ctx
+import streamlit as st
+from pages.page_base import chat_interface
+
+chat_title = "Simple Chat App"
+url = "https://damienbenveniste-backend.hf.space/simple"
+page_hash = get_script_run_ctx().page_script_hash
+
+chat_interface(chat_title, page_hash, url)

requirements.txt

@@ -0,0 +1 @@
+langserve[client]