1

app.py

@@ -42,8 +42,13 @@ def anim(f_promt, s_promt):
     )
 
     # Render each frame of animation
+    file_paths = []
     for idx, frame in enumerate(animator.render()):
-        frame.save(f"frame_{idx:05d}.png")
+        file_path = f"frame_{idx:05d}.png"
+        frame.save(file_path)
+        file_paths.append(file_path)
+
+    return file_paths
 
 with gr.Blocks() as demo:
     gr.Markdown("Stability Animation")
@@ -53,4 +58,7 @@ with gr.Blocks() as demo:
     btn = gr.Button('Anim')
     btn.click(fn=anim, inputs=[f_promt,s_promt], outputs=[outimg],api_name="AnimAPI")
 
+    # Add a File component to display the generated files
+    gr.Files(label="Generated Files", source=btn)
+
 demo.launch()