Update app.py

app.py

@@ -15,6 +15,7 @@ import random
 import tempfile
 import re
 from gradio_client import Client
+import moviepy.editor as mp
 
 
 def animate_img(encoded_string, model):
@@ -152,7 +153,16 @@ def create_video(prompt, model):
                 photo = r2.json()['data']['image_urls'][0]
                 break
             c += 1
-        return photo
+        image = base64.b64encode(requests.get(photo).content).decode("utf-8")
+        
+        with tempfile.NamedTemporaryFile(delete=False) as temp:
+            temp.write(base64.decodebytes(bytes(image, "utf-8")))
+            temp_file_path = temp.name
+            
+        clip = mp.VideoFileClip(temp_file_path)
+        temp_file2 = tempfile.NamedTemporaryFile(delete=False)
+        clip.write_videofile(temp_file2.name)
+        return temp_file2.name
 
 
 def flip_text1(prompt, motion):
@@ -309,7 +319,7 @@ with gr.Blocks(css=css) as demo:
             # prompt12 = gr.Textbox(placeholder="Введите описание видео...", show_label=True, label='Описание видео (опционально):', lines=3)
             # motion2 = gr.Dropdown(value="Приближение →←", interactive=True, show_label=True, label="Движение камеры:", choices=["Приближение →←", "Отдаление ←→", "Вверх ↑", "Вниз ↓", "Влево ←", "Вправо →", "По часовой стрелке ⟳", "Против часовой стрелки ⟲"])
             model2 = gr.Radio(interactive=True, value="Stable Video Diffusion", show_label=True, 
-                             label="Модель нейросети:", choices=['Stable Video Diffusion', 'AnimateDiff'])
+                             label="Модель нейросети:", choices=['Stable Video Diffusion'])
         with gr.Column():
             text_button2 = gr.Button("Анимировать изображение", variant='primary', elem_id="generate")
         with gr.Column():