Update app.py

Updated to new version

app.py

@@ -1,69 +1,72 @@
 import os
-import pathlib
-import random
-import shlex
-import subprocess
 import gradio as gr
-from gradio import inputs, outputs
-import torch
-from huggingface_hub import snapshot_download
 from modelscope.pipelines import pipeline
 from modelscope.outputs import OutputKeys
+import requests
+import tempfile
 import boto3
-from botocore.client import Config
-import gradio as gr
-
-# Downloading and setting up the model
-model_dir = pathlib.Path('weights')
-if not model_dir.exists():
-    model_dir.mkdir()
-    snapshot_download('damo-vilab/modelscope-damo-text-to-video-synthesis',
-                      repo_type='model',
-                      local_dir=model_dir)
+from botocore.exceptions import NoCredentialsError
 
+# Hardcoded S3 keys
 s3_access_key = "juj22qxqxql7u2pl6nomgxu3ip7a"
 s3_secret_key = "j3uwidtozhboy5vczhymhzkkjsaumznnqlzck5zjs5qxgsung4ukk"
 s3_endpoint = "https://gateway.storjshare.io"
 
-s3 = boto3.client("s3", aws_access_key_id=s3_access_key, aws_secret_access_key=s3_secret_key, endpoint_url=s3_endpoint, config=Config(signature_version="s3v4"))
+# Initialize Storj with boto3
+s3 = boto3.client(
+    "s3",
+    endpoint_url=s3_endpoint,
+    aws_access_key_id=s3_access_key,
+    aws_secret_access_key=s3_secret_key,
+)
+
+p = pipeline("text-to-video-synthesis", "damo/text-to-video-synthesis")
 
-# Function to generate video and upload it to Storj
-def generate_video(prompt: str, seed: int) -> str:
-    if seed == -1:
-        seed = random.randint(0, 1000000)
-    torch.manual_seed(seed)
-    result = pipe({'text': prompt})[OutputKeys.OUTPUT_VIDEO]
+async def generate_video(text_input: str, duration: int) -> (str, str):
+    test_text = {'text': text_input, 'duration': duration}
+    output_video_path = p(test_text,)[OutputKeys.OUTPUT_VIDEO]
+
+    # Upload to Storj
+    with open(output_video_path, "rb") as f:
+        video_key = os.path.basename(output_video_path)
+        s3.upload_fileobj(f, "huggingface-demo", video_key)
     
-    # Upload video to Storj
-    bucket_name = "huggingface-demo"
-    # Add the code to upload the video to the Storj bucket
-    # and return the URL of the uploaded video
+    # Generate signed URL for uploaded video
+    try:
+        storj_video_url = s3.generate_presigned_url(
+            "get_object",
+            Params={"Bucket": "huggingface-demo", "Key": video_key},
+            ExpiresIn=3600,
+        )
+    except NoCredentialsError:
+        print("Credentials not available")
+        return None
 
-    return result
+    # Download the video from Storj to a temporary file
+    response = requests.get(storj_video_url, stream=True)
+    response.raise_for_status()
 
-# Gradio Interface
-examples = [
-    ['An astronaut riding a horse.', 0],
-    ['A panda eating bamboo on a rock.', 0],
-    ['Spiderman is surfing.', 0],
-]
+    with tempfile.NamedTemporaryFile(delete=False, suffix=".mp4") as temp_file:
+        for chunk in response.iter_content(chunk_size=8192):
+            temp_file.write(chunk)
+        local_video_path = temp_file.name
 
-# Import Storj Theme from the hub
-storj_theme = gr.Theme.from_hub("bethecloud/storj_theme")
+    return local_video_path, storj_video_url
 
-inputs = [
-    gr.inputs.Textbox(lines=1, placeholder="Enter your text here..."),
-    gr.inputs.Slider(minimum=-1, maximum=1000000, step=1, default=-1, label="Seed")
-]
+storj_theme = gr.Theme.from_hub("bethecloud/storj_theme")
 
 iface = gr.Interface(
     fn=generate_video,
-    inputs=inputs,
-    outputs=gr.outputs.Video(),
-    theme=storj_theme,
-    allow_flagging=False,
-    examples=examples
+    inputs=[
+        gr.inputs.Textbox(lines=3, label="Input text"),
+        gr.inputs.Slider(minimum=1, maximum=60, step=1, default=5, label="Duration (seconds)")
+    ],
+    outputs=[
+        gr.outputs.Video(label="Generated Video"),
+        gr.outputs.Textbox(label="Storj Video URL")
+    ],
+    title="Text to Video to Storj",
+    theme=storj_theme
 )
 
-## Run app
 iface.launch(share=True, debug=True)