Update app.py

app.py

@@ -1,15 +1,14 @@
 from flask import Flask, request, jsonify, send_file
-from moviepy.editor import ColorClip,ImageClip, concatenate_videoclips,VideoFileClip
+from moviepy.editor import ColorClip, ImageClip, concatenate_videoclips, VideoFileClip
 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
 
-app = Flask(__name__)
+app = Flask(_name_)
 
 @app.route("/")
 def home():
@@ -21,31 +20,33 @@ def generate_video():
         data = request.get_json()
         id = data["id"]
         lines = data["lines"]
+
         image_folder = "/tmp/images"
-        #line=prompts.splitlines()
-        #asyncio.run(main(line))
+        image_olst = []
+
+        # 1. Create Image
+        create_text_image(lines[id], id, image_olst)
+
+        # 2. Generate Video (Only once!)
+        video_path = video_func(id, lines)
+
+        # 3. Check video exists
+        if not os.path.exists(video_path):
+            raise Exception("Video not found: " + video_path)
 
-        
-        image_olst=[]
-        create_text_image(lines[id],id,image_olst)
+        print("Generated video at:", video_path)
+
+        # 4. Cleanup images
         image_files = sorted(glob.glob(os.path.join(image_folder, "*.png")))
-        video_path = video_func(id,lines)
         for img in image_files:
             os.remove(img)
 
-        # Your existing processing logic
-        # create_text_image, video_func, etc.
-
-        video_path = video_func(id, lines)
-        print(video_path,"--------------------------------video_path-------------------------------")
-
-        return send_file(video_path, mimetype='video/mp4')
+        # 5. Return video
+        return send_file(video_path, mimetype='video/mp4', as_attachment=True)
 
     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)
+if _name_ == "_main_":
+    app.run(host="0.0.0.0", port=7860)