Run each selected flag separately, one output tab per flag

app.py

@@ -25,17 +25,22 @@ def version(tool):
     return subprocess.run([TOOLS[tool], "--version"], capture_output=True, text=True).stdout.split("release ")[-1].splitlines()[0]
 
 
-def run(file, tool, flags, extra):
-    if file is None:
-        return "Upload a cubin first."
+def run_one(tool, args, file):
     try:
-        r = subprocess.run([TOOLS[tool], *flags, *shlex.split(extra or ""), file],
-                           capture_output=True, text=True, timeout=60)
+        r = subprocess.run([TOOLS[tool], *args, file], capture_output=True, text=True, timeout=60)
     except subprocess.TimeoutExpired:
         return "Timed out after 60s."
     return r.stdout + r.stderr + (f"\n[exit code {r.returncode}]" if r.returncode else "")
 
 
+def run(file, tool, flags, extra):
+    if file is None:
+        return {"error": "Upload a cubin first."}
+    extra = shlex.split(extra or "")
+    return {" ".join([f, *extra]): run_one(tool, [f, *extra], file) for f in flags} or \
+           {" ".join(extra) or tool: run_one(tool, extra, file)}
+
+
 with gr.Blocks(title="cuobjdump / nvdisasm") as demo:
     gr.Markdown("# cuobjdump / nvdisasm\n"
                 "Disassemble CUDA cubins, fatbins and ELF objects with NVIDIA's tools from PyPI.\n\n"
@@ -44,13 +49,21 @@ with gr.Blocks(title="cuobjdump / nvdisasm") as demo:
         with gr.Column():
             file = gr.File(label="cubin / fatbin / CUDA ELF", type="filepath")
             tool = gr.Radio(list(TOOLS), value="cuobjdump", label="Tool")
-            flags = gr.CheckboxGroup(FLAGS["cuobjdump"], value=DEFAULT_FLAGS["cuobjdump"], label="Flags")
+            flags = gr.CheckboxGroup(FLAGS["cuobjdump"], value=DEFAULT_FLAGS["cuobjdump"], label="Flags (each runs separately, one tab per flag)")
             extra = gr.Textbox(label="Extra arguments", placeholder="-fun vecadd")
             btn = gr.Button("Run", variant="primary")
-        out = gr.Code(label="Output", lines=30, max_lines=60)
+        with gr.Column(scale=2):
+            results = gr.State({})
+
+            @gr.render(inputs=results)
+            def show(results):
+                with gr.Tabs():
+                    for label, output in results.items():
+                        with gr.Tab(label):
+                            gr.Code(output, label=label, lines=30, max_lines=60)
     gr.Examples([[e, t] for e in EXAMPLES for t in TOOLS], [file, tool])
     tool.change(lambda t: gr.CheckboxGroup(FLAGS[t], value=DEFAULT_FLAGS[t]), tool, flags)
-    btn.click(run, [file, tool, flags, extra], out)
+    btn.click(run, [file, tool, flags, extra], results)
 
 if __name__ == "__main__":
     demo.launch()