Update app.py

app.py

@@ -145,8 +145,11 @@ 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"].format(
+                ITEM_CATEGORY=st.session_state['product_description']
+            )
             caption_from_claude = create_claude_image_request_for_image_captioning(
-                st.session_state["caption_system_prompt"], st.session_state["caption_prompt"], file_embeds_base64)
+                st.session_state["caption_system_prompt"], caption_prompt_st, 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 = []
@@ -163,8 +166,11 @@ 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"].format(
+                ITEM_CATEGORY=st.session_state['product_description']
+            )
             caption_from_claude = create_claude_image_request_for_image_captioning(
-                st.session_state["caption_system_prompt"], st.session_state["caption_prompt"], file_embeds_base64)
+                st.session_state["caption_system_prompt"], caption_prompt_st_n, 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