Spaces:
Running
Running
| from pathlib import Path | |
| import sys | |
| import subprocess | |
| import traceback | |
| import psutil | |
| import shlex | |
| import shutil | |
| import os | |
| import gradio as gr | |
| # ── File Save ────────────────────────────────────────────── | |
| def save_file(file): | |
| if file is None: | |
| return "❌ No file uploaded." | |
| dest = os.path.join(".", os.path.basename(file.name)) | |
| shutil.copy(file.name, dest) | |
| return f"✅ Saved to: {dest}" | |
| # ── Streaming runner ──────────────────────────────────────── | |
| def run_stream(cmd, timeout=300): | |
| try: | |
| process = subprocess.Popen( | |
| cmd, | |
| stdout=subprocess.PIPE, | |
| stderr=subprocess.PIPE, # stderr alag | |
| text=True, | |
| bufsize=1 | |
| ) | |
| output = "" | |
| while True: | |
| line = process.stdout.readline() | |
| if not line and process.poll() is not None: | |
| break | |
| if line: | |
| output += line | |
| yield output | |
| stderr_out = process.stderr.read() | |
| if stderr_out: | |
| output += f"\n⚠️ STDERR:\n{stderr_out}" | |
| yield output | |
| try: | |
| process.wait(timeout=timeout) | |
| yield output + "\n✅ Finished" | |
| except subprocess.TimeoutExpired: | |
| process.kill() | |
| yield output + "\n⏰ Timeout — Process killed" | |
| except Exception: | |
| yield f"❌ Error:\n{traceback.format_exc()}" | |
| # ── Tabs ──────────────────────────────────────────────────── | |
| def run_python(code): | |
| yield from run_stream([sys.executable, "-u", "-c", code]) | |
| def install_package(pkgs): | |
| if not pkgs.strip(): | |
| yield "❌ No package name entered." | |
| return | |
| yield from run_stream(["pip", "install"] + pkgs.split()) | |
| def run_linux(cmd): | |
| yield from run_stream(shlex.split(cmd)) | |
| def file_func(paths, op): | |
| if not paths: | |
| return None | |
| if op == "download": | |
| return paths | |
| elif op == "remove": | |
| removed, failed = [], [] | |
| for p in paths: | |
| try: | |
| Path(p).unlink(missing_ok=True) | |
| removed.append(p) | |
| except Exception as e: | |
| failed.append(f"{p}: {e}") | |
| msg = "" | |
| if removed: msg += "✅ Removed:\n" + "\n".join(removed) | |
| if failed: msg += "\n❌ Failed:\n" + "\n".join(failed) | |
| return msg | |
| def system_monitor(): | |
| vm = psutil.virtual_memory() | |
| disk = psutil.disk_usage("/") | |
| lines = [ | |
| f"🖥️ CPU Usage : {psutil.cpu_percent(interval=1)}%", | |
| f"🧠 RAM Usage : {vm.percent}% ({vm.used//1024**2} MB / {vm.total//1024**2} MB)", | |
| f"💾 Disk Usage : {disk.percent}% ({disk.used//1024**3} GB / {disk.total//1024**3} GB)", | |
| "\n⚙️ Top Processes (by CPU):", | |
| ] | |
| procs = sorted(psutil.process_iter(['pid','name','cpu_percent','memory_percent']), | |
| key=lambda p: p.info['cpu_percent'] or 0, reverse=True)[:10] | |
| for p in procs: | |
| lines.append(f" PID {p.info['pid']:>6} | {p.info['name']:<25} | " | |
| f"CPU {p.info['cpu_percent']:>5.1f}% | RAM {p.info['memory_percent']:>5.1f}%") | |
| return "\n".join(lines) | |
| # ── UI ────────────────────────────────────────────────────── | |
| tabs = gr.TabbedInterface( | |
| [ | |
| gr.Interface(save_file, gr.File(), gr.Textbox(label="Status"), title="Upload"), | |
| gr.Interface(run_linux, gr.Code(language="shell", label="Shell Command"), gr.Textbox(lines=20, label="Output")), | |
| gr.Interface(install_package, gr.Textbox(label="Package(s) e.g. numpy pandas"), gr.Textbox(lines=10, label="pip Output")), | |
| gr.Interface(file_func, [gr.FileExplorer(label="Files"), | |
| gr.Dropdown(["download","remove"], label="Action")], | |
| gr.Files(label="Result")), | |
| gr.Interface(run_python, gr.Code(language="python", label="Python Code"), gr.Textbox(lines=20, label="Output")), | |
| gr.Interface(system_monitor, None, gr.Textbox(lines=15, label="System Info")), | |
| ], | |
| tab_names=["📁 File Upload", "🐧 Shell", "📦 Pip Install", "📂 File Manager", "🐍 Python Runner", "⚙️ System Monitor"] | |
| ) | |
| tabs.launch() |