Update app.py

app.py

@@ -97,9 +97,8 @@ def _run(file, gk, mk, gek, n_trials, progress=gr.Progress(track_tqdm=True)):
     sp = r.get("sheet_paths",{})
     mdf = pd.DataFrame(r.get("mismatch_table",[]))
     progress(1.0, desc="✅ Done!")
-    return (summary, fig, pfig, tl_show, cdf, s1, s2, s3, s4,
-            sp.get(1), sp.get(2), sp.get(3), sp.get(4),
-            r.get("json_path"), mdf)
+    dl_files = [sp.get(i) for i in [1,2,3,4]] + [r.get("json_path")]
+    return (summary, fig, pfig, tl_show, cdf, s1, s2, s3, s4, dl_files, mdf)
 
 # ── UI ───────────────────────────────────────────────────────────────────────
 css = ".gradio-container{background:#0d1117!important;color:#c9d1d9!important}" \
@@ -126,23 +125,19 @@ with gr.Blocks(theme=gr.themes.Base(primary_hue="blue",neutral_hue="slate"),
                 with gr.Tab("Pareto Front"): pareto_out = gr.Plot()
                 with gr.Tab("Trial Log"): trial_out = gr.Dataframe()
                 with gr.Tab("Clusters"): cluster_out = gr.Dataframe()
-                with gr.Tab("Sheet 1 — Groq"):
-                    s1_out = gr.Dataframe(); s1_dl = gr.File(label="⬇ Download")
-                with gr.Tab("Sheet 2 — Mistral"):
-                    s2_out = gr.Dataframe(); s2_dl = gr.File(label="⬇ Download")
-                with gr.Tab("Sheet 3 — Gemini"):
-                    s3_out = gr.Dataframe(); s3_dl = gr.File(label="⬇ Download")
-                with gr.Tab("Sheet 4 — Consolidated"):
-                    s4_out = gr.Dataframe(); s4_dl = gr.File(label="⬇ Download")
-                with gr.Tab("Downloads"):
-                    json_dl = gr.File(label="topics.json")
+                with gr.Tab("Sheet 1 — Groq"): s1_out = gr.Dataframe()
+                with gr.Tab("Sheet 2 — Mistral"): s2_out = gr.Dataframe()
+                with gr.Tab("Sheet 3 — Gemini"): s3_out = gr.Dataframe()
+                with gr.Tab("Sheet 4 — Consolidated"): s4_out = gr.Dataframe()
                 with gr.Tab("RQ Mismatch"): mismatch_out = gr.Dataframe()
+                with gr.Tab("Downloads"):
+                    dl_out = gr.File(label="All sheet CSVs + topics.json",
+                                     file_count="multiple")
     file_in.change(_preview, inputs=[file_in], outputs=[preview_out])
     run_btn.click(_run,
         inputs=[file_in, groq_in, mistral_in, gemini_in, trials_in],
         outputs=[summary_out, scatter_out, pareto_out, trial_out, cluster_out,
-                 s1_out, s2_out, s3_out, s4_out,
-                 s1_dl, s2_dl, s3_dl, s4_dl, json_dl, mismatch_out])
+                 s1_out, s2_out, s3_out, s4_out, dl_out, mismatch_out])
 
 if __name__ == "__main__":
     demo.launch(server_name="0.0.0.0", server_port=7860)