Update app.py

app.py

@@ -82,18 +82,33 @@ with gr.Blocks() as demo:
 
     prompt_input = gr.Textbox(label="Описание (русский язык)",
                               placeholder="Пример: Киберпанк-город ночью, синтезаторная музыка")
-    btn_generate = gr.Button("✨ Сгенерировать все")
+    btn_image = gr.Button("🖼️ Сгенерировать изображение")
+    btn_video = gr.Button("🎥 Сгенерировать видео")
+    btn_music = gr.Button("🎵 Сгенерировать музыку")
+    btn_merge = gr.Button("✨ Объединить видео и музыку")
 
     image_output = gr.Image(label="Изображение")
     video_output = gr.Video(label="Видео")
     audio_output = gr.Audio(label="Музыка")
     final_video_output = gr.Video(label="Итоговое видео с музыкой")
 
-    btn_generate.click(
-        fn=create_all,
-        inputs=prompt_input,
-        outputs=[image_output, video_output, audio_output, final_video_output]
-    )
+    btn_image.click(fn=generate_image, inputs=prompt_input, outputs=image_output)
+    btn_video.click(fn=generate_video, inputs=image_output, outputs=video_output)
+    btn_music.click(fn=generate_music, inputs=prompt_input, outputs=audio_output)
+    btn_merge.click(fn=merge_video_audio, inputs=[video_output, audio_output], outputs=final_video_output)
+    
+    #btn_generate = gr.Button("✨ Сгенерировать все")
+
+    #image_output = gr.Image(label="Изображение")
+    #video_output = gr.Video(label="Видео")
+    #audio_output = gr.Audio(label="Музыка")
+    #final_video_output = gr.Video(label="Итоговое видео с музыкой")
+
+    #btn_generate.click(
+    #    fn=create_all,
+    #    inputs=prompt_input,
+    #    outputs=[image_output, video_output, audio_output, final_video_output]
+    #)
 
 if __name__ == "__main__":
     demo.launch()