Upload app.py with huggingface_hub

app.py

@@ -677,7 +677,7 @@ class NutritionBot:
             context += "---\n"
 
         # Print context for debugging purposes
-        #st.write("Context: ", context)
+        print("Context: ", context)
 
         # Prepare a prompt combining past context and the current query
         prompt = f"""
@@ -691,7 +691,7 @@ class NutritionBot:
         #st.write("Context: ", prompt)
 
         # Generate a response using the agent
-        response = self.agent_executor.invoke({"input": prompt, "agent_scratchpad": ''})
+        response = self.agent_executor.invoke({"input": prompt})
         st.write("Context: ", response)
 
         # Store the current interaction for future reference
@@ -763,7 +763,13 @@ def nutrition_disorder_streamlit():
 
             # 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
+            #filtered_result = filtered_result.replace("\n", " ")  # Normalize the result
+            if filtered_result is None:
+                print("Agent: Sorry, I encountered an error while filtering your input. Please try again.")
+                continue
+            else:
+                filtered_result = filtered_result.replace("\n", " ")  # Normalize the result
+                print(f"filtered_result (2): {filtered_result}")
 
             # Check if input is safe based on allowed statuses
             if filtered_result in ["safe", "safe S7", "safe S6"]: # Blanks #3, #4, #5: Fill in with allowed safe statuses (e.g., "safe", "unsafe S7", "unsafe S6")
@@ -771,11 +777,13 @@ def nutrition_disorder_streamlit():
                 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)
-                    #st.write("chatbot instantiated - calling handle_customer_query...")
-
+                    #st.write("chatbot is calling handle_customer_query...")
+                    #st.write("user_id: ", st.session_state.user_id)
+                    #st.write("user_query: ", user_query)
                     response = st.session_state.chatbot.handle_customer_query(st.session_state.user_id, user_query)
                     #st.write("response is returned.")
                     # Blank #7: Fill in with the method to handle queries (e.g., handle_customer_query)
+
                     st.write(response)
                     st.session_state.chat_history.append({"role": "assistant", "content": response})
                 except Exception as e: