Update app.py

app.py

@@ -1,31 +1,23 @@
-import os
-import getpass
+#@title Connect to the Stability API
+
 from stability_sdk.api import Context
 from stability_sdk.animation_ui import create_ui
-import gradio as gr
-
-STABILITY_HOST = "grpc.stability.ai:443"
 
-def connect_to_stability_api(api_key):
-    # Connect to Stability API
-    context = Context(STABILITY_HOST, api_key)
+# @markdown To get your API key visit https://dreamstudio.ai/account
+STABILITY_HOST = "grpc.stability.ai:443" #@param {type:"string"}
 
-    # Test the connection
-    context.get_user_info()
-    print("Connection successful!")
+# Connect to Stability API
+context = Context(STABILITY_HOST, STABILITY_KEY)
 
-    outputs_path = '/tmp/SAnim'  # Hugging Face Spaces only allows writing to the /tmp directory
+# Test the connection
+context.get_user_info()
+print("Connection successful!")
 
-    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)
+#@title Animation UI
+show_ui_in_notebook = True #@param {type:"boolean"}
 
-def wrapper_func(dummy_input):
-    api_key = os.getenv('STABILITY_KEY')
-    if not api_key:
-        api_key = getpass.getpass('Enter your API Key')
-    connect_to_stability_api(api_key)
-    return "Done"
+outputs_path = r'C:\Work\tst\SAnim'
 
-iface = gr.Interface(fn=wrapper_func, inputs='text', outputs='text')
-iface.launch()
+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)