|
|
|
from flask import Flask, request, jsonify, send_file |
|
from moviepy.editor import ColorClip,ImageClip, 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 |
|
|
|
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!") |
|
|
|
|
|
for id in range(len(lines)): |
|
video_func(id,lines) |
|
clips = [] |
|
for id in range(len(lines)): |
|
clip = VideoFileClip(f"clip{id}.mp4") |
|
clips.append(clip) |
|
|
|
final_video = concatenate_videoclips(clips) |
|
video_path = f"/tmp/video_{uuid.uuid4().hex}.mp4" |
|
final_video.write_videofile(video_path, fps=24) |
|
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) |
|
|
|
|