Upload app.py with huggingface_hub

app.py

@@ -569,28 +569,12 @@ class NutritionBot:
         self.memory = MemoryClient(api_key=MEM0_api_key) # userdata.get("mem0"))  # Complete the code to define the memory client API key
 
         self.client = ChatOpenAI(
-            model_name="gpt-4o-mini",  # Specify the model to use (e.g., GPT-4 optimized version)
+            model_name="gpt-4o",  # Specify the model to use (e.g., GPT-4 optimized version)
             api_key = my_api_key, # config.get("API_KEY"),  # API key for authentication
             #endpoint = endpoint, # config.get("OPENAI_API_BASE"),
             temperature=0  # Controls randomness in responses; 0 ensures deterministic results
         )
 
-        '''
-        try:
-          self.client = ChatOpenAI(
-              api_key=my_api_key,
-              base_url=endpoint,
-              openai_api_base=endpoint,
-              openai_api_key=my_api_key,
-              model="gpt-4o-mini",
-              streaming=False, # Explicitly disabling streaming
-              temperature=0
-          )
-        except Exception as e:
-          print(f"Error with OpenAI: {str(e)}")
-          return None
-        '''
-
         # Define tools available to the chatbot, such as web search
         tools = [agentic_rag]
 
@@ -694,7 +678,7 @@ class NutritionBot:
             context += "---\n"
 
         # Print context for debugging purposes
-        st.write("Context: ", context)
+        #st.write("Context: ", context)
 
         # Prepare a prompt combining past context and the current query
         prompt = f"""
@@ -702,11 +686,9 @@ class NutritionBot:
         Current customer query: {query}
         Provide a helpful response that takes into account any relevant past interactions.
         """
-        st.write("prompt: ", prompt)
 
         # Generate a response using the agent
         response = self.agent_executor.invoke({"input": prompt})
-        st.write("response: ", response)
 
         # Store the current interaction for future reference
         self.store_customer_interaction(
@@ -781,17 +763,11 @@ def nutrition_disorder_streamlit():
 
             # 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")
-                st.write("Got a safe input...")
                 try:
                     if 'chatbot' not in st.session_state:
-                        st.write("In not in st.session_state...")
                         st.session_state.chatbot = NutritionBot() # Blank #6: Fill in with the chatbot class initialization (e.g., NutritionBot)
-                    else:
-                        st.write("In st.session_state...")
 
-                    st.write(f"Getting response...\nUser ID: {st.session_state.user_id}\nUser query: {user_query}")
                     response = st.session_state.chatbot.handle_customer_query(st.session_state.user_id, user_query) 
-                    st.write(f"Got response: {response}")
                     # 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})
@@ -800,13 +776,9 @@ def nutrition_disorder_streamlit():
                     st.write(error_msg)
                     st.session_state.chat_history.append({"role": "assistant", "content": error_msg})
             else:
-                try:
-                  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})
-                except Exception as e:
-                  st.write(str(e))
-                  st.session_state.chat_history.append({"role": "assistant", "content": str(e)})
+                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()