Upload app.py with huggingface_hub

app.py

@@ -128,6 +128,7 @@ def expand_query(state: AgentState) -> AgentState:
     Convert the user query into something that a nutritionist would understand. Use domain related words.
     Perform query expansion on the question received. If there are multiple common ways of phrasing a user question \
     or common synonyms for key words in the question, make sure to return multiple versions of the query with the different phrasings.
+    In case you have a valid and not empty feedback, you should use that feedback to improve the expanded query.
 
     If the query has multiple parts, split them into separate simpler queries. This is the only case where you can generate more than 1 query.
 
@@ -245,8 +246,8 @@ def score_groundedness(state: Dict) -> Dict:
     """
     print("---------check_groundedness---------")
     system_message = '''
-    "You are an evaluator, and you are tasked with rating AI-generated responses to questions posed by users.
-    You need to assess whether the provided response is grounded in the supplied context.
+    You are an evaluator, and you are tasked with rating AI-generated responses to questions posed by users.
+    You need to assess whether the provided response is grounded in the provided context.
 
     Please act as an impartial judge and evaluate the quality of the provided response, which attempts to answer a user question based on a provided context.
     You will be presented with a context used by the AI model to generate the response and an AI-generated response to the question.
@@ -294,10 +295,10 @@ def check_precision(state: Dict) -> Dict:
     """
     print("---------check_precision---------")
     system_message = '''
-    Given the user query and response, verify if the AI-generated response precisely addresses the user’s query.
-    In the input, the user query will begin with Query:, while the AI generated response will begin with Response:.
+    Given the user query and the AI-generated response, verify if the AI-generated response precisely addresses the user’s query.
+    In the input, the user query will begin with Query: and ends before the token Response:, while the AI generated response will begin with Response:.
 
-    Give verdict as 1 if useful and 0 if not.
+    Give verdict as 1 if the AI-generated response addresses the user’s query and 0 if not.
 
     DO NOT output anything else before or after the veredict integer value of 1 or 0.
     '''
@@ -332,7 +333,7 @@ def refine_response(state: Dict) -> Dict:
 
     system_message = '''
     You are an expert in reviewing AI-generated responses, and your task is to provide constructive feedback on the following response to help improve its accuracy, clarity, and completeness.
-    In the input, the user query will begin with Query:, while the AI-generated response will begin with Response:.
+    In the input, the user query will begin with Query: and ends before the token Response:, while the AI-generated response will begin with Response:.
     You need to identify potential gaps, unsupported claims, ambiguous language, or missing details in the response.
 
     Do not rewrite the response; only suggest improvements in a concise, bullet-point format when possible, to enhance accuracy and completeness.
@@ -515,7 +516,7 @@ def filter_input_with_llama_guard(user_input, model="meta-llama/llama-guard-4-12
 
     Parameters:
     - user_input: The input provided by the user.
-    - model: The Llama Guard model to be used for filtering (default is "llama-guard-3-8b").
+    - model: The Llama Guard model to be used for filtering (default is "meta-llama/llama-guard-4-12b").
 
     Returns:
     - The filtered and safe input.
@@ -693,7 +694,7 @@ def nutrition_disorder_streamlit():
     st.write("You might try asking questions like:")
     st.write("- In what ways can increasing dietary fiber help alleviate symptoms of functional bowel disorders?")
     st.write("- What dietary changes can help reduce the risk of diabetes mellitus in those suffering from obesity?")
-    st.write("- How do dietary deficiencies in essential fatty acids impact neurological function and lead to chronic symptoms?")
+    st.write("- What are some effective dietary changes to help lower high cholesterol levels?")
     st.write("Type 'exit' to end the conversation.")
 
     # Initialize session state for chat history and user_id if they don't exist