Update app.py

app.py

@@ -23,12 +23,19 @@ subprocess.run(["mv", "./network-disparity.pytorch", "./3d-ken-burns/models/disp
 subprocess.run(["mv", "./network-refinement.pytorch", "./3d-ken-burns/models/disparity-refinement.pytorch"])
 subprocess.run(["mv", "./network-inpainting.pytorch", "./3d-ken-burns/models/pointcloud-inpainting.pytorch"])
 
-# with open ('./3d-ken-burns/autozoom.py','r+') as az:
-#     script = az.read()
-#    script = script.replace("'fltShift': 100.0","'fltShift': 0.0")
-#    script = script.replace("'fltZoom': 1.25","'fltZoom': 1.35")
-#    script = script.replace("'fltSteps': numpy.linspace(0.0, 1.0, 75).tolist(),","'fltSteps': numpy.linspace(0.0, 1.0, 250).tolist(),")
-#    az.write(script)
+subprocess.run(["cp", "./3d-ken-burns/autozoom.py", "./3d-ken-burns/autozoom-original.py"])
+
+def update_parameters(fltShift=100, fltZoom=1.25, steps1=0.0, steps2=1.0, frames=240, fps=30):
+  with open ('./3d-ken-burns/autozoom-original.py','r') as azo:
+     script = azo.read()
+     script = script.replace("'fltShift': 100.0",f"'fltShift': {fltShift}")
+     script = script.replace("'fltZoom': 1.25",f"'fltZoom': {fltZoom}")
+     script = script.replace("'fltSteps': numpy.linspace(0.0, 1.0, 75).tolist(),",f"'fltSteps': numpy.linspace({steps1}, {steps2}, {frames}).tolist(),")
+     script = script.replace(", fps=25)",f", fps={fps})")
+     with open ('./3d-ken-burns/autozoom.py','w') as az:
+         az.write(script)
+         az.truncate()
+         az.seek(0)
 
 @spaces.GPU
 def generate_video(image):
@@ -42,7 +49,8 @@ def generate_video(image):
     subprocess.run(["ls","-al","/tmp"])
     return "/tmp/autozoom.mp4"
 
-def get_video(image):
+def get_video(image,fltShift=100, fltZoom=1.25, steps1=0.0, steps2=1.0, frames=240, fps=30):
+    update_parameters(fltShift, fltZoom, steps1, steps2, frames, fps)
     video_path = generate_video(image)
     clip = VideoFileClip("/tmp/autozoom.mp4")
     duration = clip.duration / 2
@@ -53,6 +61,11 @@ def get_video(image):
 with gr.Blocks() as demo:
     gr.Markdown("""# 3D Zoom Video Generator!""")
     image = gr.Image()
+    fltShift = gr.Slider(label="Flat Shift",minimum=0,maximum=1000,step=1,value=100)
+    fltZoom = gr.Slider(label="Flat Zoom",minimum=0.0,maximum=5.0,value=1.3,step=0.1)
+    steps1 = gr.Slider(label="Steps1",minimum=0.0,maximum=5.0,value=0.0,step=0.1)
+    steps2 = gr.Slider(label="Steps2",minimum=0.0,maximum=5.0,value=1.0,step=0.1)
+    frames = gr.Slider(label="Frames",minimum=50,maximum=10000,step=1,value=240)
     submit_image = gr.Button(value="Generate Video")
     video = gr.Video()
     submit_image.click(fn=get_video,inputs=image,outputs=video)