1

app.py

@@ -45,10 +45,13 @@ def anim(f_promt, s_promt, stability_key):
 
     # Render each frame of animation
     for idx, frame in enumerate(animator.render()):
-        file_path = f"frame_{idx:05d}.png"
+        file_path = os.path.join("output", f"frame_{idx:05d}.png") 
         frame.save(file_path)
-        print("saved: "+file_path)
-        yield file_path  # Yield the path to the generated frame
+        yield file_path
+
+def file():
+    app_dir = os.getcwd()
+    return app_dir
       
 with gr.Blocks() as demo:
     gr.Markdown("Stability Animation")
@@ -58,7 +61,11 @@ with gr.Blocks() as demo:
     
     outimg = gr.Files(label="Generated Files")
     btn = gr.Button('Anim')
- 
+    fbtn = gr.Button('Get Dir')
+    files = gr.Textbox("dir")
+
+    btn.click(fn=file, inputs=[], outputs=[files])
+
     btn.click(fn=anim, inputs=[f_promt, s_promt, stability_key], outputs=[outimg],api_name="AnimAPI")
 
 demo.launch()