kanakkk789 commited on
Commit
7e181cd
Β·
verified Β·
1 Parent(s): 4d17849

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. 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 # βœ… added
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']}") # βœ… added
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
- prompt = f"""
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
- inputs = tokenizer(prompt, return_tensors="pt")
 
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)