Upload app.py with huggingface_hub

app.py

@@ -749,6 +749,7 @@ class NutritionBot:
 
 #=====================User Interface using streamlit ===========================#
 import streamlit as st
+import os
 
 def nutrition_disorder_streamlit():
     """
@@ -766,7 +767,6 @@ def nutrition_disorder_streamlit():
 
     # Login form: Only if user is not logged in
     if st.session_state.user_id is None:
-        # Use a unique key for the form
         with st.form(key="login_form", clear_on_submit=True):
             user_id = st.text_input("Please enter your name to begin:")
             submit_button = st.form_submit_button("Login")
@@ -776,10 +776,10 @@ def nutrition_disorder_streamlit():
                     "role": "assistant",
                     "content": f"Welcome, {user_id}! How can I help you with nutrition disorders today?"
                 })
-                st.session_state.login_submitted = True  # Set flag to trigger rerun
+                st.session_state.login_submitted = True
         if st.session_state.get("login_submitted", False):
             st.session_state.pop("login_submitted")
-            st.rerun()  # Replace st.experimental_rerun with st.rerun
+            st.rerun()
     else:
         # Display chat history
         for message in st.session_state.chat_history:
@@ -789,13 +789,11 @@ def nutrition_disorder_streamlit():
         # Chat input with custom placeholder text
         user_query = st.chat_input("Type your question here (or 'exit' to end): ")
 
-        # 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 = ""
 
-        # Only process non-empty input
         if user_query:
             if user_query.lower() == "exit":
                 st.session_state.chat_history.append({"role": "user", "content": "exit"})
@@ -806,23 +804,25 @@ def nutrition_disorder_streamlit():
                 with st.chat_message("assistant"):
                     st.write(goodbye_msg)
                 st.session_state.user_id = None
-                st.rerun()  # Replace st.experimental_rerun with st.rerun
+                st.rerun()
                 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)
 
             # 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
+                filtered_result = filter_input_with_llama_guard(user_query)  # Replace with the actual function
+                filtered_result = filtered_result.replace("\n", " ")
 
-                # Check if input is safe based on allowed statuses
-                if filtered_result in ["safe", "unsafe S7", "unsafe S6"]:  # Adjust these statuses as needed
+                if filtered_result in ["safe", "unsafe S7", "unsafe S6"]:
                     if "chatbot" not in st.session_state:
-                        st.session_state.chatbot = NutritionBot()  # Replace with your chatbot initialization
+                        # Pass the API key and base to the chatbot initialization
+                        st.session_state.chatbot = NutritionBot(
+                            api_key=os.getenv("OPENAI_API_KEY"),
+                            api_base=os.getenv("OPENAI_API_BASE")
+                        )
 
                     # Handle the query
                     response = st.session_state.chatbot.handle_customer_query(
@@ -833,21 +833,18 @@ def nutrition_disorder_streamlit():
                     st.session_state.chat_history.append({"role": "assistant", "content": response})
 
                 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:
-            # Notify the user if the input is empty
             with st.chat_message("assistant"):
                 st.write("Please type something to ask a question!")