Update app.py

app.py

@@ -1,4 +1,41 @@
+import os
+import time
+from datetime import timedelta
+import numpy as np
+from threading import Thread
+
 import streamlit as st
 
-x = st.slider('Select a value')
-st.write(x, 'squared is', x * x)
+def cpu_benchmark(cores, iterations=1000000):
+    start = time.perf_counter()
+    
+    num_threads = min([int(core * iterations / cores) for core in range(cores)] + [1])
+    threads = []
+
+    results = []
+    for i in range(cores):
+        t = Thread(target=np.random.randn, args=(num_threads[i],))
+        threads.append(t)
+        t.start()
+
+    for t in threads:
+        t.join()
+
+    end = time.perf_counter()
+    elapsed = end - start
+    return elapsed / iterations, elapsed
+
+def main():
+    st.title("CPU Benchmark")
+    st.write("Select the number of cores to use for the benchmark.")
+
+    cores = int(st.slider("Number of Cores", 1, os.cpu_count(), step=1))
+    submit_button = st.button("Run Benchmark")
+
+    if submit_button:
+        elapsed_time, total_time = cpu_benchmark(cores)
+        result = f"Elapsed Time per Iteration: {elapsed_time:.6f} seconds\nTotal Time: {total_time:.6f} seconds"
+        st.success(result)
+
+if __name__ == "__main__":
+    main()