Making the GIF Downloadable in Gradio

app.py

@@ -31,7 +31,7 @@ def simulate_and_render(
     run_counter += 1
     unique_id = f"{seed}_{run_counter}_{int(time.time())}"  # Unique identifier for this run
     
-    gif_path,physics_state, json_path  = run_tray_simulation(
+    gif_path, physics_state, json_path  = run_tray_simulation(
         seed=seed,
         azimuth=azimuth,
         elevation=elevation,
@@ -72,11 +72,15 @@ with gr.Blocks(
                 height=600  # Set default height
             )
 
+            output_file = gr.File(label="Download Simulation GIF")
+            state_output = gr.JSON(label="Physics State")
+            json_file = gr.File(label="Download Physics State JSON")
+
     # Connect the button to the simulation function
     submit_button.click(
         fn=simulate_and_render,
         inputs=[seed_input, azimuth_input, elevation_input, distance_input],
-        outputs=output_image
+        outputs=[output_image, output_file, state_output, json_file]
     )
 
 if __name__ == "__main__":