Update app.py

app.py

@@ -1,28 +1,56 @@
 import gradio as gr
+import multiprocessing
 import time
+from math import isqrt
 
+# Function to check if a number is prime
 def is_prime(n):
     if n < 2:
         return False
-    for i in range(2, int(n**0.5) + 1):
-        if n % i == 0:
+    if n in (2, 3):
+        return True
+    if n % 2 == 0 or n % 3 == 0:
+        return False
+    for i in range(5, isqrt(n) + 1, 6):
+        if n % i == 0 or n % (i + 2) == 0:
             return False
     return True
 
-def prime_generator():
-    num = 2
-    while True:
-        if is_prime(num):
-            yield str(num)  # Yield the prime number as a string
-        num += 1
-        time.sleep(0.5)  # Adjust the sleep time to control the speed of prime generation
+# Function to find the next prime number in a given range
+def find_primes_in_range(start, end, queue):
+    primes = [str(num) for num in range(start, end) if is_prime(num)]
+    queue.put(primes)
 
+# Function that utilizes multiple processes to find primes
 def start_prime_printer():
-    prime_stream = prime_generator()
+    num = 2  # Start from the first prime
+    batch_size = 1000  # Size of the range each process will handle
     output = []
-    for prime in prime_stream:
-        output.append(prime)
-        yield "\n".join(output)  # Update the output in the Gradio interface
+    num_processes = multiprocessing.cpu_count()  # Utilize all available CPU cores
+
+    while True:
+        processes = []
+        queue = multiprocessing.Queue()
+        
+        # Create and start processes
+        for i in range(num_processes):
+            start = num + i * batch_size
+            end = start + batch_size
+            process = multiprocessing.Process(target=find_primes_in_range, args=(start, end, queue))
+            process.start()
+            processes.append(process)
+        
+        # Collect results from all processes
+        for process in processes:
+            process.join()  # Wait for process to finish
+            primes = queue.get()  # Get primes from this process
+            output.extend(primes)
+        
+        # Display the collected prime numbers
+        yield "\n".join(output)
+        
+        num += num_processes * batch_size
+        time.sleep(0.1)  # Adjust the sleep time if needed to manage UI responsiveness
 
 # Gradio Interface
 interface = gr.Interface(