Update app.py

app.py

@@ -1,15 +1,13 @@
-#@title Connect to the Stability API
 import os
 import gradio as gr
 from stability_sdk.api import Context
-from stability_sdk.animation_ui import create_ui
+from stability_sdk.animation import AnimationArgs, Animator
 
 from dotenv import load_dotenv
 
 load_dotenv(".env") 
 
-# @markdown To get your API key visit https://dreamstudio.ai/account
-STABILITY_HOST = "grpc.stability.ai:443" #@param {type:"string"}
+STABILITY_HOST = "grpc.stability.ai:443" 
 STABILITY_KEY = os.getenv("STABILITY_KEY")
 
 # Connect to Stability API
@@ -19,12 +17,42 @@ context = Context(STABILITY_HOST, STABILITY_KEY)
 context.get_user_info()
 print("Connection successful!")
 
-#@title Animation UI
-show_ui_in_notebook = True #@param {type:"boolean"}
-
 outputs_path = r'C:\Work\tst\SAnim'
 
+def anim(f_promt, s_promt):
+    # Configure the animation
+    args = AnimationArgs()
+    args.interpolate_prompts = True
+    args.locked_seed = True
+    args.max_frames = 48
+    args.seed = 42
+    args.strength_curve = "0:(0)"
+    args.diffusion_cadence_curve = "0:(4)"
+    args.cadence_interp = "film"
+
+    animation_prompts = {
+        0: f_promt,
+        24: s_promt,
+    }
+    negative_prompt = ""
+
+    # Create Animator object to orchestrate the rendering
+    animator = Animator(
+        api_context=context,
+        animation_prompts=animation_prompts,
+        negative_prompt=negative_prompt,
+        args=args
+    )
+
+    # Render each frame of animation
+    for idx, frame in enumerate(animator.render()):
+        frame.save(f"frame_{idx:05d}.png")
+
 with gr.Blocks() as demo:
     gr.Markdown("Stability Animation")
+    f_promt = gr.Textbox(label="Fist Promt", value="a photo of a cute cat")
+    s_promt = gr.Textbox(label="Fist Promt", value="a photo of a cute dog")
+    btn = gr.Button('Anim')
+    btn.click(fn=anim, inputs=[f_promt,s_promt], outputs=[],api_name="AnimAPI")
 
 demo.launch()