added print logs

app.py

@@ -7,6 +7,8 @@ import urllib.request
 import tarfile
 from yaml.loader import SafeLoader
 import warnings
+import utilities.invoke_models as invoke_models
+import threading
 warnings.filterwarnings("ignore", category=DeprecationWarning)
 
 # Page setup
@@ -16,6 +18,20 @@ st.set_page_config(
     
 )
 
+def async_bedrock_warmup():
+    try:
+        _ = invoke_models.invoke_model_mm("hi", "none")
+        _ = invoke_models.invoke_model("hi")
+        _ = invoke_models.invoke_llm_model("hi", False)
+        st.session_state.bedrock_warmup_done = True
+        print("[Background Warmup] Done.")
+    except Exception as e:
+        print(f"[Background Warmup] Failed: {e}")
+
+if "bedrock_warmup_done" not in st.session_state:
+    threading.Thread(target=async_bedrock_warmup).start()
+
+
 st.markdown("""
     <div id="home-page">
 """, unsafe_allow_html=True)

pages/Multimodal_Conversational_Search.py

@@ -100,7 +100,6 @@ if "input_query" not in st.session_state:
     if(st.session_state.input_index == "covid19ie"):
         st.session_state.input_query="How many aged above 85 years died due to covid ?"#"What is the projected energy percentage from renewable sources in future?"#"Which city in United Kingdom has the highest average housing price ?"#"How many aged above 85 years died due to covid ?"# What is the projected energy from renewable sources ?"
 
-
 st.markdown("""
     <style>
     [data-testid=column]:nth-of-type(2) [data-testid=stVerticalBlock]{

utilities/invoke_models.py

@@ -7,6 +7,7 @@ from langchain_experimental.agents.agent_toolkits import create_pandas_dataframe
 #from langchain_core.prompts import ChatPromptTemplate
 from langchain_community.chat_models import BedrockChat
 import streamlit as st
+import threading
 #from transformers import VisionEncoderDecoderModel, ViTImageProcessor, AutoTokenizer
 #import torch