fix: improve input validation in arithmetic and average programs

Author: nithish

basics/02_variables_types/02_average.py

@@ -1,4 +1,19 @@
-first_number = int(input("Enter the First Number : "))
-second_number = int(input("Enter the Second Number : "))
-average = (first_number + second_number) / 2
-print(f"Avg of {first_number} and {second_number} is {average}")
+# Average of Two Numbers
+
+while True:
+    try:
+        num1 = int(input("Enter first number: "))
+        break
+    except ValueError:
+        print("Invalid input! Please enter a valid integer (e.g., 10, -5)")
+
+while True:
+    try:
+        num2 = int(input("Enter second number: "))
+        break
+    except ValueError:
+        print("Invalid input! Please enter a valid integer (e.g., 10, -5)")
+
+average = (num1 + num2) / 2
+
+print(f"The average of {num1} and {num2} is {average}")