Upload app.py with huggingface_hub

app.py

@@ -116,28 +116,28 @@ def expand_query(state):
     """
     print("---------Expanding Query---------")
     system_message = '''You are an AI specializing in improving search queries to retrieve the most relevant nutrition disorder-related information.
-      Your task is to **refine** and **expand** the given query so that better search results are obtained, while **keeping the original intent** unchanged.
+    Your task is to **refine** and **expand** the given query so that better search results are obtained, while **keeping the original intent** unchanged.
 
-      Guidelines:
-      - Add **specific details** where needed. Example: If a user asks about "anorexia," specify aspects like symptoms, causes, or treatment options.
-      - Include **related terms** to improve retrieval (e.g., “bulimia” → “bulimia nervosa vs binge eating disorder”).
-      - If the user provides an unclear query, suggest necessary clarifications.
-      - **DO NOT** answer the question. Your job is only to enhance the query.
+    Guidelines:
+    - Add **specific details** where needed. Example: If a user asks about "anorexia," specify aspects like symptoms, causes, or treatment options.
+    - Include **related terms** to improve retrieval (e.g., “bulimia” → “bulimia nervosa vs binge eating disorder”).
+    - If the user provides an unclear query, suggest necessary clarifications.
+    - **DO NOT** answer the question. Your job is only to enhance the query.
 
-      Examples:
-      1. User Query: "Tell me about eating disorders."
-        Expanded Query: "Provide details on eating disorders, including types (e.g., anorexia nervosa, bulimia nervosa), symptoms, causes, and treatment options."
+    Examples:
+    1. User Query: "Tell me about eating disorders."
+      Expanded Query: "Provide details on eating disorders, including types (e.g., anorexia nervosa, bulimia nervosa), symptoms, causes, and treatment options."
 
-      2. User Query: "What is anorexia?"
-        Expanded Query: "Explain anorexia nervosa, including its symptoms, causes, risk factors, and treatment options."
+    2. User Query: "What is anorexia?"
+      Expanded Query: "Explain anorexia nervosa, including its symptoms, causes, risk factors, and treatment options."
 
-      3. User Query: "How to treat bulimia?"
-        Expanded Query: "Describe treatment options for bulimia nervosa, including psychotherapy, medications, and lifestyle changes."
+    3. User Query: "How to treat bulimia?"
+      Expanded Query: "Describe treatment options for bulimia nervosa, including psychotherapy, medications, and lifestyle changes."
 
-      4. User Query: "What are the effects of malnutrition?"
-        Expanded Query: "Explain the effects of malnutrition on physical and mental health, including specific nutrient deficiencies and their consequences."
+    4. User Query: "What are the effects of malnutrition?"
+      Expanded Query: "Explain the effects of malnutrition on physical and mental health, including specific nutrient deficiencies and their consequences."
 
-      Now, expand the following query:'''
+    Now, expand the following query:'''
 
     expand_prompt = ChatPromptTemplate.from_messages([
         ("system", system_message),
@@ -845,7 +845,7 @@ def nutrition_disorder_streamlit():
             # Process through the agent
             with st.chat_message("assistant"):
                 # if filtered_result in ["safe", "unsafe S7", "unsafe S6"]:
-                if filtered_result in ["SAFE", "S6", "S7"]:
+                if filtered_result in ["safe", "S6", "S7"]:
                     try:
                         # Initialize chatbot if not already done
                         if 'chatbot' not in st.session_state: