Upload app.py with huggingface_hub

app.py

@@ -748,6 +748,8 @@ class NutritionBot:
             traceback.print_exc()
 
 #=====================User Interface using streamlit ===========================#
+import streamlit as st
+
 def nutrition_disorder_streamlit():
     """
     A Streamlit-based UI for the Nutrition Disorder Specialist Agent.
@@ -757,15 +759,15 @@ def nutrition_disorder_streamlit():
     st.write("Type 'exit' to end the conversation.")
 
     # Initialize session state for chat history and user_id if they don't exist
-    if 'chat_history' not in st.session_state:
+    if "chat_history" not in st.session_state:
         st.session_state.chat_history = []
-    if 'user_id' not in st.session_state:
+    if "user_id" not in st.session_state:
         st.session_state.user_id = None
 
     # Login form: Only if user is not logged in
     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:")
+            user_id = st.text_input("Please enter your name or id and press Login to begin:")
             submit_button = st.form_submit_button("Login")
             if submit_button and user_id:
                 st.session_state.user_id = user_id
@@ -776,7 +778,7 @@ def nutrition_disorder_streamlit():
                 st.session_state.login_submitted = True  # Set flag to trigger rerun
         if st.session_state.get("login_submitted", False):
             st.session_state.pop("login_submitted")
-            st.rerun()
+            st.experimental_rerun()
     else:
         # Display chat history
         for message in st.session_state.chat_history:
@@ -784,14 +786,15 @@ def nutrition_disorder_streamlit():
                 st.write(message["content"])
 
         # Chat input with custom placeholder text
-        user_query = st.chat_input("Type your question here (or 'exit' to end: ").strip()  # Blank #1: Fill in the chat input prompt (e.g., "Type your question here (or 'exit' to end)...")
+        user_query = st.chat_input("Type your question here (or 'exit' to end): ")
 
-        # Check if user_query is None before calling .strip()
-        if user_query:
-          user_query = user_query.strip()
+        # Check if user_query is None before processing
+        if user_query is not None:
+            user_query = user_query.strip()
         else:
-          user_query = ""  # Default to an empty string if no input is provided
+            user_query = ""  # Default to an empty string if no input is provided
 
+        # Only process non-empty input
         if user_query:
             if user_query.lower() == "exit":
                 st.session_state.chat_history.append({"role": "user", "content": "exit"})
@@ -802,34 +805,53 @@ def nutrition_disorder_streamlit():
                 with st.chat_message("assistant"):
                     st.write(goodbye_msg)
                 st.session_state.user_id = None
-                st.rerun()
+                st.experimental_rerun()  # Reset the app for a new user
                 return
 
+            # Add the user query to the chat history
             st.session_state.chat_history.append({"role": "user", "content": user_query})
             with st.chat_message("user"):
                 st.write(user_query)
 
-            # Filter input using Llama Guard
-            filtered_result = filter_input_with_llama_guard(user_query)  # Blank #2: Fill in with the function name for filtering input (e.g., filter_input_with_llama_guard)
-            filtered_result = filtered_result.replace("\n", " ")  # Normalize the result
-
-            # Check if input is safe based on allowed statuses
-            if filtered_result in ["safe", "unsafe S7", "unsafe S6"]:  # Blanks #3, #4, #5: Fill in with allowed safe statuses (e.g., "safe", "unsafe S7", "unsafe S6")
-                try:
-                    if 'chatbot' not in st.session_state:
-                        st.session_state.chatbot = NutritionBot()  # Blank #6: Fill in with the chatbot class initialization (e.g., NutritionBot)
-                    response = st.session_state.chatbot.andle_customer_query(st.session_state.user_id, user_query)
-                    # Blank #7: Fill in with the method to handle queries (e.g., handle_customer_query)
-                    st.write(response)
+            # Input filtering with Llama Guard
+            try:
+                filtered_result = filter_input_with_llama_guard(user_query)  # Replace with your actual function
+                filtered_result = filtered_result.replace("\n", " ")  # Normalize the result
+
+                # Check if input is safe based on allowed statuses
+                if filtered_result in ["safe", "unsafe S7", "unsafe S6"]:  # Adjust these statuses as needed
+                    if "chatbot" not in st.session_state:
+                        st.session_state.chatbot = NutritionBot()  # Replace with your chatbot initialization
+
+                    # Handle the query
+                    response = st.session_state.chatbot.handle_customer_query(
+                        st.session_state.user_id, user_query
+                    )
+                    with st.chat_message("assistant"):
+                        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)}"
+
+                else:
+                    # Handle inappropriate input
+                    inappropriate_msg = "I apologize, but I cannot process that input as it may be inappropriate. Please try again."
+                    with st.chat_message("assistant"):
+                        st.write(inappropriate_msg)
+                    st.session_state.chat_history.append({"role": "assistant", "content": inappropriate_msg})
+
+            except Exception as e:
+                # Handle errors in query processing
+                error_msg = f"Sorry, I encountered an error while processing your query. Please try again. Error: {str(e)}"
+                with st.chat_message("assistant"):
                     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})
+                st.session_state.chat_history.append({"role": "assistant", "content": error_msg})
+
+        else:
+            # Notify the user if the input is empty
+            with st.chat_message("assistant"):
+                st.write("Please type something to ask a question!")
+
+if __name__ == "__main__":
+    nutrition_disorder_streamlit()
 
 if __name__ == "__main__":
     nutrition_disorder_streamlit()