Spaces:
Sleeping
Sleeping
Update src/streamlit_app.py
Browse files- src/streamlit_app.py +6 -9
src/streamlit_app.py
CHANGED
|
@@ -63,7 +63,6 @@ if page == "π Fitness Form":
|
|
| 63 |
|
| 64 |
name = st.text_input("Name")
|
| 65 |
|
| 66 |
-
# β
ADDED GENDER
|
| 67 |
gender = st.selectbox(
|
| 68 |
"Gender",
|
| 69 |
["Male", "Female"]
|
|
@@ -102,7 +101,7 @@ if page == "π Fitness Form":
|
|
| 102 |
category = get_category(bmi)
|
| 103 |
|
| 104 |
st.session_state["name"] = name
|
| 105 |
-
st.session_state["gender"] = gender
|
| 106 |
st.session_state["fitness_level"] = fitness_level
|
| 107 |
st.session_state["bmi"] = bmi
|
| 108 |
st.session_state["category"] = category
|
|
@@ -125,16 +124,13 @@ elif page == "π BMI Result":
|
|
| 125 |
else:
|
| 126 |
|
| 127 |
st.write(f"Name: {st.session_state['name']}")
|
| 128 |
-
st.write(f"Gender: {st.session_state['gender']}")
|
| 129 |
st.write(f"Fitness Level: {st.session_state['fitness_level']}")
|
| 130 |
st.write(f"BMI: {st.session_state['bmi']}")
|
| 131 |
st.write(f"Category: {st.session_state['category']}")
|
| 132 |
st.write(f"Goal: {st.session_state['goal']}")
|
| 133 |
st.write(f"Equipment: {', '.join(st.session_state['equipment'])}")
|
| 134 |
|
| 135 |
-
|
| 136 |
-
# β
FIXED VARIABLES FOR PROMPT
|
| 137 |
-
|
| 138 |
gender = st.session_state["gender"]
|
| 139 |
bmi = st.session_state["bmi"]
|
| 140 |
bmi_status = st.session_state["category"]
|
|
@@ -143,7 +139,7 @@ elif page == "π BMI Result":
|
|
| 143 |
equipment_list = ", ".join(st.session_state["equipment"])
|
| 144 |
|
| 145 |
|
| 146 |
-
|
| 147 |
You are a certified professional fitness trainer.
|
| 148 |
Generate a structured 5-day workout plan based on the following user profile.
|
| 149 |
|
|
@@ -187,7 +183,8 @@ Rest: 60 seconds
|
|
| 187 |
|
| 188 |
Only return the workout plan.
|
| 189 |
"""
|
| 190 |
-
|
|
|
|
| 191 |
|
| 192 |
with st.spinner("Generating AI Fitness Plan..."):
|
| 193 |
|
|
@@ -203,4 +200,4 @@ inputs = tokenizer(prompt, return_tensors="pt")
|
|
| 203 |
|
| 204 |
st.subheader("AI Fitness Plan")
|
| 205 |
|
| 206 |
-
st.write(result)
|
|
|
|
| 63 |
|
| 64 |
name = st.text_input("Name")
|
| 65 |
|
|
|
|
| 66 |
gender = st.selectbox(
|
| 67 |
"Gender",
|
| 68 |
["Male", "Female"]
|
|
|
|
| 101 |
category = get_category(bmi)
|
| 102 |
|
| 103 |
st.session_state["name"] = name
|
| 104 |
+
st.session_state["gender"] = gender
|
| 105 |
st.session_state["fitness_level"] = fitness_level
|
| 106 |
st.session_state["bmi"] = bmi
|
| 107 |
st.session_state["category"] = category
|
|
|
|
| 124 |
else:
|
| 125 |
|
| 126 |
st.write(f"Name: {st.session_state['name']}")
|
| 127 |
+
st.write(f"Gender: {st.session_state['gender']}")
|
| 128 |
st.write(f"Fitness Level: {st.session_state['fitness_level']}")
|
| 129 |
st.write(f"BMI: {st.session_state['bmi']}")
|
| 130 |
st.write(f"Category: {st.session_state['category']}")
|
| 131 |
st.write(f"Goal: {st.session_state['goal']}")
|
| 132 |
st.write(f"Equipment: {', '.join(st.session_state['equipment'])}")
|
| 133 |
|
|
|
|
|
|
|
|
|
|
| 134 |
gender = st.session_state["gender"]
|
| 135 |
bmi = st.session_state["bmi"]
|
| 136 |
bmi_status = st.session_state["category"]
|
|
|
|
| 139 |
equipment_list = ", ".join(st.session_state["equipment"])
|
| 140 |
|
| 141 |
|
| 142 |
+
prompt = f"""
|
| 143 |
You are a certified professional fitness trainer.
|
| 144 |
Generate a structured 5-day workout plan based on the following user profile.
|
| 145 |
|
|
|
|
| 183 |
|
| 184 |
Only return the workout plan.
|
| 185 |
"""
|
| 186 |
+
|
| 187 |
+
inputs = tokenizer(prompt, return_tensors="pt")
|
| 188 |
|
| 189 |
with st.spinner("Generating AI Fitness Plan..."):
|
| 190 |
|
|
|
|
| 200 |
|
| 201 |
st.subheader("AI Fitness Plan")
|
| 202 |
|
| 203 |
+
st.write(result)
|