UI

app.py

@@ -21,14 +21,19 @@ from utils import save_uploaded_file, encode_video_H264
 
 
 st.set_page_config(layout="wide", initial_sidebar_state="collapsed")
+
+st.title("Welcome to the AI Bros playground! 🏋️‍♂️🤖🏋️‍♀️")
+
 # Create two columns
 col1, col2 = st.columns(2)
 video_uploaded = None
 llm = ChatMistralAI(model="mistral-large-latest", mistral_api_key=mistral_api_key, temperature=0)
 prompt = ChatPromptTemplate.from_template(
-    template =""" You are a personal AI sports coach with an expertise in nutrition and fitness. 
+    template =""" You are a personal coach with an expertise in nutrition and fitness. 
     You are having a conversation with your client, which is either a beginner or an advanced athlete. 
-    You must be gentle, kind, and motivative.
+    You are motivative and have very good vibe.
+    Call your client buddy or sweatheart.
+    You are a human, not a robot.
     Always try to answer concisely to the queries.
     
     User: {question}
@@ -85,7 +90,7 @@ with col1:
                     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"]
+                    workout_df.columns = ["exercice", "sets", "reps", "rest"]
                     display_workout=True
             print(type(response))
             st.session_state.messages.append({"role": "assistant", "content": response})