Upload app.py with huggingface_hub

app.py

@@ -764,10 +764,17 @@ 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
+            if filtered_result is None:
+            # fail-open or fail-closed; most teams choose cautious allow for health Q&A
+            filtered_result = "SAFE"
+
+            label = filtered_result.replace("\n", " ").strip().upper()
+
+            #filtered_result = filtered_result.replace("\n", " ")  # Normalize the result
 
             # Check if input is safe based on allowed statuses
-            if filtered_result in ["SAFE", "S6", "S7"]:  # Blanks #3, #4, #5: Fill in with allowed safe statuses (e.g., "safe", "unsafe S7", "unsafe S6")
+            #if filtered_result in ["SAFE", "S6", "S7"]:  # Blanks #3, #4, #5: Fill in with allowed safe statuses (e.g., "safe", "unsafe S7", "unsafe S6")
+            if any(tag in label for tag in ("SAFE", "S6", "S7")):
                 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)