import csv

def write_into_csv(info_list):
    with open('student_info.csv', 'a', newline='') as csv_file:
        writer = csv.writer(csv_file)
        if csv_file.tell() == 0:
            writer.writerow(["Name", "Age", "Contact_Number", "E-mail_ID"])

        writer.writerow(info_list)

if __name__ == '__main__':
     condition = True
     num = 1

     while(condition):
         student_info = input("Enter student information for student #{} in the format (Name Age Contact_Number E-mail-ID)".format(num))
         print("Entered student information" + student_info)

         student_info_list = student_info.split(" ")
         print("Entered split information:" + str(student_info_list))

         print("\nEntered information is: \nName: {}\nAge: {}\nContact_number: {}\nE-mail_ID: {}".format(student_info_list[0],student_info_list[1],
         student_info_list[2],student_info_list[3]))

         choice_check = input("Is the entered information correct (yes/no)")
         if choice_check == "yes":
              write_into_csv(student_info_list)
              condition_check = input("Enter (yes/no) if you want to enter information for another student")
          
              if condition_check == "yes":
                  condition = True
                  num = num + 1
              elif condition_check == "no":
                  condition = False
         elif choice_check == "no":
             print("\n Please enter valid information\n")
