1

app.py

@@ -23,7 +23,7 @@ def anim(f_promt, s_promt, stability_key):
     args = AnimationArgs()
     args.interpolate_prompts = True
     args.locked_seed = True
-    args.max_frames = 20
+    args.max_frames = 5
     args.seed = 42
     args.strength_curve = "0:(0)"
     args.diffusion_cadence_curve = "0:(4)"
@@ -31,7 +31,7 @@ def anim(f_promt, s_promt, stability_key):
 
     animation_prompts = {
         0: f_promt,
-        10: s_promt,
+        2: s_promt,
     }
     negative_prompt = ""
 
@@ -42,23 +42,20 @@ def anim(f_promt, s_promt, stability_key):
         negative_prompt=negative_prompt,
         args=args
     )
+    # Get root folder of the repository
+    repo_dir = os.path.dirname(os.path.abspath(__file__)) 
 
-    # Render each frame of animation
-    for idx, frame in enumerate(animator.render()):
-        file_path = os.path.join("output", f"frame_{idx:05d}.png") 
-        frame.save(file_path)
-        yield file_path
-
-def file():
-    # Get current working directory 
-    app_dir = os.getcwd()
-
-    # Set output folder for files
-    output_dir = os.path.join(app_dir, "moutput")  
+    # Define output folder path
+    output_dir = os.path.join(repo_dir, "Outputs")
 
     if not os.path.exists(output_dir):
         os.makedirs(output_dir)
         print("Created folder at:"+output_dir)
+    # Render each frame of animation
+    for idx, frame in enumerate(animator.render()):
+        file_path = os.path.join(output_dir, f"frame_{idx:05d}.png") 
+        frame.save(file_path)
+        yield file_path
       
 with gr.Blocks() as demo:
     gr.Markdown("Stability Animation")
@@ -68,10 +65,6 @@ with gr.Blocks() as demo:
     
     outimg = gr.Files(label="Generated Files")
     btn = gr.Button('Anim')
-    fbtn = gr.Button('Get Dir')
-    files = gr.Text()
-
-    fbtn.click(fn=file, inputs=[], outputs=[files])
 
     btn.click(fn=anim, inputs=[f_promt, s_promt, stability_key], outputs=[outimg],api_name="AnimAPI")