Added Zip instead of single files to make loading in Hueforge correct

app.py

@@ -71,7 +71,6 @@ gradio.blocks.Blocks.call_function = sentry_call_fn
 import gradio as gr
 import pandas as pd
 import os
-import subprocess
 import time
 import sys
 from datetime import datetime
@@ -333,7 +332,7 @@ def run_autoforge_process(cmd, log_path):
 
     exit_code = 0
     with open(log_path, "w", buffering=1, encoding="utf-8") as log_f, \
-         redirect_stdout(log_f), redirect_stderr(log_f), parallel_backend("threading", n_jobs=-1):
+            redirect_stdout(log_f), redirect_stderr(log_f), parallel_backend("threading", n_jobs=-1):
         try:
             sys.argv = ["autoforge"] + cli_args
             autoforge_main.main()         # runs until completion
@@ -532,7 +531,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
                     )
                     expected_cols = ["Brand", " Name", " TD", " Color"]
                     if not all(
-                        col in loaded_script_df.columns for col in expected_cols
+                            col in loaded_script_df.columns for col in expected_cols
                     ):
                         gr.Error(
                             f"CSV must contain columns: {', '.join(expected_cols)}. Found: {loaded_script_df.columns.tolist()}"
@@ -544,8 +543,8 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
                         )
                         current_script_df = filament_df_state.value
                         if (
-                            current_script_df is not None
-                            and not current_script_df.empty
+                                current_script_df is not None
+                                and not current_script_df.empty
                         ):
                             return current_script_df.rename(
                                 columns={
@@ -583,8 +582,8 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
 
             def save_filaments_to_file_for_download(current_script_df_from_state):
                 if (
-                    current_script_df_from_state is None
-                    or current_script_df_from_state.empty
+                        current_script_df_from_state is None
+                        or current_script_df_from_state.empty
                 ):
                     gr.Warning("Filament table is empty. Nothing to save.")
                     return None
@@ -742,8 +741,8 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
 
             with gr.Row():
                 download_results = gr.File(
-                    label="Download results",
-                    file_count="multiple",
+                    label="Download Results (zip)",
+                    file_count="single",
                     interactive=True,
                     visible=False,
                 )
@@ -751,9 +750,9 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
     # --- Backend Function for Running the Script ---
     @spaces.GPU(duration=150)
     def execute_autoforge_script(
-        current_filaments_df_state_val, input_image, *accordion_param_values
+            current_filaments_df_state_val, input_image, *accordion_param_values
     ):
-        
+
         log_output = []
 
         # 0. Validate Inputs
@@ -769,8 +768,8 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
 
         # 1. Save current filaments
         if (
-            current_filaments_df_state_val is None
-            or current_filaments_df_state_val.empty
+                current_filaments_df_state_val is None
+                or current_filaments_df_state_val.empty
         ):
             gr.Error("Filament table is empty. Please add filaments.")
             capture_exception(
@@ -826,9 +825,9 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
                     item for item in get_script_args_info() if item["name"] == arg_name
                 ]  # get full list to check type
                 if (
-                    arg_info_list
-                    and arg_info_list[0]["type"] == "checkbox"
-                    and arg_widget_val is False
+                        arg_info_list
+                        and arg_info_list[0]["type"] == "checkbox"
+                        and arg_widget_val is False
                 ):
                     continue
                 else:
@@ -906,7 +905,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
                     exc_str = exc_text(e)
                     self.exc = e
                     capture_exception(e)  # still goes to Sentry
-                    
+
                     # make the error visible in the UI console
                     with open(self.log_path, "a", encoding="utf-8") as lf:
                         lf.write(
@@ -923,11 +922,11 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
         try:
             worker = Worker(command, log_file)
             worker.start()
-    
+
             preview_mtime = 0
             last_push = 0
             file_pos = 0  # how far we've read
-    
+
             while worker.is_alive() or file_pos < os.path.getsize(log_file):
                 # read any new console text
                 with open(log_file, "r", encoding="utf-8") as lf:
@@ -935,7 +934,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
                     new_txt = lf.read()
                     file_pos = lf.tell()
                     log_output += new_txt
-    
+
                 now = time.time()
                 if now - last_push >= 1.0:  # one-second UI tick
                     current_preview = _maybe_new_preview()
@@ -945,9 +944,9 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
                         gr.update(),  # placeholder for download widget
                     )
                     last_push = now
-    
+
                 time.sleep(0.05)
-    
+
             worker.join()  # make sure it’s done
         except RuntimeError as e:
             # Show toast to user
@@ -956,10 +955,10 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
             capture_exception(e)
 
             with open(log_file, "r", encoding="utf-8") as lf:
-                    lf.seek(file_pos)
-                    new_txt = lf.read()
-                    file_pos = lf.tell()
-                    log_output += new_txt
+                lf.seek(file_pos)
+                new_txt = lf.read()
+                file_pos = lf.tell()
+                log_output += new_txt
             yield (
                 "".join(log_output),
                 current_preview,
@@ -1034,6 +1033,20 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
         if out_png is None:
             log_output += "\nWarning: final_model.png not found in output."
 
+        zip_path = None
+        if files_to_offer:
+            zip_path = os.path.join(run_output_dir_val, "autoforge_results.zip")
+            log_output += f"\nZipping results to {os.path.basename(zip_path)}..."
+            try:
+                with zipfile.ZipFile(zip_path, "w", compression=zipfile.ZIP_STORED) as zf:
+                    for f in files_to_offer:
+                        zf.write(f, os.path.basename(f))
+                log_output += " done."
+            except Exception as e:
+                capture_exception(e)
+                log_output += f"\nError creating zip file: {e}"
+                zip_path = None # Don't offer a broken zip
+
         sentry_sdk.capture_event(  # moved inside the same scope
             {
                 "message": "Autoforge process finished",
@@ -1084,9 +1097,9 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
             "".join(log_output),  # progress_output
             out_png,  # final_image_preview (same as before)
             gr.update(  # download_results
-                value=files_to_offer,
-                visible=True,
-                interactive=True,
+                value=zip_path,
+                visible=bool(zip_path),
+                interactive=bool(zip_path),
             ),
         )