Anustup commited on
Commit
5b6e5b7
·
verified ·
1 Parent(s): 5c1cf92

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -11
app.py CHANGED
@@ -145,8 +145,8 @@ def page3():
145
  file_base64_embeds_product = [embed_base64_for_claude(location) for location in file_locations_for_product]
146
  caption_list_from_claude_product = []
147
  for file_embeds_base64 in file_base64_embeds_product:
148
- caption_prompt_st = st.session_state["caption_prompt"].format(
149
- ITEM_CATEGORY=st.session_state['product_description']
150
  )
151
  caption_from_claude = create_claude_image_request_for_image_captioning(
152
  st.session_state["caption_system_prompt"], caption_prompt_st, file_embeds_base64)
@@ -166,8 +166,8 @@ def page3():
166
  st.session_state.assistant_initialized = True
167
  caption_list_from_claude = []
168
  for file_embeds_base64 in file_base64_embeds:
169
- caption_prompt_st_n = st.session_state["caption_prompt"].format(
170
- ITEM_CATEGORY=st.session_state['product_description']
171
  )
172
  caption_from_claude = create_claude_image_request_for_image_captioning(
173
  st.session_state["caption_system_prompt"], caption_prompt_st_n, file_embeds_base64)
@@ -179,10 +179,9 @@ def page3():
179
  f"```{string_caption_list}```. Apart from this brand needs" \
180
  f"{st.session_state['shoot_type']}"
181
  product_info = str(string_caption_list_product) + st.session_state['product_info']
182
- updated_prompt_for_claude = st.session_state["system_prompt"].format(
183
- BRAND_DETAILS=st.session_state['brand_summary'],
184
- PRODUCT_DETAILS=product_info,
185
- ADDITIONAL_INFO=additional_info_param_for_prompt
186
  )
187
  st.session_state["updated_prompt"] = updated_prompt_for_claude
188
  message_schema_for_claude = [
@@ -327,9 +326,9 @@ def page4():
327
  user_input = st.chat_input("Type your message here...")
328
  if user_input:
329
  st.session_state["mood_chat_messages"].append({"role": "User", "message": user_input})
330
- updated_flux_prompt = prompts["PROMPT_TO_UPDATE_IDEA_OR_MOOD"].format(
331
- EXISTING_MODEL_BG_PROMPT=model__bg_preference,
332
- USER_INSTRUCTIONS=user_input
333
  )
334
  message_schema_for_claude = [
335
  {
 
145
  file_base64_embeds_product = [embed_base64_for_claude(location) for location in file_locations_for_product]
146
  caption_list_from_claude_product = []
147
  for file_embeds_base64 in file_base64_embeds_product:
148
+ caption_prompt_st = st.session_state["caption_prompt"].replace(
149
+ "{{ITEM_CATEGORY}}", st.session_state['product_description']
150
  )
151
  caption_from_claude = create_claude_image_request_for_image_captioning(
152
  st.session_state["caption_system_prompt"], caption_prompt_st, file_embeds_base64)
 
166
  st.session_state.assistant_initialized = True
167
  caption_list_from_claude = []
168
  for file_embeds_base64 in file_base64_embeds:
169
+ caption_prompt_st_n = st.session_state["caption_prompt"].replace(
170
+ "{{ITEM_CATEGORY}}", st.session_state['product_description']
171
  )
172
  caption_from_claude = create_claude_image_request_for_image_captioning(
173
  st.session_state["caption_system_prompt"], caption_prompt_st_n, file_embeds_base64)
 
179
  f"```{string_caption_list}```. Apart from this brand needs" \
180
  f"{st.session_state['shoot_type']}"
181
  product_info = str(string_caption_list_product) + st.session_state['product_info']
182
+ updated_prompt_for_claude = st.session_state["system_prompt"].replace(
183
+ "{{BRAND_DETAILS}}", st.session_state['brand_summary']).replace(
184
+ "{{PRODUCT_DETAILS}}", product_info
 
185
  )
186
  st.session_state["updated_prompt"] = updated_prompt_for_claude
187
  message_schema_for_claude = [
 
326
  user_input = st.chat_input("Type your message here...")
327
  if user_input:
328
  st.session_state["mood_chat_messages"].append({"role": "User", "message": user_input})
329
+ updated_flux_prompt = prompts["PROMPT_TO_UPDATE_IDEA_OR_MOOD"].replace(
330
+ "{{EXISTING_MODEL_BG_PROMPT}}", model__bg_preference).replace(
331
+ "{{USER_INSTRUCTIONS}}", user_input
332
  )
333
  message_schema_for_claude = [
334
  {