Update main.py with csv read/write functionality.

Author: Kingstead

input.csv

@@ -0,0 +1,30 @@
+Amoxicillin 500 mg: Take 1 capsule by mouth every 8 hours.
+Metformin 500 mg: Take 1 tablet by mouth twice daily with meals.
+Lisinopril 10 mg: Take 1 tablet by mouth once daily.
+Atorvastatin 20 mg: Take 1 tablet by mouth once daily in the evening.
+Omeprazole 20 mg: Take 1 capsule by mouth once daily before breakfast.
+Levothyroxine 50 mcg: Take 1 tablet by mouth once daily on an empty stomach.
+Hydrochlorothiazide 25 mg: Take 1 tablet by mouth once daily in the morning.
+Gabapentin 300 mg: Take 1 capsule by mouth three times daily.
+Albuterol Inhaler: Inhale 2 puffs by mouth every 4-6 hours as needed for wheezing.
+Ibuprofen 400 mg: Take 1 tablet by mouth every 6 hours as needed for pain.
+Sertraline 50 mg: Take 1 tablet by mouth once daily.
+Amlodipine 5 mg: Take 1 tablet by mouth once daily.
+Prednisone 10 mg: Take 1 tablet by mouth once daily for 5 days.
+Ciprofloxacin 500 mg: Take 1 tablet by mouth every 12 hours for 7 days.
+Clonazepam 0.5 mg: Take 1 tablet by mouth twice daily.
+Montelukast 10 mg: Take 1 tablet by mouth once daily in the evening.
+Tramadol 50 mg: Take 1 tablet by mouth every 4-6 hours as needed for pain.
+Fluoxetine 20 mg: Take 1 capsule by mouth once daily.
+Metoprolol 50 mg: Take 1 tablet by mouth twice daily.
+Furosemide 40 mg: Take 1 tablet by mouth once daily in the morning.
+Lorazepam 1 mg: Take 1 tablet by mouth every 8 hours as needed for anxiety.
+Simvastatin 20 mg: Take 1 tablet by mouth once daily in the evening.
+Doxycycline 100 mg: Take 1 tablet by mouth twice daily for 10 days.
+Trazodone 50 mg: Take 1 tablet by mouth at bedtime.
+Warfarin 5 mg: Take 1 tablet by mouth once daily.
+Zolpidem 10 mg: Take 1 tablet by mouth at bedtime as needed for sleep.
+Cyclobenzaprine 10 mg: Take 1 tablet by mouth three times daily.
+Bupropion 150 mg: Take 1 tablet by mouth once daily.
+Meloxicam 15 mg: Take 1 tablet by mouth once daily.
+Ranitidine 150 mg: Take 1 tablet by mouth twice daily.

main.py

@@ -12,15 +12,15 @@ def get_input():
         try:
             value = int(
                 input(
-                    "Enter [1] for a single sig. Enter [2] for bulk sigs via .csv file: "
+                    "Enter 1 for a single sig. Enter 2 for bulk sigs via .csv file: "
                 )
             )
-            if value not in (1, 2):
-                pass
-            else:
+            if value in (1, 2):
                 return value
+            else:
+                print("Invalid value. Enter either a 1 or 2 to continue.")
         except ValueError:
-            pass
+            print("Invalid value enter a 1 or 2 to continue.")
 
 
 def generate_output(n):
@@ -35,12 +35,18 @@ def generate_output(n):
                     " "
                 )
                 if inputfile.endswith(".csv") and outputfile.endswith(".csv"):
-                    with open(inputfile, "r") as file:
-                        print(file)
+                    with open(inputfile, "r") as infile, open(outputfile, "w", newline='') as outfile:
+                        reader = csv.reader(infile)
+                        writer = csv.writer(outfile)
+                        for row in reader:
+                            parsed_sig = SigParser().parse(row[0])
+                            writer.writerow([parsed_sig])
+                    print(f"Output written to {outputfile}")
+                    break
                 else:
                     print("Both files must end with .csv. Please try again.")
             except ValueError:
-                pass
+                print("Invalid. Enter input and output file names separated by a space.")
             except FileNotFoundError:
                 print("Input file not found. Please try again.")
                 pass