Update app.py

app.py

@@ -145,11 +145,8 @@ def page3():
         file_base64_embeds_product = [embed_base64_for_claude(location) for location in file_locations_for_product]
         caption_list_from_claude_product = []
         for file_embeds_base64 in file_base64_embeds_product:
-            caption_prompt_st = st.session_state["caption_prompt"].replace(
-                "{{ITEM_CATEGORY}}", st.session_state['product_description']
-            )
             caption_from_claude = create_claude_image_request_for_image_captioning(
-                st.session_state["caption_system_prompt"], caption_prompt_st, file_embeds_base64)
+                st.session_state["caption_system_prompt"], st.session_state["caption_prompt"], file_embeds_base64)
             caption_list_from_claude_product.append(caption_from_claude)
         string_caption_list_product = str(caption_list_from_claude_product)
         file_locations_for_others = []
@@ -166,11 +163,8 @@ def page3():
         st.session_state.assistant_initialized = True
         caption_list_from_claude = []
         for file_embeds_base64 in file_base64_embeds:
-            caption_prompt_st_n = st.session_state["caption_prompt"].replace(
-                "{{ITEM_CATEGORY}}", st.session_state['product_description']
-            )
             caption_from_claude = create_claude_image_request_for_image_captioning(
-                st.session_state["caption_system_prompt"], caption_prompt_st_n, file_embeds_base64)
+                st.session_state["caption_system_prompt"], st.session_state["caption_prompt"], file_embeds_base64)
             caption_list_from_claude.append(caption_from_claude)
         string_caption_list = str(caption_list_from_claude)
         st.session_state["caption_product"] = string_caption_list_product
@@ -180,9 +174,11 @@ def page3():
                                            f"{st.session_state['shoot_type']}"
         product_info = str(string_caption_list_product) + st.session_state['product_info']
         updated_prompt_for_claude = st.session_state["system_prompt"].replace(
-            "{{BRAND_DETAILS}}", st.session_state['brand_summary']).replace(
-            "{{PRODUCT_DETAILS}}", product_info
+            "{{BRAND_DETAILS}}", str(st.session_state['brand_summary'])).replace(
+            "{{PRODUCT_DETAILS}}", str(product_info)).replace(
+            "{{ADDITIONAL_INFO}}", str(additional_info_param_for_prompt)
         )
+        print(f"UP PROMPT:{updated_prompt_for_claude}")
         st.session_state["updated_prompt"] = updated_prompt_for_claude
         message_schema_for_claude = [
             {
@@ -202,13 +198,21 @@ def page3():
         for idx, campaign in enumerate(campaigns, start=1):
             get_model_prompt = extract_data_from_text_xml(campaign, "model_prompt")
             get_background_prompt = extract_data_from_text_xml(campaign, "background_prompt")
-            concat_prompt_flux = get_model_prompt + get_background_prompt
-            concat_prompt_list.append(concat_prompt_flux)
+
+            if get_model_prompt and get_background_prompt:  # Ensure both prompts exist
+                # Clean and concatenate the prompts
+                concat_prompt_flux = (get_model_prompt.strip() + " " + get_background_prompt.strip()).strip()
+                concat_prompt_list.append(concat_prompt_flux)
+
         flux_generated_theme_image = []
         for concat_prompt in concat_prompt_list:
             theme_image = flux_generated_image(concat_prompt)
             flux_generated_theme_image.append(theme_image["file_name"])
-        print(flux_generated_theme_image)
+
+        # Debugging: print generated image file names
+        # print(flux_generated_theme_image)
+
+        # Store the session state
         st.session_state["descriptions"] = concat_prompt_list
         st.session_state["claude_context"] = response_from_claude
         st.session_state["images"] = flux_generated_theme_image
@@ -256,12 +260,21 @@ def page3():
         for idx, campaign in enumerate(campaigns, start=1):
             get_model_prompt = extract_data_from_text_xml(campaign, "model_prompt")
             get_background_prompt = extract_data_from_text_xml(campaign, "background_prompt")
-            concat_prompt_flux = get_model_prompt + get_background_prompt
-            concat_prompt_list.append(concat_prompt_flux)
+
+            if get_model_prompt and get_background_prompt:  # Ensure both prompts exist
+                # Clean and concatenate the prompts
+                concat_prompt_flux = (get_model_prompt.strip() + " " + get_background_prompt.strip()).strip()
+                concat_prompt_list.append(concat_prompt_flux)
+
         flux_generated_theme_image = []
         for concat_prompt in concat_prompt_list:
             theme_image = flux_generated_image(concat_prompt)
             flux_generated_theme_image.append(theme_image["file_name"])
+
+        # Debugging: print generated image file names
+        # print(flux_generated_theme_image)
+
+        # Store the session state
         st.session_state["descriptions"] = concat_prompt_list
         st.session_state["claude_context"] = response_from_claude
         st.session_state["images"] = flux_generated_theme_image
@@ -326,9 +339,9 @@ def page4():
         user_input = st.chat_input("Type your message here...")
         if user_input:
             st.session_state["mood_chat_messages"].append({"role": "User", "message": user_input})
-            updated_flux_prompt = prompts["PROMPT_TO_UPDATE_IDEA_OR_MOOD"].replace(
-                "{{EXISTING_MODEL_BG_PROMPT}}", model__bg_preference).replace(
-                "{{USER_INSTRUCTIONS}}", user_input
+            updated_flux_prompt = prompts["PROMPT_TO_UPDATE_IDEA_OR_MOOD"].format(
+                EXISTING_MODEL_BG_PROMPT=model__bg_preference,
+                USER_INSTRUCTIONS=user_input
             )
             message_schema_for_claude = [
                 {