Create app.py

app.py

@@ -0,0 +1,112 @@
+import os
+import cv2
+import numpy as np
+from PIL import Image, ImageDraw, ImageFont
+from moviepy.editor import VideoFileClip, AudioFileClip, CompositeAudioClip
+import gradio as gr
+
+def create_video(imageFolder, textToInsertLine1, textToInsertLine2, language):
+    temp_folder = 'temp_images'
+    if not os.path.exists(temp_folder):
+        os.makedirs(temp_folder)
+
+    image_paths = []
+    for image in imageFolder:
+        image_filename = os.path.basename(image.name)
+        image_path = os.path.join(temp_folder, image_filename)
+        with open(image_path, "wb") as buffer:
+            buffer.write(image.read())
+        image_paths.append(image_path)
+
+    video_path = 'output.mp4'
+    fps = 30
+    frame_size = (1080, 1920)
+    total_duration = 60  # Thời lượng video mặc định là 60 giây
+    frame_duration = total_duration / len(image_paths)  # Thời lượng mỗi ảnh
+
+    fourcc = cv2.VideoWriter_fourcc(*'mp4v')
+    video_writer = cv2.VideoWriter(video_path, fourcc, fps, frame_size)
+
+    logo_path = os.path.join("public", 'logo.png')
+    logo = cv2.imread(logo_path, cv2.IMREAD_UNCHANGED)
+    if logo is None:
+        return "Không tìm thấy file logo"
+    if logo.shape[2] == 4:  # Kiểm tra nếu logo có kênh alpha
+        logo = cv2.cvtColor(logo, cv2.COLOR_RGBA2RGB)
+    logo = cv2.resize(logo, (100, 100))
+
+    font_path = os.path.join("public", 'fonts', 'Montserrat-Bold.ttf')
+    if not os.path.exists(font_path):
+        return "Không tìm thấy file font"
+
+    try:
+        font = ImageFont.truetype(font_path, 50)
+    except OSError as e:
+        return f"Không thể mở file font: {e}"
+
+    for image_path in image_paths:
+        image = cv2.imread(image_path)
+        if image is None:
+            return f"Không thể đọc ảnh: {image_path}"
+        
+        # Crop và resize ảnh để đảm bảo kích thước 1080x1920
+        height, width = image.shape[:2]
+        if height != 1920 or width != 1080:
+            image = cv2.resize(image, frame_size)
+
+        # Chèn logo và text vào ảnh
+        image[10:110, 10:110] = logo
+        img_pil = Image.fromarray(cv2.cvtColor(image, cv2.COLOR_BGR2RGB))
+        draw = ImageDraw.Draw(img_pil)
+
+        # Vẽ dòng text thứ nhất
+        text_bbox1 = draw.textbbox((0, 0), textToInsertLine1, font=font)
+        text_size1 = (text_bbox1[2] - text_bbox1[0], text_bbox1[3] - text_bbox1[1])
+        text_position1 = ((frame_size[0] - text_size1[0]) // 2, (frame_size[1] - text_size1[1]) // 2 - 50)
+        draw.text(text_position1, textToInsertLine1, font=font, fill=(255, 255, 255), stroke_width=2, stroke_fill=(255, 0, 0))
+
+        # Vẽ dòng text thứ hai
+        text_bbox2 = draw.textbbox((0, 0), textToInsertLine2, font=font)
+        text_size2 = (text_bbox2[2] - text_bbox2[0], text_bbox2[3] - text_bbox2[1])
+        text_position2 = ((frame_size[0] - text_size2[0]) // 2, (frame_size[1] - text_size2[1]) // 2 + 50)
+        draw.text(text_position2, textToInsertLine2, font=font, fill=(255, 255, 255), stroke_width=2, stroke_fill=(255, 0, 0))
+
+        image = cv2.cvtColor(np.array(img_pil), cv2.COLOR_RGB2BGR)
+
+        for _ in range(int(frame_duration * fps)):
+            video_writer.write(image)
+
+    video_writer.release()
+
+    # Chèn âm thanh nền mặc định vào video
+    default_audio_path = os.path.join("public", 'bg.mp3')
+    if not os.path.exists(default_audio_path):
+        return "Không tìm thấy file âm thanh nền mặc định"
+
+    video_clip = VideoFileClip(video_path)
+    audio_clip = AudioFileClip(default_audio_path)
+    final_audio = CompositeAudioClip([audio_clip])
+    final_clip = video_clip.set_audio(final_audio)
+    final_clip.write_videofile('final_output.mp4', codec='libx264')
+
+    for image_path in image_paths:
+        os.remove(image_path)
+
+    return 'final_output.mp4'
+
+# Tạo giao diện Gradio
+iface = gr.Interface(
+    fn=create_video,
+    inputs=[
+        gr.File(label="Upload Folder Dataset", file_count="directory"),
+        gr.Textbox(label="Text to Insert Line 1"),
+        gr.Textbox(label="Text to Insert Line 2"),
+        gr.Dropdown(label="Language", choices=["English", "Vietnamese"])
+    ],
+    outputs=gr.Video(label="Output Video"),
+    title="Video Production Tool",
+    description="Upload a folder containing images to create a video with text overlay."
+)
+
+# Chạy ứng dụng Gradio
+iface.launch()