Update app.py

app.py

@@ -16,10 +16,10 @@ app = Flask(__name__)
 SAVE_DIR = "/app/data/video/"
 os.makedirs(SAVE_DIR, exist_ok=True)
 
-def send_request_to_space(id,lines,url):
+def send_task(id,url,lines):
     payload = {
         "id": id,
-        "lines": lines  # or [line] depending on your expected payload
+        "lines": lines  
     }
 
     try:
@@ -122,33 +122,10 @@ def generate_video():
             return send_file(video_link, mimetype='video/mp4')
         else:
             space_b_url = ['https://sreepathi-ravikumar-subprocess1.hf.space/generate', 'https://sreepathi-ravikumar-subprocess2.hf.space/generate', 'https://sreepathi-ravikumar-subprocess3.hf.space/generate', 'https://sreepathi-ravikumar-subprocess4.hf.space/generate', 'https://sreepathi-ravikumar-subprocess5.hf.space/generate', 'https://sreepathi-ravikumar-subprocess6.hf.space/generate', 'https://sreepathi-ravikumar-subprocess7.hf.space/generate', 'https://sreepathi-ravikumar-subprocess8.hf.space/generate', 'https://sreepathi-ravikumar-subprocess9.hf.space/generate', 'https://sreepathi-ravikumar-subprocess10.hf.space/generate']
-            for id in range(len(lines)):
-                payload = {
-                    "id": id,
-                    "lines": lines
-                }
-            
-                # Replace with your actual Space B URL
-                
-            
-                response = requests.post(space_b_url[id+1], json=payload)
-            
-                if response.status_code == 200:
-                    # Save received video file
-                    with open(f"/app/data/video/clip{id}.mp4", "wb") as f:
-                        f.write(response.content)
-                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)
+            with ThreadPoolExecutor(max_workers=len(lines)) as executor:
+                for id in range(len(lines)):
+                    executor.submit(send_task,id,space_b_url[id],lines)
 
             video_path = video_com(lines)
             #for img in image_files: