Create app.py

app.py

@@ -0,0 +1,98 @@
+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()