Update app.py

app.py

@@ -30,10 +30,17 @@ async def text_to_speech(text, voice, rate, pitch):
     return tmp_path, None
 
 # 文字转视频功能
-def text_to_video(text, voice, rate, pitch, video_width, video_height, bg_color, text_color, text_font, text_size):
+def text_to_video(text, voice, rate, pitch, video_width, video_height, bg_image, text_color, text_font, text_size):
     # 字体文件路径
     font_path = os.path.abspath(text_font)
 
+    # 读取背景图像
+    if bg_image is not None:
+        bg_image_path = bg_image.name
+        bg_clip = ImageClip(bg_image_path).set_duration(0)  # Create a static clip
+    else:
+        bg_clip = ColorClip((video_width, video_height), color=(255, 255, 255))  # Default white background
+
     # 计算每页可以容纳的行数和每行可以容纳的字符数
     max_chars_per_line = video_width // (text_size // 2)  # 字体宽度假设为字体大小的一半
     max_lines_per_page = video_height // (text_size + 15)  # 10是行间距
@@ -70,7 +77,6 @@ def text_to_video(text, voice, rate, pitch, video_width, video_height, bg_color,
         audio_clip = AudioFileClip(audio)
         audio_clips.append(audio_clip)
 
-        # 使用 wand 生成视频片段
         # 使用 wand 生成视频片段
         with Drawing() as draw:
             draw.font = font_path
@@ -79,17 +85,24 @@ def text_to_video(text, voice, rate, pitch, video_width, video_height, bg_color,
             draw.text_alignment = 'center'
             draw.text_interline_spacing = 10
             
-            with Image(width=video_width, height=video_height, background=Color(bg_color)) as img:
+            with Image(width=video_width, height=video_height, background=Color('transparent')) as img:
                 lines = page.split("\n")
+                # Centering text vertically
+                total_text_height = len(lines) * (text_size + 10)  # Height of text area
+                start_y = (video_height - total_text_height) // 2  # Start position to center vertically
+                
                 for j, line in enumerate(lines):
-                    draw.text(int(video_width / 2), (j + 1) * (text_size + 10), line)
+                    draw.text(int(video_width / 2), start_y + (j * (text_size + 10)), line)
                 
                 draw(img)  # Apply the drawing to the image
                 img.format = 'png'
                 img_path = os.path.join(tempfile.gettempdir(), f"page_{i}.png")
                 img.save(filename=img_path)
+
                 text_clip = ImageClip(img_path).set_duration(audio_clip.duration).set_audio(audio_clip)
-                video_clips.append(text_clip)
+                text_clip = text_clip.set_position('center').set_duration(audio_clip.duration)
+                final_clip = CompositeVideoClip([bg_clip, text_clip])  # Overlay text on background
+                video_clips.append(final_clip)
 
     # 合并所有视频片段
     final_video = concatenate_videoclips(video_clips)
@@ -98,8 +111,8 @@ def text_to_video(text, voice, rate, pitch, video_width, video_height, bg_color,
     return final_video_path, None
 
 # Gradio接口函数
-def tts_interface(text, voice, rate, pitch, video_width, video_height, bg_color, text_color, text_font, text_size):
-    video, warning = text_to_video(text, voice, rate, pitch, video_width, video_height, bg_color, text_color, text_font, text_size)
+def tts_interface(text, voice, rate, pitch, video_width, video_height, bg_image, text_color, text_font, text_size):
+    video, warning = text_to_video(text, voice, rate, pitch, video_width, video_height, bg_image, text_color, text_font, text_size)
     return None, video, warning
 
 # 创建Gradio应用
@@ -115,7 +128,7 @@ async def create_demo():
             gr.Slider(minimum=-20, maximum=20, value=0, label="Pitch Adjustment (Hz)", step=1),
             gr.Slider(minimum=640, maximum=1920, value=1080, label="Video Width", step=10),
             gr.Slider(minimum=480, maximum=1080, value=720, label="Video Height", step=10),
-            gr.ColorPicker(value="#000000", label="Background Color"),
+            gr.File(label="Upload Background Image or Video", type="file", optional=True),
             gr.ColorPicker(value="#FFFFFF", label="Text Color"),
             gr.Textbox(label="Text Font", value="msyh.ttf"),  # 请确保字体文件路径正确
             gr.Slider(minimum=10, maximum=100, value=24, label="Text Size", step=1)