Update app.py

app.py

@@ -8,9 +8,9 @@ import numpy as np
 import tensorflow as tf
 import mediapy
 from PIL import Image
-
+import base64
 import gradio as gr
-
+import tempfile
 from huggingface_hub import snapshot_download
 
 from image_tools.sizes import resize_and_crop
@@ -26,6 +26,12 @@ mediapy.set_ffmpeg(ffmpeg_path)
 
 SECRET_TOKEN = os.getenv('SECRET_TOKEN', 'default_secret')
 
+
+def base64_to_video(base64_string, output_file):
+    video_data = base64.b64decode(base64_string)
+    with open(output_file, 'wb') as f:
+        f.write(video_data)
+        
 def do_interpolation(frame1, frame2, times_to_interpolate):
     print(frame1, frame2)
     input_frames = [frame1, frame2]
@@ -84,10 +90,15 @@ def create_video(frames, fps, type):
     return type + "_result.mp4"
 
     
-def infer(secret_token, video_in,interpolation,fps_output):
+def infer(secret_token, video_in_base64, interpolation, fps_output):
     
     if secret_token != SECRET_TOKEN:
         raise gr.Error(f'Invalid secret token. Please fork the original space if you want to use it for yourself.')
+
+        
+    # Decode the base64 string to a video file
+    video_in = "video_in.mp4"  # or choose any other filename/path
+    base64_to_video(video_in_base64, video_in)
     
     # 1. break video into frames and get FPS
     break_vid = get_frames(video_in, "vid_input_frame", "origin")
@@ -103,9 +114,7 @@ def infer(secret_token, video_in,interpolation,fps_output):
     
     # 2. prepare frames result arrays
     result_frames = []
-    print("set stop frames to: " + str(n_frame))
-    
-    
+    # print("set stop frames to: " + str(n_frame))
     
     
     for idx, frame in enumerate(frames_list[0:int(n_frame)]):
@@ -138,7 +147,7 @@ with gr.Blocks() as demo:
               </div>
     </div>""")
     secret_token = gr.Textbox(label="Secret token")
-    video_input = gr.Video(source="upload", type="filepath")
+    video_input = gr.Video(source="upload", type="base64")
    
     interpolation = gr.Slider(minimum=1,maximum=4,step=1, value=3, label="Interpolation Steps")
     fps_output = gr.Radio([8, 12, 24], label="FPS output", value=24)