Update app.py

app.py

@@ -49,7 +49,7 @@ def add_new_image(image):
     image = Image.fromarray(image)
     height = image.height
     width = image.width
-    lr = height/width
+    lr = 1.0
     ip_img = "${.condition_images}"
     image.save("./img.png")
     img_settings = f"""- condition_images: ./img.png
@@ -60,7 +60,7 @@ def add_new_image(image):
   name: image
   prompt: (masterpiece, best quality, highres:1),(1person, solo:1),(eye blinks:1.8),(head wave:1.3)
   refer_image: {ip_img}
-  video_path: ./final.mp4
+  video_path: null
   width: {width}"""
     print(img_settings)
     with open ("/home/user/app/MuseV/configs/tasks/example.yaml","r+") as configs:
@@ -68,23 +68,23 @@ def add_new_image(image):
         configs.truncate()
         configs.seek(0)
 
-def add_new_video(image):
+def add_new_video(video):
     image = Image.fromarray(image)
     height = image.height
     width = image.width
-    lr = height/width
+    lr = 1.0
     ip_img = "${.condition_images}"
     image.save("./img.png")
-    img_settings = f"""- condition_images: ./img.png
-  eye_blinks_factor: 1.8
-  height: {height}
-  img_length_ratio: {lr}
-  ipadapter_image: {ip_img}
-  name: image
-  prompt: (masterpiece, best quality, highres:1),(1person, solo:1),(eye blinks:1.8),(head wave:1.3)
-  refer_image: {ip_img}
-  video_path: ./final.mp4
-  width: {width}"""
+    img_settings = f"""- name: "dance2"
+  prompt: "(best quality), ((masterpiece)), (highres), illustration, original, extremely detailed wallpaper"
+  video_path:  ./MuseV/data/source_video/video1_girl_poseseq.mp4 
+  condition_images: ./MuseV/data/images/cyber_girl.png
+  refer_image: ${.condition_images}
+  ipadapter_image: ${.condition_images}
+  height: 960
+  width: 512
+  img_length_ratio: 1.0
+  video_is_middle: True """
     print(img_settings)
     with open ("/home/user/app/MuseV/configs/tasks/example.yaml","r+") as configs:
         configs.write(img_settings)