Merge pull request #58 from jyotish6699/fix/improve-calculator-input-flow

fix(calculator): retry invalid operands and prevent division/modulus by zero

Author: steam-bell-92

math/Simple-Calculator/Simple-Calculator.py

@@ -20,42 +20,49 @@
         break
 
     if choice in ['1', '2', '3', '4', '5', '6']:
-        try:
-            num1 = float(input("\n📥 Enter first number: "))
-            num2 = float(input("📥 Enter second number: "))
-            
-            if choice == '1':
-                result = num1 + num2
-                print(f"\n✨ Result: {num1} + {num2} = {result}\n")
-            
-            elif choice == '2':
-                result = num1 - num2
-                print(f"\n✨ Result: {num1} - {num2} = {result}\n")
-            
-            elif choice == '3':
-                result = num1 * num2
-                print(f"\n✨ Result: {num1} × {num2} = {result}\n")
-            
-            elif choice == '4':
-                if num2 == 0:
-                    print("\n❌ Error! Division by zero is not allowed.\n")
-                else:
-                    result = num1 / num2
-                    print(f"\n✨ Result: {num1} ÷ {num2} = {result}\n")
-            
-            elif choice == '5':
-                if num2 == 0:
-                    print("\n❌ Error! Modulus by zero is not allowed.\n")
-                else:
-                    result = num1 % num2
-                    print(f"\n✨ Result: {num1} % {num2} = {result}\n")
+        while True:
+            try:
+                num1 = float(input("\n📥 Enter first number: "))
+
+                num2 = float(input("📥 Enter second number: "))
+                
+                if choice == '1':
+                    result = num1 + num2
+                    print(f"\n✨ Result: {num1} + {num2} = {result}\n")
+                
+                elif choice == '2':
+                    result = num1 - num2
+                    print(f"\n✨ Result: {num1} - {num2} = {result}\n")
+                
+                elif choice == '3':
+                    result = num1 * num2
+                    print(f"\n✨ Result: {num1} × {num2} = {result}\n")
+                
+                elif choice == '4':
+                    if num2 == 0:
+                        print("\n❌ Error! Division by zero is not allowed.\n")
+                        continue
+                    else:
+                        result = num1 / num2
+                        print(f"\n✨ Result: {num1} ÷ {num2} = {result}\n")
+                
+                elif choice == '5':
+                    if num2 == 0:
+                        print("\n❌ Error! Modulus by zero is not allowed.\n")
+                        continue
+                    else:
+                        result = num1 % num2
+                        print(f"\n✨ Result: {num1} % {num2} = {result}\n")
+                
+                elif choice == '6':
+                    result = num1 ** num2
+                    print(f"\n✨ Result: {num1} ^ {num2} = {result}\n")
 
-            elif choice == '6':
-                result = num1 ** num2
-                print(f"\n✨ Result: {num1} ^ {num2} = {result}\n")
-        
-        except ValueError:
-            print("\n❌ Invalid input! Please enter valid numbers.\n")
+            except ValueError:
+                print("\n❌ Invalid input! Please enter valid numbers.\n")
+                continue
+
+            break
 
     else:
         print("\n❌ Invalid choice! Please select 1-7.\n")