Update app.py

app.py

@@ -18,8 +18,11 @@ import cv2
 from PIL import Image
 import tempfile
 import os
+import errno
 import gc
 from openai import OpenAI
+import base64
+from io import BytesIO
 
 # Load Hugging Face token if needed
 hf_token = os.getenv("HF_TOKEN")
@@ -34,6 +37,14 @@ with open(system_prompt_t2v_path, "r") as f:
 with open(system_prompt_i2v_path, "r") as f:
     system_prompt_i2v = f.read()
 
+
+def silentremove(filename):
+    try:
+        os.remove(filename)
+    except OSError as e: # this would be "except OSError, e:" before Python 2.6
+        if e.errno != errno.ENOENT: # errno.ENOENT = no such file or directory
+            raise # re-raise exception if a different error occurred
+            
 # Set model download directory within Hugging Face Spaces
 model_path = "asset"
 if not os.path.exists(model_path):
@@ -272,7 +283,7 @@ def generate_video_from_text(
         gc.collect()
 
     output_path = tempfile.mktemp(suffix=".mp4")
-    print(images.shape)
+    #print(images.shape)
     video_np = images.squeeze(0).permute(1, 2, 3, 0).cpu().float().numpy()
     video_np = (video_np * 255).astype(np.uint8)
     height, width = video_np.shape[1:3]
@@ -286,8 +297,15 @@ def generate_video_from_text(
     del images
     del video_np
     torch.cuda.empty_cache()
-    return output_path
-
+    
+    # Read the content of the video file and encode it to base64
+    with open(output_path, "rb") as video_file:
+        video_base64 = base64.b64encode(video_file.read()).decode('utf-8')
+    
+    silentremove(output_path)
+    
+    # Prepend the appropriate data URI header with MIME type
+    return 'data:video/mp4;base64,' + video_base64
 
 def generate_video_from_image(
     secret_token="",
@@ -380,7 +398,14 @@ def generate_video_from_image(
         torch.cuda.empty_cache()
         gc.collect()
 
-    return output_path
+    # Read the content of the video file and encode it to base64
+    with open(output_path, "rb") as video_file:
+        video_base64 = base64.b64encode(video_file.read()).decode('utf-8')
+    
+    silentremove(output_path)
+    
+    # Prepend the appropriate data URI header with MIME type
+    return 'data:video/mp4;base64,' + video_base64
 
 
 def create_advanced_options():
@@ -486,7 +511,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as iface:
                     )
 
                 with gr.Column():
-                    txt2vid_output = gr.Video(label="Generated Output")
+                    txt2vid_output = gr.Text(label="Generated Output (in base64)")
 
             with gr.Row():
                 gr.Examples(
@@ -559,7 +584,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as iface:
                     )
 
                 with gr.Column():
-                    img2vid_output = gr.Video(label="Generated Output")
+                    img2vid_output = gr.Text(label="Generated Output (in base64)")
 
             with gr.Row():
                 gr.Examples(
@@ -635,6 +660,6 @@ with gr.Blocks(theme=gr.themes.Soft()) as iface:
     )
 
 if __name__ == "__main__":
-    iface.queue(max_size=32, default_concurrency_limit=1, api_open=False).launch(
-        share=True, show_api=True
+    iface.queue(max_size=32, default_concurrency_limit=1, api_open=True).launch(
+        share=False, show_api=True
     )