Spaces:
Sleeping
Sleeping
File size: 4,462 Bytes
646aeb2 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 |
import streamlit as st
import sys
import os
# Add project directories to Python path
sys.path.append(os.path.join(os.path.dirname(__file__), 'services'))
sys.path.append(os.path.join(os.path.dirname(__file__), 'core'))
sys.path.append(os.path.join(os.path.dirname(__file__), 'config'))
from services.nutrition_bot import NutritionBot
from services.guardrails import filter_input_with_llama_guard
def nutrition_disorder_streamlit():
"""A Streamlit-based UI for the Nutrition Disorder Specialist Agent."""
st.title("Nutrition Disorder Specialist")
st.write("Ask me anything about nutrition disorders, symptoms, causes, treatments, and more.")
st.write("Type 'exit' to end the conversation.")
# Initialize session state
if 'chat_history' not in st.session_state:
st.session_state.chat_history = []
if 'user_id' not in st.session_state:
st.session_state.user_id = None
# Login form
if st.session_state.user_id is None:
with st.form("login_form", clear_on_submit=True):
user_id = st.text_input("Please enter your name to begin:")
submit_button = st.form_submit_button("Login")
if submit_button and user_id:
st.session_state.user_id = user_id
st.session_state.chat_history.append({
"role": "assistant",
"content": f"Welcome, {user_id}! How can I help you with nutrition disorders today?"
})
st.session_state.login_submitted = True
if st.session_state.get("login_submitted", False):
st.session_state.pop("login_submitted")
st.rerun()
else:
# Display chat history
for message in st.session_state.chat_history:
with st.chat_message(message["role"]):
st.write(message["content"])
# Chat input
user_query = st.chat_input("Type your question here (or 'exit' to end)...")
if user_query:
# Check if user wants to exit
if user_query.lower() == "exit":
st.session_state.chat_history.append({"role": "user", "content": "exit"})
with st.chat_message("user"):
st.write("exit")
goodbye_msg = "Goodbye! Feel free to return if you have more questions about nutrition disorders."
st.session_state.chat_history.append({"role": "assistant", "content": goodbye_msg})
with st.chat_message("assistant"):
st.write(goodbye_msg)
st.session_state.user_id = None
st.rerun()
return
# Add user message to chat history
st.session_state.chat_history.append({"role": "user", "content": user_query})
with st.chat_message("user"):
st.write(user_query)
# Filter input
filtered_result = filter_input_with_llama_guard(user_query)
# Process through the agent
with st.chat_message("assistant"):
if filtered_result in ["safe", "unsafe S7", "unsafe S6"]:
try:
# Initialize chatbot if not already done
if 'chatbot' not in st.session_state:
st.session_state.chatbot = NutritionBot()
# Get response from the chatbot
response = st.session_state.chatbot.handle_customer_query(
st.session_state.user_id,
user_query
)
st.write(response)
st.session_state.chat_history.append({"role": "assistant", "content": response})
except Exception as e:
error_msg = f"Sorry, I encountered an error while processing your query. Please try again. Error: {str(e)}"
st.write(error_msg)
st.session_state.chat_history.append({"role": "assistant", "content": error_msg})
else:
inappropriate_msg = "I apologize, but I cannot process that input as it may be inappropriate. Please try again."
st.write(inappropriate_msg)
st.session_state.chat_history.append({"role": "assistant", "content": inappropriate_msg})
if __name__ == "__main__":
nutrition_disorder_streamlit()
|