UI

app.py

@@ -40,7 +40,7 @@ display_workout = False
 # First column containers
 with col1:
     
-    st.subheader("LLM answering")
+    st.subheader("Coach chat")
 
     if "messages" not in st.session_state:
         st.session_state.messages = []
@@ -48,11 +48,11 @@ with col1:
     if "file_name" not in st.session_state:
         st.session_state.file_name = None
 
-    for message in st.session_state.messages:
-        with st.chat_message(message["role"]):
-            st.markdown(message["content"])
+    # for message in st.session_state.messages:
+    #     with st.chat_message(message["role"]):
+    #         st.markdown(message["content"])
 
-    if prompt := st.chat_input("What is up?"):
+    if prompt := st.chat_input("What's up?"):
         st.session_state.messages.append({"role": "user", "content": prompt})
         with st.chat_message("user"):
             st.markdown(prompt)
@@ -63,7 +63,7 @@ with col1:
             print(type(direction))
             print(direction)
             if direction=='fitness_advices':
-                with st.spinner("Thinking..."):
+                with st.spinner("Retrieving relevant data..."):
                     response = rag_chain.invoke(
                                 prompt
                                 )
@@ -80,12 +80,13 @@ with col1:
                         {"input" : prompt}
                     )["output"]
             else:
-                response = "Sure! I just made a workout for you. Check on the table I just provided you."
-                json_output = workout_chain.invoke({"query":prompt})
-                exercises_list = json_output['exercises']
-                workout_df = pd.DataFrame(exercises_list)
-                workout_df.columns = ["exercice", "nombre_series", "nombre_repetitions", "temps_repos"]
-                display_workout=True
+                with st.spinner("Creating workout program..."):
+                    response = "Sure! I just made a workout for you. Check on the table I just provided you."
+                    json_output = workout_chain.invoke({"query":prompt})
+                    exercises_list = json_output['exercises']
+                    workout_df = pd.DataFrame(exercises_list)
+                    workout_df.columns = ["exercice", "nombre_series", "nombre_repetitions", "temps_repos"]
+                    display_workout=True
             print(type(response))
             st.session_state.messages.append({"role": "assistant", "content": response})
             st.markdown(response)
@@ -97,7 +98,7 @@ with col1:
 with col2:
     # st.subheader("Sports Agenda")
         # TO DO
-    st.subheader("Video Analysis")
+    st.subheader("Technique Analysis")
 
     video_uploaded = st.file_uploader("Choose a video file", type=["mp4", "ogg", "webm", "MOV"])
     if video_uploaded: