add create_draft api

downloader.py

@@ -7,123 +7,123 @@ from urllib.parse import urlparse, unquote
 
 def download_video(video_url, draft_name, material_name):
     """
-    下载视频到指定目录
-    :param video_url: 视频URL
-    :param draft_name: 草稿名称
-    :param material_name: 素材名称
-    :return: 本地视频路径
+    Download video to specified directory
+    :param video_url: Video URL
+    :param draft_name: Draft name
+    :param material_name: Material name
+    :return: Local video path
     """
-    # 确保目录存在
+    # Ensure directory exists
     video_dir = f"{draft_name}/assets/video"
     os.makedirs(video_dir, exist_ok=True)
     
-    # 生成本地文件名
+    # Generate local filename
     local_path = f"{video_dir}/{material_name}"
     
-    # 检查文件是否已存在
+    # Check if file already exists
     if os.path.exists(local_path):
-        print(f"视频文件已存在: {local_path}")
+        print(f"Video file already exists: {local_path}")
         return local_path
     
     try:
-        # 使用ffmpeg下载视频
+        # Use ffmpeg to download video
         command = [
             'ffmpeg',
             '-i', video_url,
-            '-c', 'copy',  # 直接复制，不重新编码
+            '-c', 'copy',  # Direct copy, no re-encoding
             local_path
         ]
         subprocess.run(command, check=True, capture_output=True)
         return local_path
     except subprocess.CalledProcessError as e:
-        raise Exception(f"下载视频失败: {e.stderr.decode('utf-8')}")
+        raise Exception(f"Failed to download video: {e.stderr.decode('utf-8')}")
 
 def download_image(image_url, draft_name, material_name):
     """
-    下载图片到指定目录，并统一转换为PNG格式
-    :param image_url: 图片URL
-    :param draft_name: 草稿名称
-    :param material_name: 素材名称
-    :return: 本地图片路径
+    Download image to specified directory, and convert to PNG format
+    :param image_url: Image URL
+    :param draft_name: Draft name
+    :param material_name: Material name
+    :return: Local image path
     """
-    # 确保目录存在
+    # Ensure directory exists
     image_dir = f"{draft_name}/assets/image"
     os.makedirs(image_dir, exist_ok=True)
     
-    # 统一使用png格式
+    # Uniformly use png format
     local_path = f"{image_dir}/{material_name}"
     
-    # 检查文件是否已存在
+    # Check if file already exists
     if os.path.exists(local_path):
-        print(f"图片文件已存在: {local_path}")
+        print(f"Image file already exists: {local_path}")
         return local_path
     
     try:
-        # 使用ffmpeg下载并转换图片为PNG格式
+        # Use ffmpeg to download and convert image to PNG format
         command = [
             'ffmpeg',
             '-i', image_url,
-            '-vf', 'format=rgba',  # 转换为RGBA格式以支持透明度
-            '-frames:v', '1',      # 确保只处理一帧
-            '-y',                  # 覆盖已存在的文件
+            '-vf', 'format=rgba',  # Convert to RGBA format to support transparency
+            '-frames:v', '1',      # Ensure only one frame is processed
+            '-y',                  # Overwrite existing files
             local_path
         ]
         subprocess.run(command, check=True, capture_output=True)
         return local_path
     except subprocess.CalledProcessError as e:
-        raise Exception(f"下载图片失败: {e.stderr.decode('utf-8')}")
+        raise Exception(f"Failed to download image: {e.stderr.decode('utf-8')}")
 
 def download_audio(audio_url, draft_name, material_name):
     """
-    下载音频并转码为MP3格式到指定目录
-    :param audio_url: 音频URL
-    :param draft_name: 草稿名称
-    :param material_name: 素材名称
-    :return: 本地音频路径
+    Download audio and transcode to MP3 format to specified directory
+    :param audio_url: Audio URL
+    :param draft_name: Draft name
+    :param material_name: Material name
+    :return: Local audio path
     """
-    # 确保目录存在
+    # Ensure directory exists
     audio_dir = f"{draft_name}/assets/audio"
     os.makedirs(audio_dir, exist_ok=True)
     
-    # 生成本地文件名（保留.mp3后缀）
+    # Generate local filename (keep .mp3 extension)
     local_path = f"{audio_dir}/{material_name}"
     
-    # 检查文件是否已存在
+    # Check if file already exists
     if os.path.exists(local_path):
-        print(f"音频文件已存在: {local_path}")
+        print(f"Audio file already exists: {local_path}")
         return local_path
     
     try:
-        # 使用ffmpeg下载并转码为MP3（关键修改：指定MP3编码器）
+        # Use ffmpeg to download and transcode to MP3 (key modification: specify MP3 encoder)
         command = [
             'ffmpeg',
-            '-i', audio_url,          # 输入URL
-            '-c:a', 'libmp3lame',     # 强制将音频流编码为MP3
-            '-q:a', '2',              # 设置音频质量（0-9，0为最佳，2为平衡质量与文件大小）
-            '-y',                     # 覆盖已存在文件（可选）
-            local_path                # 输出路径
+            '-i', audio_url,          # Input URL
+            '-c:a', 'libmp3lame',     # Force encode audio stream to MP3
+            '-q:a', '2',              # Set audio quality (0-9, 0 is best, 2 balances quality and file size)
+            '-y',                     # Overwrite existing files (optional)
+            local_path                # Output path
         ]
         subprocess.run(command, check=True, capture_output=True, text=True)
         return local_path
     except subprocess.CalledProcessError as e:
-        raise Exception(f"下载音频失败:\n{e.stderr}")
+        raise Exception(f"Failed to download audio:\n{e.stderr}")
 
 def download_file(url:str, local_filename, max_retries=3, timeout=180):
-    # 提取目录部分
+    # Extract directory part
     directory = os.path.dirname(local_filename)
 
     retries = 0
     while retries < max_retries:
         try:
             if retries > 0:
-                wait_time = 2 ** retries  # 指数退避策略
+                wait_time = 2 ** retries  # Exponential backoff strategy
                 print(f"Retrying in {wait_time} seconds... (Attempt {retries+1}/{max_retries})")
                 time.sleep(wait_time)
             
             print(f"Downloading file: {local_filename}")
             start_time = time.time()
             
-            # 创建目录（如果不存在）
+            # Create directory (if it doesn't exist)
             if directory and not os.path.exists(directory):
                 os.makedirs(directory, exist_ok=True)
                 print(f"Created directory: {directory}")
@@ -143,13 +143,13 @@ def download_file(url:str, local_filename, max_retries=3, timeout=180):
                             
                             if total_size > 0:
                                 progress = bytes_written / total_size * 100
-                                # 对于频繁更新的进度，可以考虑使用logger.debug或更细粒度的控制，避免日志文件过大
-                                # 或者只在控制台输出进度，不写入文件
+                                # For frequently updated progress, consider using logger.debug or more granular control to avoid large log files
+                                # Or only output progress to console, not write to file
                                 print(f"\r[PROGRESS] {progress:.2f}% ({bytes_written/1024:.2f}KB/{total_size/1024:.2f}KB)", end='')
-                                pass # 避免在日志文件中打印过多进度信息
+                                pass # Avoid printing too much progress information in log files
                 
                 if total_size > 0:
-                    # print() # 原始的换行符
+                    # print() # Original newline
                     pass
                 print(f"Download completed in {time.time()-start_time:.2f} seconds")
                 print(f"File saved as: {os.path.abspath(local_filename)}")