zf.write(image_path, arcname=os.path.basename(image_path))

app.py

@@ -225,7 +225,7 @@ def export_images_to_zip(gallery) -> str:
     with zipfile.ZipFile(tmp_zip.name, "w", compression=zipfile.ZIP_DEFLATED) as zf:
         for i in range(len(gallery)):
             image_path = gallery[i]
-            zf.write(image_path, arcname=str(i) + "_" + os.path.basename(image_path))
+            zf.write(image_path, arcname=os.path.basename(image_path))
 
     print(str(len(gallery)) + " images zipped")
     return tmp_zip.name
@@ -265,7 +265,7 @@ def infer_example(prompt, input_images=None, mode_choice="Base (50 steps)", seed
     except:
         print("Error")
     zip_path = export_images_to_zip(gallery)
-    return [seed, zip_path]
+    return [seed, zip_path, "Done!"]
     
 
 examples = [
@@ -424,7 +424,7 @@ FLUX.2 [klein] is a fast, unified image generation and editing model designed fo
     
     with gr.Row(visible=False):
         download_button = gr.DownloadButton(elem_id="download_btn", interactive = True)
-        result_gallery = gr.Gallery(interactive = False, elem_id = "gallery1")
+        info_debug = gr.HTML(value = "")
         prompt_debug = gr.Text(
             max_lines=2,
             container=False,
@@ -442,7 +442,7 @@ FLUX.2 [klein] is a fast, unified image generation and editing model designed fo
             ],
             fn=infer_example,
             inputs=[prompt, input_image],
-            outputs=[seed, download_button],
+            outputs=[seed, download_button, info_debug],
             run_on_click=True,
             cache_examples=True,
             cache_mode='lazy'