Update app.py

app.py

@@ -1,24 +1,30 @@
-#@title Connect to the Stability API
+import os
 import getpass
 from stability_sdk.api import Context
 from stability_sdk.animation_ui import create_ui
+import gradio as gr
 
-# @markdown To get your API key visit https://dreamstudio.ai/account
-STABILITY_HOST = "grpc.stability.ai:443" #@param {type:"string"}
-STABILITY_KEY = ""#getpass.getpass('Enter your API Key')
+STABILITY_HOST = "grpc.stability.ai:443"
 
-# Connect to Stability API
-context = Context(STABILITY_HOST, STABILITY_KEY)
+def connect_to_stability_api(api_key):
+    # Connect to Stability API
+    context = Context(STABILITY_HOST, api_key)
 
-# Test the connection
-context.get_user_info()
-print("Connection successful!")
+    # Test the connection
+    context.get_user_info()
+    print("Connection successful!")
 
-#@title Animation UI
-show_ui_in_notebook = True #@param {type:"boolean"}
+    outputs_path = '/tmp/SAnim'  # Hugging Face Spaces only allows writing to the /tmp directory
 
-outputs_path = r'C:\Work\tst\SAnim'
+    ui = create_ui(context, outputs_path)
+    ui.queue(concurrency_count=2, max_size=2)
+    ui.launch(show_api=False, debug=True, inline=True, height=768, share=True, show_error=True)
 
-ui = create_ui(context, outputs_path)
-ui.queue(concurrency_count=2, max_size=2)
-ui.launch(show_api=False, debug=True, inline=show_ui_in_notebook, height=768, share=True, show_error=True)
+def wrapper_func():
+    api_key = os.getenv('STABILITY_KEY')
+    if not api_key:
+        api_key = getpass.getpass('Enter your API Key')
+    connect_to_stability_api(api_key)
+
+iface = gr.Interface(fn=wrapper_func, inputs=[], outputs=[])
+iface.launch()