Anustup commited on
Commit
208d07b
·
verified ·
1 Parent(s): 5b137a6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -7,7 +7,7 @@ from prompts import prompts
7
  from constants import JSON_SCHEMA_FOR_GPT, UPDATED_MODEL_ONLY_SCHEMA, JSON_SCHEMA_FOR_LOC_ONLY
8
  from gpt import runAssistant, checkRunStatus, retrieveThread, createAssistant, saveFileOpenAI, startAssistantThread, \
9
  create_chat_completion_request_open_ai_for_summary, addMessageToThread, create_image_completion_request_gpt
10
- from summarizer import create_brand_html, create_langchain_openai_query, create_screenshot_from_scrap_fly
11
  from theme import flux_generated_image, flux_generated_image_seed
12
  import time
13
  from PIL import Image
@@ -94,8 +94,9 @@ def page2():
94
  st.text("Using Scrapfly")
95
  brand_summary_html = create_screenshot_from_scrap_fly(brand_link)
96
  if brand_summary_html["success"]:
97
- st.image(brand_summary_html["location"])
98
- brand_image_embed = embed_base64_for_claude(brand_summary_html["location"])
 
99
  brand_summary_response = create_claude_image_request_for_image_captioning(
100
  "Fashion expert of understanding brand details",
101
  st.session_state["brand_prompt"], brand_image_embed)
 
7
  from constants import JSON_SCHEMA_FOR_GPT, UPDATED_MODEL_ONLY_SCHEMA, JSON_SCHEMA_FOR_LOC_ONLY
8
  from gpt import runAssistant, checkRunStatus, retrieveThread, createAssistant, saveFileOpenAI, startAssistantThread, \
9
  create_chat_completion_request_open_ai_for_summary, addMessageToThread, create_image_completion_request_gpt
10
+ from summarizer import create_brand_html, create_langchain_openai_query, create_screenshot_from_scrap_fly, check_and_compress_image
11
  from theme import flux_generated_image, flux_generated_image_seed
12
  import time
13
  from PIL import Image
 
94
  st.text("Using Scrapfly")
95
  brand_summary_html = create_screenshot_from_scrap_fly(brand_link)
96
  if brand_summary_html["success"]:
97
+ compressed_image = f"comp_brand_{random.randint(1, 100000000)}.png"
98
+ check_and_compress_image(brand_summary_html["location"], compressed_image)
99
+ brand_image_embed = embed_base64_for_claude(compressed_image)
100
  brand_summary_response = create_claude_image_request_for_image_captioning(
101
  "Fashion expert of understanding brand details",
102
  st.session_state["brand_prompt"], brand_image_embed)