Update app.py

app.py

@@ -2,6 +2,7 @@ from flask import Flask, request, jsonify, send_file
 from moviepy.editor import ColorClip,ImageClip, concatenate_videoclips,VideoFileClip
 import traceback
 import uuid
+from concurrent.futures import ThreadPoolExecutor, as_completed
 import glob
 import os
 import requests
@@ -15,10 +16,32 @@ app = Flask(__name__)
 SAVE_DIR = "/app/data/video/"
 os.makedirs(SAVE_DIR, exist_ok=True)
 
+def send_request_to_space(id,lines,url):
+    payload = {
+        "id": id,
+        "lines": lines  # or [line] depending on your expected payload
+    }
+
+    try:
+        response = requests.post(url, json=payload)
+        if response.status_code == 200:
+            video_path = f"/app/data/video/clip{id}.mp4"
+            with open(video_path, "wb") as f:
+                f.write(response.content)
+            print(video_path)
+        else:
+            raise Exception(f"[{url}] HTTP {response.status_code}: {response.text}")
+    except Exception as e:
+        print(f"Error for ID {id}: {e}")
+        return None
+
+
 @app.route("/")
 def home():
     return "Flask Video Generator is Running"
 
+
+
 @app.route("/generate", methods=["POST"])
 def generate_video():
     try:
@@ -117,6 +140,16 @@ def generate_video():
                 else:
                     raise Exception(f"Error from Space B: {response.text}")
 
+            with ThreadPoolExecutor(max_workers=10) as executor:
+            for id, line in enumerate(lines):
+                url = space_b_urls[id % len(space_b_urls)]  # reuse if more lines than URLs
+                futures.append(executor.submit(send_request_to_space, id, lines, url))
+    
+            for future in as_completed(futures):
+                result = future.result()
+                if result:
+                    video_paths.append(result)
+
             video_path = video_com(lines)
             #for img in image_files:
                 #os.remove(img)