Update app.py

app.py

@@ -308,9 +308,9 @@ def analyze_claims(claims, assistant_id):
     
     return claims_analysis_str
 
-def generate_final_analysis(brand_name, product_name, processing_level, harmful_ingredient_analysis, claims_analysis):
+def generate_final_analysis(brand_name, product_name, processing_level, harmful_ingredient_analysis, claims_analysis, system_prompt):
     global debug_mode, client
-    system_prompt = """You are provided with a detailed analysis of a food product. Your task is to generate actionable insights to help the user decide whether to consume the product, at what frequency, and identify any potential harms or benefits. Consider the context of consumption to ensure the advice is personalized and practical.
+    system_prompt_orig = """You are provided with a detailed analysis of a food product. Your task is to generate actionable insights to help the user decide whether to consume the product, at what frequency, and identify any potential harms or benefits. Consider the context of consumption to ensure the advice is personalized and practical.
 
 Use the following criteria to generate your response:
 
@@ -363,7 +363,7 @@ Claims Analysis:
     return completion.choices[0].message.content
 
 
-def analyze_product(product_info_raw):
+def analyze_product(product_info_raw, system_prompt):
     global assistant1, assistant2, assistant3
     
     if product_info_raw != "{}":
@@ -379,7 +379,7 @@ def analyze_product(product_info_raw):
         if len(claims_list) > 0:                    
             claims_analysis = analyze_claims(claims_list, assistant3.id) if claims_list else ""
                 
-        final_analysis = generate_final_analysis(brand_name,product_name,processing_level,harmful_ingredient_analysis,claims_analysis)
+        final_analysis = generate_final_analysis(brand_name,product_name,processing_level,harmful_ingredient_analysis,claims_analysis, system_prompt)
         return final_analysis
     else:
         return "I'm sorry, product information could not be extracted from the url."    
@@ -389,7 +389,7 @@ def analyze_product(product_info_raw):
 if 'messages' not in st.session_state:
     st.session_state.messages = []
 
-def chatbot_response(image_urls_str, product_name_by_user, data_extractor_url, extract_info = True):
+def chatbot_response(image_urls_str, product_name_by_user, data_extractor_url, system_prompt, extract_info = True):
     # Process the user input and generate a response
     processing_level = ""
     harmful_ingredient_analysis = ""
@@ -407,7 +407,7 @@ def chatbot_response(image_urls_str, product_name_by_user, data_extractor_url, e
             with st.spinner("Analyzing the product... This may take a moment."):
                 product_info_raw = get_product_data_from_db(product_name_by_user, data_extractor_url)
                 print(f"DEBUG product_info_raw : {product_info_raw}")
-                final_analysis = analyze_product(product_info_raw)
+                final_analysis = analyze_product(product_info_raw, system_prompt)
                 return [], final_analysis
                 
         else:
@@ -426,7 +426,7 @@ def chatbot_response(image_urls_str, product_name_by_user, data_extractor_url, e
             product_info_raw = extract_data_from_product_image(image_urls, data_extractor_url)
             print(f"DEBUG product_info_raw : {product_info_raw}")
             if product_info_raw:
-                final_analysis = analyze_product(product_info_raw)
+                final_analysis = analyze_product(product_info_raw, system_prompt)
                 return [], final_analysis
             else:
                 return [], "Information could not be extracted from the image!"
@@ -449,6 +449,8 @@ if "yes_no_choice" not in st.session_state:  # Track if Yes/No choice was made
     st.session_state.yes_no_choice = None
 if "welcome_msg" not in st.session_state:  # Track if Yes/No choice was made
     st.session_state.welcome_msg = "Welcome to ConsumeWise! What product would you like me to analyze today?"
+if "system_prompt" not in st.session_state:
+    st.session_state.system_prompt = ""
     
 print(f"st.session_state.product_selected : {st.session_state.product_selected}")
 print(f"st.session_state.product_shared : {st.session_state.product_shared}")
@@ -459,6 +461,19 @@ print(f"st.session_state.yes_no_choice : {st.session_state.yes_no_choice}")
 
 st.title("ConsumeWise - Your Food Product Analysis Assistant")
 
+# Add a text area for system prompt input
+st.sidebar.header("System Prompt")
+system_prompt = st.sidebar.text_area(
+    "Enter your system prompt here:",
+    value=st.session_state.system_prompt,
+    height=150,
+    key="system_prompt_input"
+)
+
+# Update session state when the system prompt changes
+if system_prompt != st.session_state.system_prompt:
+    st.session_state.system_prompt = system_prompt
+    
 # Chat history in a scrollable container
 with st.container():
     st.markdown("<div style='height: 400px; overflow-y: auto;'>", unsafe_allow_html=True)
@@ -504,13 +519,13 @@ def get_yes_no_choice(key):
     
     
 # Define chatbot interaction logic
-def chat_response(user_input):
+def chat_response(user_input, system_prompt):
     """Simulate product selection and analysis logic."""
     if not st.session_state.product_selected:
         # Fetch similar products
         if "http:/" not in user_input and "https:/" not in user_input:
             st.session_state.product_shared = True
-            similar_products, msg = chatbot_response("", user_input, data_extractor_url, extract_info=False)
+            similar_products, msg = chatbot_response("", user_input, data_extractor_url, system_prompt, extract_info=False)
             if len(similar_products) > 0:
                 st.session_state.messages.append(
                     {"role": "assistant", "content": "Here are some similar products. Please select one or enter 'None':"}
@@ -522,7 +537,7 @@ def chat_response(user_input):
                 choice = st.radio("Select a product:", similar_products + ["None"], key="product_choice")
                 if choice and choice != "None":
                     st.session_state.product_selected = True
-                    _, msg = chatbot_response("", user_input, data_extractor_url, extract_info=True)
+                    _, msg = chatbot_response("", user_input, data_extractor_url, system_prompt, extract_info=True)
                     return msg
                     
                 elif choice == "None":
@@ -532,7 +547,7 @@ def chat_response(user_input):
         
         else:
             if ("http:/" in user_input or "https:/" in user_input) and (".jpeg" in user_input or ".jpg" in user_input) and st.session_state.product_shared:
-                _, msg = chatbot_response(user_input, "", data_extractor_url, extract_info=True)
+                _, msg = chatbot_response(user_input, "", data_extractor_url, system_prompt, extract_info=True)
                     
                 st.session_state.product_selected = True
                 
@@ -556,9 +571,10 @@ show_welcome_message()
 
 # Input Box for User Interaction
 user_input = st.chat_input("Enter your message:", key="user_input")
-            
-if user_input:
-    response = chat_response(handle_user_input())
+#How can I provide a text box to enter system_prompt?
+
+if user_input and system_prompt:
+    response = chat_response(handle_user_input(), system_prompt)
     
     if response != "Next Product":
         st.session_state.messages.append({"role": "assistant", "content": response})