Initial commit

app.py

@@ -0,0 +1,71 @@
+
+import streamlit as st
+import os
+import urllib.request
+import json
+import ssl
+from menu import menu
+
+
+menu()
+
+# Retrieve Answer through API -----------------------------------------------
+def chat_api(history, question):
+    data = {'chat_history': history,"question": question}
+    body = str.encode(json.dumps(data))
+    url = st.secrets["url"]
+    api_key = st.secrets["ms_api_key"]
+    # Replace this with the primary/secondary key, AMLToken, or Microsoft Entra ID token for the endpoint
+    if not api_key:
+        raise Exception("A key should be provided to invoke the endpoint")
+
+
+    headers = {'Content-Type':'application/json', 'Accept': 'application/json', 'Authorization':('Bearer '+ api_key)}
+
+    req = urllib.request.Request(url, body, headers)
+    response = urllib.request.urlopen(req)
+    result = response.read()
+    answer = json.loads(result)["answer"]
+
+    return(answer)
+
+
+# Header UI ------------------------------------------------------------------
+# st.image('./Cognizant_Logo.jpg', width=300)
+st.header("RCM Chat", divider = 'blue')
+
+# Chat UI --------------------------------------------------------------------
+st.write("This chat as access to search the web to answer your quesitons.")
+st.header("", divider = 'blue')
+
+# Initialize chat history
+if "messages" not in st.session_state:
+    st.session_state.messages = []
+
+# Display chat messages from history on app rerun
+for message in st.session_state.messages:
+    with st.chat_message(message["role"]):
+        st.markdown(message["content"])
+
+# Accept user input
+if prompt := st.chat_input("What is your question?"):
+    # Add user message to chat history
+    st.session_state.messages.append({"role": "user", "content": prompt})
+    # Display user message in chat message container
+    with st.chat_message("user"):
+        st.markdown(prompt)
+
+    # Display assistant response in chat message container
+    with st.chat_message("assistant"):
+        message_placeholder = st.empty()
+        full_response = ""
+        with st.status("Getting the answer...") as status:
+            #assistant_response = get_completion(prompt, ft_model)
+            chat_history = st.session_state.messages
+            print(chat_history)
+            assistant_response = chat_api(history = chat_history, question = prompt)
+            full_response = assistant_response
+            status.update(label="Done.", state = "complete")
+        message_placeholder.markdown(full_response)
+    # Add assistant response to chat history
+    st.session_state.messages.append({"role": "assistant", "content": full_response})

menu.py

@@ -0,0 +1,76 @@
+import streamlit as st
+import hmac
+
+
+def _check_password():
+    """Authenticate user and manage login state."""
+    if st.session_state.get("authenticated"):
+        return True
+
+    with st.form("Credentials"):
+        username = st.text_input("Username")
+        password = st.text_input("Password", type="password")
+        submitted = st.form_submit_button("Log in")
+
+    if submitted:
+        if username in st.secrets["passwords"] and hmac.compare_digest(
+            password,
+            st.secrets.passwords[username],
+        ):
+            st.session_state.authenticated = True
+            st.session_state.username = username
+            st.session_state.role = st.secrets.roles.get(username, "user")  # Default to "user" if role not specified
+            st.rerun()
+        else:
+            st.error("User not known or password incorrect.")
+
+    return False
+
+
+def _authenticated_menu():
+    # Show a navigation menu for authenticated users
+    st.sidebar.page_link("app.py", label="Home Page", icon="🏡")
+    # st.sidebar.page_link("pages/composer.py", label="Insight Composer", icon ="🎨")
+    # st.sidebar.page_link("pages/user.py", label="Regular User Page", icon="🚎")
+    if st.session_state.role in ["admin", "super-admin"]:
+        st.sidebar.page_link("pages/admin.py", label="Admin User Page", icon="🔧")
+        st.sidebar.page_link(
+            "pages/super-admin.py",
+            label="Super Admin User Page",
+            icon="🔧",    
+            disabled=st.session_state.role != "super-admin",
+        )
+    st.sidebar.divider()
+    st.sidebar.write(f"Welcome, {st.session_state.username}!")
+    st.sidebar.write(f"Your role is: {st.session_state.role}")
+    if st.sidebar.button("Logout"):
+        _logout()
+
+
+def _unauthenticated_menu():
+    # Show a navigation menu for unauthenticated users
+    # st.sidebar.page_link("home.py", label="Log in")
+    pass
+
+def _logout():
+    """Log out the current user."""
+    st.session_state.clear()
+    st.success("You have been logged out successfully.")
+    st.rerun()
+
+
+def menu():
+    # Determine if a user is logged in or not, then show the correct
+    if not _check_password():
+        _unauthenticated_menu()
+        st.stop()
+    else:
+        _authenticated_menu()
+
+
+def menu_with_redirect():
+    # Redirect users to the main page if not logged in, otherwise continue to
+    # render the navigation menu
+    if not _check_password():
+        st.switch_page("app.py")
+    menu()