Update app.py

app.py

@@ -1,10 +1,9 @@
 from flask import Flask, request, jsonify, send_file
-from moviepy.editor import ColorClip,ImageClip, concatenate_videoclips
+from moviepy.editor import VideoFileClip, concatenate_videoclips
 import traceback
 import uuid
 import glob
 import os
-import asyncio
 from image_fetcher import main
 from video import create_text_image
 from video2 import video_func
@@ -20,86 +19,81 @@ def generate_video():
     try:
         data = request.get_json()
         prompt = data.get("duration", '').strip()
-        prompts=prompt.replace("**","")
+        prompts = prompt.replace("**", "")
         print(prompts)
         if prompts == '':
-            return jsonify({"error": "prompts be must"}), 400
-        image_folder = "/tmp/images"
-        #line=prompts.splitlines()
-        #asyncio.run(main(line))
-        raw_lines = prompts.splitlines(keepends=False)
+            return jsonify({"error": "prompts must not be empty"}), 400
+
+        # Ensure all temp dirs exist
+        for d in ["/tmp/images", "/tmp/sound", "/tmp/video"]:
+            os.makedirs(d, exist_ok=True)
+
+        raw_lines = prompts.splitlines()
         lines = []
-        
+
         i = 0
         while i < len(raw_lines):
             line = raw_lines[i].strip()
-        
-            # Check if current line is a heading
-            if line.strip().startswith("#") and (line.endswith('?') or line.endswith(':')):
-                block = line  # Start block with heading
+            if line.startswith("#") and (line.endswith('?') or line.endswith(':')):
+                block = line
                 i += 1
-        
-                # Accumulate body lines until next heading or 5+ lines
                 paragraph_lines = []
                 while i < len(raw_lines):
                     next_line = raw_lines[i].strip()
-        
-                    # Stop if next line is a heading
-                    if next_line.strip().startswith("#") and (next_line.endswith('?') or next_line.endswith(':')):
+                    if next_line.startswith("#") and (next_line.endswith('?') or next_line.endswith(':')):
                         break
-        
                     paragraph_lines.append(next_line)
                     i += 1
-        
-                    # If we've gathered enough lines for a slide, break to next
                     if len(paragraph_lines) >= 5:
                         break
-        
-                # Combine heading + paragraph
                 if paragraph_lines:
                     block += '\n' + '\n'.join(paragraph_lines)
-        
                 lines.append(block)
-        
             else:
-                # Group normal lines (not part of any heading)
                 block_lines = []
                 count = 0
-        
                 while i < len(raw_lines) and count < 5:
                     next_line = raw_lines[i].strip()
-        
-                    # If this is a heading, break to handle it separately
-                    if next_line.strip().startswith("#") and (next_line.endswith('?') or next_line.endswith(':')):
+                    if next_line.startswith("#") and (next_line.endswith('?') or next_line.endswith(':')):
                         break
-        
                     block_lines.append(next_line)
                     i += 1
                     count += 1
-        
                 if block_lines:
                     lines.append('\n'.join(block_lines))
-        
-        # Print or use lines as slides
-        
-        image_olst=[]
+
+        # Generate images
+        image_olst = []
         for id in range(len(lines)):
-          create_text_image(lines[id],id,image_olst)
-        image_files = sorted(glob.glob(os.path.join(image_folder, "*.png")))
+            create_text_image(lines[id], id, image_olst)
+
+        image_files = sorted(glob.glob("/tmp/images/*.png"))
         if not image_files:
             raise ValueError("No images found in folder!")
-    
-        video_location = video_func(0,lines)
-        for img in image_files:
-            os.remove(img)
 
-        return send_file(video_location, mimetype='video/mp4')
+        # Generate video clips
+        for id in range(len(lines)):
+            video_func(id, lines)
+
+        clips = []
+        for id in range(len(lines)):
+            clip_path = f"/tmp/video/clip{id}.mp4"
+            clip = VideoFileClip(clip_path)
+            clips.append(clip)
+
+        final_video = concatenate_videoclips(clips)
+        output_path = "/tmp/final_output.mp4"
+        final_video.write_videofile(output_path, fps=24)
+
+        # Clean up
+        for f in image_files:
+            os.remove(f)
+
+        return send_file(output_path, mimetype='video/mp4')
 
     except Exception as e:
         traceback.print_exc()
         return jsonify({"error": str(e)}), 500
 
 if __name__ == "__main__":
-    app.run(host="0.0.0.0", port=7860)
-
-    # Example call (remove or change in your actual app)
+    app.run(host="0.0.0.0", port=7860)