Update app.py

app.py

@@ -15,7 +15,28 @@ import random
 import tempfile
 
 
-def flip_text(prompt, motion):
+def animate_img(encoded_string):
+    r = requests.post("https://stable-video-diffusion.com/api/upload", files={"file": open(encoded_string, 'rb')})
+    print(r.text)
+    hash_ = r.json()['hash']
+    time.sleep(10)
+    while True:
+        r2 = requests.get(f"https://stable-video-diffusion.com/result?hash={hash_}")
+        source_string = r2.text
+        if "Generation has been in progress for" in source_string:
+            time.sleep(10)
+            continue
+        if "Generation has been in progress for" not in source_string:
+            pattern = r'https://storage.stable-video-diffusion.com/([a-f0-9]{32})\.mp4'
+            matches = re.findall(pattern, source_string)
+            sd_video = []
+            for match in matches:
+                sd_video.append(f"https://storage.stable-video-diffusion.com/{match}.mp4")
+            print(sd_video[0])
+            return sd_video[0]
+
+
+def flip_text1(prompt, motion):
     try:
         language = detect(prompt)
         if language == 'ru':
@@ -68,7 +89,9 @@ def flip_text(prompt, motion):
                 break
 
 
-def flip_text2(encoded_string, prompt, motion):
+
+
+def flip_text21(encoded_string, prompt, motion):
 
     url_video_g = os.getenv("url_video_g")
     url_video_c = os.getenv("url_video_c")
@@ -126,6 +149,8 @@ def flip_text2(encoded_string, prompt, motion):
                 break
 
                 
+
+
 css = """
 #generate {
     width: 100%;
@@ -166,7 +191,7 @@ with gr.Blocks(css=css) as demo:
             text_button2 = gr.Button("Анимировать изображение", variant='primary', elem_id="generate")
         with gr.Column():
             video_output2 = gr.Video(show_label=True, label='Результат:', type="file")
-            text_button2.click(flip_text2, inputs=[prompt2, prompt12, motion2], outputs=video_output2)
+            text_button2.click(animate_img, inputs=[prompt2], outputs=video_output2)
     
 demo.queue(concurrency_count=12)
 demo.launch()