Dhruv-Ty commited on
Commit
97207fb
·
verified ·
1 Parent(s): beb73d5

Update src/report_generator.py

Browse files
Files changed (1) hide show
  1. src/report_generator.py +7 -1
src/report_generator.py CHANGED
@@ -440,8 +440,14 @@ def generate_and_download_report():
440
  value=st.session_state.patient_info.get("name", ""))
441
  st.text_input("Age", key="patient_age",
442
  value=st.session_state.patient_info.get("age", ""))
 
 
 
 
 
 
443
  st.selectbox("Gender", ["Male", "Female", "Other"], key="patient_gender",
444
- index=["Male", "Female", "Other"].index(st.session_state.patient_info.get("gender", "Male")))
445
 
446
  if st.button("Generate Report"):
447
  # Save patient info to session state
 
440
  value=st.session_state.patient_info.get("name", ""))
441
  st.text_input("Age", key="patient_age",
442
  value=st.session_state.patient_info.get("age", ""))
443
+
444
+ # Get gender with fallback to "Male" if empty or not in options
445
+ gender = st.session_state.patient_info.get("gender", "Male")
446
+ if gender not in ["Male", "Female", "Other"]:
447
+ gender = "Male"
448
+
449
  st.selectbox("Gender", ["Male", "Female", "Other"], key="patient_gender",
450
+ index=["Male", "Female", "Other"].index(gender))
451
 
452
  if st.button("Generate Report"):
453
  # Save patient info to session state