1

.gitignore

@@ -110,4 +110,5 @@ venv.bak/
 *.pt
 
 # Files
-api.py
+api.py
+/frames

app.py

@@ -1,4 +1,6 @@
 import os
+import io
+import base64
 import gradio as gr
 from stability_sdk.api import Context
 from stability_sdk.animation import AnimationArgs, Animator
@@ -43,10 +45,12 @@ 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"
-        frame.save(file_path)
-        yield file_path  # Yield the path to the generated frame
-
+        # Convert PIL Image to base64
+        buffered = io.BytesIO()
+        frame.save(buffered, format="PNG")
+        img_str = base64.b64encode(buffered.getvalue()).decode()
+        yield img_str  # Yield the base64 string of the generated frame
+        
 with gr.Blocks() as demo:
     gr.Markdown("Stability Animation")
     f_promt = gr.Textbox(label="First Prompt", value="a photo of a cute cat")