Update app.py

app.py

@@ -16,7 +16,8 @@ import time
 import os
 import re
 
-def main():
+@st.cache
+def get_args():
     st.set_page_config(layout="wide")
 
     # -- 1. Setup arguments
@@ -33,9 +34,22 @@ def main():
     logger = get_logger(__name__)
 
     # -- 3. Setup constants
+    args = parser.parse_args()
+    return args
+
+@st.cache
+def get_podcast_data(transcription_path):
+    together.api_key = os.environ["TOGETHER_API_KEY"]
+    together.Models.start(MODEL)
+    podcast_url_video_df = pd.read_csv(PODCAST_URL_VIDEO_PATH, sep=';')
+    return podcast_url_video_df
+    
+def main():
+    args = get_args()
     B_INST, E_INST = "[INST]", "[/INST]"
     B_SYS, E_SYS   = "<<SYS>>\n", "\n<</SYS>>\n\n"
-    args = parser.parse_args()
+
+    # -- 4. Get parameters
     PODCAST_URL_VIDEO_PATH     = args.PODCAST_URL_VIDEO_PATH
     DEFAULT_SYSTEM_PROMPT_LINK = args.DEFAULT_SYSTEM_PROMPT_LINK
     TRANSCRIPTION              = args.TRANSCRIPTION
@@ -43,9 +57,7 @@ def main():
     MODEL                      = args.MODEL
     EMB_MODEL                  = args.EMB_MODEL
 
-    together.api_key = os.environ["TOGETHER_API_KEY"]
-    together.Models.start(MODEL)
-    podcast_url_video_df = pd.read_csv(PODCAST_URL_VIDEO_PATH, sep=';')
+    podcast_url_video_df = get_podcast_data(TRANSCRIPTION_PATH)
 
     r = requests.get("https://raw.githubusercontent.com/AlbertoUAH/Castena/main/media/castena-animated-icon.gif", stream=True)
     icon = Image.open(r.raw)