Fix error of "return to settings"

app.py

@@ -502,20 +502,32 @@ def main():
                         st.rerun()
                     else:
                         st.error("No output video was generated. Check logs for details.")
-                        if st.button("Return to Settings"):
-                            return_to_input()
+                        return_to_input_button = st.button("Return to Settings")
+                        if return_to_input_button:
+                            st.session_state.page = 'input'
+                            st.session_state.temp_dir = None
+                            st.session_state.final_video_path = None
+                            st.session_state.process_started = False
                             st.rerun()
                         
                 except subprocess.CalledProcessError as e:
                     st.error(f"Error running morphing pipeline: {e}")
-                    if st.button("Return to Settings"):
-                        return_to_input()
+                    return_to_input_button = st.button("Return to Settings")
+                    if return_to_input_button:
+                        st.session_state.page = 'input'
+                        st.session_state.temp_dir = None
+                        st.session_state.final_video_path = None
+                        st.session_state.process_started = False
                         st.rerun()
                 
             except Exception as e:
                 st.error(f"An error occurred during processing: {e}")
-                if st.button("Return to Settings"):
-                    return_to_input()
+                return_to_input_button = st.button("Return to Settings")
+                if return_to_input_button:
+                    st.session_state.page = 'input'
+                    st.session_state.temp_dir = None
+                    st.session_state.final_video_path = None
+                    st.session_state.process_started = False
                     st.rerun()
 
     # =============== RESULT PAGE ===============