Update src/report_generator.py

src/report_generator.py

@@ -561,22 +561,15 @@ def generate_and_download_report():
                 if st.session_state.show_email_form:
                     show_email_form()
                 
-                # Option to start over
-                if st.button("New Report", use_container_width=True):
-                    # Reset report step
-                    st.session_state.report_step = 1
-                    # Clear PDF data
-                    if 'pdf_data' in st.session_state:
-                        del st.session_state.pdf_data
-                    # Reset patient info
-                    st.session_state.patient_info = {"name": "", "age": "", "gender": ""}
-                    # Hide email form if shown
-                    st.session_state.show_email_form = False
-                    # Force refresh
-                    st.rerun()
+                # After displaying download/email options, reset report_step to prevent re-generation
+                # UNLESS an email form is active, in which case, it will be reset by show_email_form or cancel.
+                if not st.session_state.get('show_email_form', False):
+                    st.session_state.report_step = 0 # Reset to neutral state
                     
             except Exception as e:
                 st.error(f"Error generating report: {str(e)}")
+                # Ensure report_step is reset even on error to allow trying again from step 1
+                st.session_state.report_step = 1 
                 st.button("Try Again", on_click=lambda: setattr(st.session_state, "report_step", 1))
 
 
@@ -616,6 +609,7 @@ def show_email_form():
                         st.success(message)
                         # Hide the form after successful send
                         st.session_state.show_email_form = False
+                        st.session_state.report_step = 0 # Reset report step
                         st.rerun()
                     else:
                         st.error(message)
@@ -623,4 +617,5 @@ def show_email_form():
     # Cancel button to hide email form
     if st.button("Cancel"):
         st.session_state.show_email_form = False
+        st.session_state.report_step = 0 # Reset report step
         st.rerun()