Update app.py

app.py

@@ -85,28 +85,33 @@ def generate_video():
                     lines.append('\n'.join(block_lines))
         
         # Print or use lines as slides
-        id=0
-        payload = {
-            "id": id,
-            "lines": lines
-        }
-    
-        # Replace with your actual Space B URL
-        space_b_url = "https://sreepathi-ravikumar-subprocess1.hf.space/generate"
-    
-        response = requests.post(space_b_url, 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)
+        if len(lines)==1:
+            video_link = "/app/data/video/clip0.mp4"
+            video_func(0,lines)
+            return send_file(video_link, mimetype='video/mp4')
         else:
-            raise Exception(f"Error from Space B: {response.text}")
+            for id in len(lines):
+                payload = {
+                    "id": id,
+                    "lines": lines
+                }
+            
+                # Replace with your actual Space B URL
+                space_b_url = "https://sreepathi-ravikumar-subprocess1.hf.space/generate"
+            
+                response = requests.post(space_b_url, 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}")
 
-        video_path = video_com(lines)
-        #for img in image_files:
-            #os.remove(img)
-        return send_file(video_path, mimetype='video/mp4')
+            video_path = video_com(lines)
+            #for img in image_files:
+                #os.remove(img)
+            return send_file(video_path, mimetype='video/mp4')
     except Exception as e:
         traceback.print_exc()
         return jsonify({"error": str(e)}), 500