Update app.py

app.py

@@ -1,6 +1,9 @@
 import streamlit as st
 import shutil
 import os 
+from PIL import Image
+import requests
+
 from gradio_client import Client
 prompt_prefix = "night, full body, closed mouth, happy face, walking front, "
 prompt_list = [
@@ -33,13 +36,41 @@ prompt_list = [
 prompt_suffix = ", masterpiece, best quality, very aesthetic, absurdres"
 if os.path.exists("./output/") == False:
     os.mkdir("./output/")
+if os.path.exists("./frames/") == False:
+    os.mkdir("./frames/")
+if os.path.exists("./interpolations/") == False:
+    os.mkdir("./interpolations/")
 project_name = str(st.text_input("Project name")).replace(" ", "-")
 character = st.text_input("character")
 anime_from = st.text_input("From")
-seed = st.slider('seed', 0, 2147483647)
+seed = st.number_input('seed')
 if st.button("Generate!"):
+    n = 0
     for p in prompt_list:
         prompt_new = "1girl, "+character+", "+anime_from+", "+prompt_prefix+p+prompt_suffix
+        client = Client("cagliostrolab/animagine-xl-3.1")
+        result = client.predict(
+	    	prompt_new,	# str  in 'Prompt' Textbox component
+		    "nsfw, lowres, (bad), text, error, fewer, extra, missing, worst quality, jpeg artifacts, low quality, watermark, unfinished, displeasing, oldest, early, chromatic aberration, signature, extra digits, artistic error, username, scan, [abstract]",	# str  in 'Negative Prompt' Textbox component
+	    	seed,	# float (numeric value between 0 and 2147483647) in 'Seed' Slider component
+	    	768,	# float (numeric value between 512 and 2048) in 'Width' Slider component
+	    	1344,	# float (numeric value between 512 and 2048) in 'Height' Slider component
+	    	7,	# float (numeric value between 1 and 12) in 'Guidance scale' Slider component
+	    	28,	# float (numeric value between 1 and 50) in 'Number of inference steps' Slider component
+	    	"Euler a",	# Literal['DPM++ 2M Karras', 'DPM++ SDE Karras', 'DPM++ 2M SDE Karras', 'Euler', 'Euler a', 'DDIM']  in 'Sampler' Dropdown component
+	    	"768 x 1344",	# Literal['1024 x 1024', '1152 x 896', '896 x 1152', '1216 x 832', '832 x 1216', '1344 x 768', '768 x 1344', '1536 x 640', '640 x 1536', 'Custom']  in 'Aspect Ratio' Radio component
+	    	"(None)",	# Literal['(None)', 'Cinematic', 'Photographic', 'Anime', 'Manga', 'Digital Art', 'Pixel art', 'Fantasy art', 'Neonpunk', '3D Model']  in 'Style Preset' Radio component
+	    	"Standard v3.1",	# Literal['(None)', 'Standard v3.0', 'Standard v3.1', 'Light v3.1', 'Heavy v3.1']  in 'Quality Tags Presets' Dropdown component
+	    	False,	# bool  in 'Use Upscaler' Checkbox component
+	    	0,	# float (numeric value between 0 and 1) in 'Strength' Slider component
+	    	1,	# float (numeric value between 1 and 1.5) in 'Upscale by' Slider component
+	    	True,	# bool  in 'Add Quality Tags' Checkbox component
+	    	api_name="/run"
+        )
+        source = 'https://cagliostrolab-animagine-xl-3-1.hf.space/file='+result[0][0]["image"]
+        frame = Image.open(requests.get(source, stream=True).raw)
+        frame.save("./output/"+str(n)+".png")
+        n += 1
         print(prompt_new)
     
 output_num = len(os.listdir("./output/"))