Update base/app.py

base/app.py

@@ -73,11 +73,22 @@ def infer(secret_token, prompt):
     videos = model(prompt, video_length=16, height = 320, width= 512, num_inference_steps=ddim_steps, guidance_scale=cfg).video
     if not os.path.exists(args.output_folder):
         os.mkdir(args.output_folder)
-    torchvision.io.write_video(args.output_folder + prompt[0:30].replace(' ', '_') + '-'+str(seed_inp)+'-'+str(ddim_steps)+'-'+str(cfg)+ '-.mp4', videos[0], fps=8)
 
+    video_path = args.output_folder + prompt[0:30].replace(' ', '_') + '-'+str(seed_inp)+'-'+str(ddim_steps)+'-'+str(cfg)+ '-.mp4'
+    
+    torchvision.io.write_video(outputPath, videos[0], fps=8)
+
+    # Read the content of the video file and encode it to base64
+    with open(video_path, "rb") as video_file:
+        video_base64 = base64.b64encode(video_file.read()).decode('utf-8')
 
-    return args.output_folder + prompt[0:30].replace(' ', '_') + '-'+str(seed_inp)+'-'+str(ddim_steps)+'-'+str(cfg)+ '-.mp4'
+    # Prepend the appropriate data URI header with MIME type
+    video_data_uri = 'data:video/mp4;base64,' + video_base64
+    
+    # Clean up the video file to avoid filling up storage
+    # os.remove(video_path)
 
+    return video_data_uri
 
 
 with gr.Blocks() as demo:
@@ -101,10 +112,10 @@ with gr.Blocks() as demo:
 
             with gr.Column():
                 submit_btn = gr.Button("Generate video")
-                video_out = gr.Video(label="Video result", elem_id="video-output")
+                base64_out = gr.Textbox(label="Base64 Video")
 
             inputs = [secret_token, prompt, seed_inp, ddim_steps, cfg, infer_type]
-            outputs = [video_out]
+            outputs = [base64_out]
 
     submit_btn.click(infer, inputs, outputs)