concurnecy

app.py

@@ -1,9 +1,10 @@
+from dotenv import load_dotenv
 import gradio as gr
 import os
 from pathlib import Path
 import tempfile
 
-from sympy import true
+from sympy import true 
 from extract_metadata import get_metadata
 from find_steps import find_all_steps
 from video_editing import crop_video, get_final_video
@@ -11,6 +12,16 @@ import json
 from video_editing_ffmpeg import concatenate_videos
 
 from grok_analyze import get_story_with_grok
+
+from concurrent.futures import ThreadPoolExecutor
+import random
+import string
+
+from experiments.upload_to_s3 import upload_multiple_files
+load_dotenv()
+
+# Access the variables
+bucket = os.getenv("BUCKET")
 # Placeholder function for video generation
 def generate_video(metadata, filenames,current_context):
     """
@@ -58,15 +69,28 @@ def upload_files(uploaded_files):
             metadata_file=file.name
 
     if metadata_file=="":
+        random_string = 'a'.join(random.choices(string.ascii_lowercase + string.digits, k=8))
+
+        tuple_list = [(path,random_string+"/"+os.path.splitext(os.path.basename(path))[0]) for path in uploaded_files]
+
+        print(tuple_list)
         metadata, filename = get_metadata(uploaded_files)
-    else:
 
+        # with ThreadPoolExecutor() as executor:
+        #     future = executor.submit(upload_multiple_files,tuple_list,bucket)
+
+        executor = ThreadPoolExecutor(max_workers=1)  # Adjust max_workers as needed
+        future = executor.submit(upload_multiple_files, tuple_list, bucket)
+    else:
         with open(metadata_file, 'r') as file:
             metadata = json.load(file)
         return metadata, metadata_file, "Files uploaded successfully."
 
+
+
     print("Uploaded files:", filename)
     print("Metadata:", metadata)
+
     
     return metadata, filename, "Files uploaded successfully."
 
@@ -78,6 +102,7 @@ with gr.Blocks() as demo:
         with gr.Column():
             file_input = gr.File(label="Upload Images, Videos or metadata", file_count="multiple", file_types=["image", "video",'.json'])
             upload_button = gr.Button("Upload Files (video, image, metadata)")
+            # get_metadata =  gr.Button("Generate Metadata")
             generate_button = gr.Button("Generate Story!")
             context = gr.Textbox(label="Context")
             generated_story = gr.Textbox(label="Generated Story")
@@ -106,4 +131,6 @@ with gr.Blocks() as demo:
     )
 
 # Launch the app (Pyodide-compatible launch)
-demo.launch(share=true)
+# demo.launch(share=true)
+
+demo.launch()