| from flask import Flask, request, jsonify, send_file |
| 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_com |
|
|
| app = Flask(__name__) |
|
|
| @app.route("/") |
| def home(): |
| return "Flask Video Generator is Running" |
|
|
| @app.route("/generate", methods=["POST"]) |
| def generate_video(): |
| try: |
| data = request.get_json() |
| prompt = data.get("duration", '').strip() |
| prompts=prompt.replace("**","") |
| print(prompts) |
| if prompts == '': |
| return jsonify({"error": "prompts be must"}), 400 |
| image_folder = "/tmp/images" |
| |
| |
| raw_lines = prompts.splitlines(keepends=False) |
| lines = [] |
| |
| i = 0 |
| while i < len(raw_lines): |
| line = raw_lines[i].strip() |
| |
| |
| if line.strip().startswith("#") and (line.endswith('?') or line.endswith(':')): |
| block = line |
| i += 1 |
| |
| |
| paragraph_lines = [] |
| while i < len(raw_lines): |
| next_line = raw_lines[i].strip() |
| |
| |
| if next_line.strip().startswith("#") and (next_line.endswith('?') or next_line.endswith(':')): |
| break |
| |
| paragraph_lines.append(next_line) |
| i += 1 |
| |
| |
| if len(paragraph_lines) >= 5: |
| break |
| |
| |
| if paragraph_lines: |
| block += '\n' + '\n'.join(paragraph_lines) |
| |
| lines.append(block) |
| |
| else: |
| |
| block_lines = [] |
| count = 0 |
| |
| while i < len(raw_lines) and count < 5: |
| next_line = raw_lines[i].strip() |
| |
| |
| if next_line.strip().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)) |
| |
| |
| |
| 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"))) |
| if not image_files: |
| raise ValueError("No images found in folder!") |
| |
| 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 |
|
|
| if __name__ == "__main__": |
| app.run(host="0.0.0.0", port=7860) |
|
|
| |