Update app.py

app.py

@@ -24,8 +24,8 @@ interpolator = interpolator.Interpolator(model, None)
 ffmpeg_path = util.get_ffmpeg_path()
 mediapy.set_ffmpeg(ffmpeg_path)
 
+SECRET_TOKEN = os.getenv('SECRET_TOKEN', 'default_secret')
 
-        
 def do_interpolation(frame1, frame2, times_to_interpolate):
     print(frame1, frame2)
     input_frames = [frame1, frame2]
@@ -84,8 +84,10 @@ def create_video(frames, fps, type):
     return type + "_result.mp4"
 
     
-def infer(video_in,interpolation,fps_output):
+def infer(secret_token, video_in,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.')
     
     # 1. break video into frames and get FPS
     break_vid = get_frames(video_in, "vid_input_frame", "origin")
@@ -129,21 +131,23 @@ def infer(video_in,interpolation,fps_output):
 title="""test space"""
 
 with gr.Blocks() as demo:
-    with gr.Column():
-        gr.HTML(title)
-        with gr.Row():
-            with gr.Column():
-                video_input = gr.Video(source="upload", type="filepath")
-                with gr.Row():
-                    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)
-                submit_btn = gr.Button("Submit")
+    gr.HTML("""
+            <div style="z-index: 100; position: fixed; top: 0px; right: 0px; left: 0px; bottom: 0px; width: 100%; height: 100%; background: white; display: flex; align-items: center; justify-content: center; color: black;">
+              <div style="text-align: center; color: black;">
+                <p style="color: black;">This space is a REST API to programmatically generate interpolate MP4s.</p>
+              </div>
+    </div>""")
+    secret_token = gr.Textbox(label="Secret token")
+    video_input = gr.Video(source="upload", type="filepath")
+   
+    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)
+    submit_btn = gr.Button("Submit")
             
-            with gr.Column():
-                video_output = gr.Video()
-                file_output = gr.File()
+    video_output = gr.Video()
+    file_output = gr.File()
 
     
-    submit_btn.click(fn=infer, inputs=[video_input,interpolation,fps_output], outputs=[video_output, file_output])
+    submit_btn.click(fn=infer, inputs=[secret_token, video_input, interpolation, fps_output], outputs=[video_output, file_output])
 
 demo.launch()