Major refactoring

Author: zack-zzq

README.md

@@ -1,82 +1,88 @@
-# Videogen - AI视频生成工具
+# Videogen - AI Video Generation Tool
 
 ![Videogen](assets/creation-light.svg)
 
-Videogen是一个基于多种AI模型的视频生成工具，支持火山引擎、阿里云百炼和智谱AI三大平台，可以通过文本提示词或图片+文本组合生成高质量视频。
+*English | [中文](README_zh.md)*
 
-## 功能特点
+Videogen is a video generation tool based on multiple AI models, supporting three major platforms: Volcengine, Aliyun Bailian, and Zhipu AI. It can generate high-quality videos through text prompts or a combination of images and text.
 
-- **多平台支持**：
-  - 火山引擎视频生成
-  - 阿里云百炼通义万相系列模型
-  - 智谱AI的CogVideoX模型系列
-- **多种生成模式**：
-  - 文生视频：仅通过文本描述生成视频
-  - 图生视频：通过图片和文本联合生成视频
-- **高度定制化**：
-  - 支持设置宽高比、分辨率、时长、帧率等参数
-  - 支持AI音效生成（智谱AI）
-- **便捷文件管理**：
-  - 集成Tebi.io云存储服务，自动上传和管理图片素材
+## Features
 
-## 环境变量配置
+- **Multi-platform Support**:
+  - Volcengine video generation
+  - Aliyun Bailian Tongyi Wanxiang series models
+  - Zhipu AI's CogVideoX model series
+- **Multiple Generation Modes**:
+  - Text-to-video: Generate videos through text descriptions only
+  - Image-to-video: Generate videos through a combination of images and text
+- **Highly Customizable**:
+  - Set aspect ratio, resolution, duration, frame rate, and other parameters
+  - Support for AI sound effect generation (Zhipu AI)
+- **Convenient File Management**:
+  - Integrated with Tebi.io cloud storage service for automatic upload and management of image materials
+- **Multilingual Support**:
+  - English and Chinese interface
+  - Auto-detects system language
+  - Easily switch languages within the interface
 
-使用本项目需要配置以下环境变量：
+## Environment Variables
+
+The following environment variables need to be configured to use this project:
 
 ```bash
-# 火山引擎配置
-ARK_API_KEY=your_ark_api_key    # 火山引擎的API KEY
-ARK_ENDPOINT=your_ark_endpoint  # 火山引擎的推理接入点
+# Volcengine Configuration
+ARK_API_KEY=your_ark_api_key    # Volcengine API KEY
+ARK_ENDPOINT=your_ark_endpoint  # Volcengine inference endpoint
 
-# 阿里云百炼配置
-DASHSCOPE_API_KEY=your_dashscope_api_key  # 阿里云百炼的API KEY
+# Aliyun Bailian Configuration
+DASHSCOPE_API_KEY=your_dashscope_api_key  # Aliyun Bailian API KEY
 
-# 智谱AI配置
-ZHIPUAI_API_KEY=your_zhipuai_api_key  # 智谱AI的API KEY
+# Zhipu AI Configuration
+ZHIPUAI_API_KEY=your_zhipuai_api_key  # Zhipu AI API KEY
 
-# Tebi.io云存储配置
-TEBI_ACCESS_KEY=your_tebi_access_key  # Tebi.io的ACCESS KEY
-TEBI_SECRET_KEY=your_tebi_secret_key  # Tebi.io的SECRET KEY
+# Tebi.io Cloud Storage Configuration
+TEBI_ACCESS_KEY=your_tebi_access_key  # Tebi.io ACCESS KEY
+TEBI_SECRET_KEY=your_tebi_secret_key  # Tebi.io SECRET KEY
 ```
 
-## 安装说明
+## Installation
 
-1. 克隆仓库：
+1. Clone the repository:
 
 ```bash
 git clone https://github.com/yourusername/videogen.git
 cd videogen
 ```
 
-2. 安装依赖：
+2. Install dependencies:
 
 ```bash
 pip install -r requirements.txt
 ```
 
-3. 设置环境变量（可使用.env文件或系统环境变量）
+3. Set environment variables (using .env file or system environment variables)
 
-4. 启动应用：
+4. Start the application:
 
 ```bash
 python main.py
 ```
 
-## 使用方法
+## Usage
 
-1. 在浏览器中访问 `http://localhost:7860`（默认端口）
-2. 选择目标AI平台（火山引擎/阿里云百炼/智谱AI）
-3. 输入详细的提示词描述您想要生成的视频内容
-4. 如果需要图生视频，请上传参考图片
-5. 根据您选择的平台，设置合适的参数（宽高比、分辨率、时长等）
-6. 点击"生成视频"按钮，等待生成完成
-7. 生成完成后，可以直接在界面中预览或下载视频
+1. Access `http://localhost:7860` in your browser (default port)
+2. Select the target AI platform (Volcengine/Aliyun Bailian/Zhipu AI)
+3. Enter detailed prompt describing the video content you want to generate
+4. If you need image-to-video generation, upload a reference image
+5. Set appropriate parameters based on your chosen platform (aspect ratio, resolution, duration, etc.)
+6. Click the "Generate Video" button and wait for generation to complete
+7. Once generation is complete, you can preview or download the video directly from the interface
 
-## Docker部署
+## Docker Deployment
 
-### 使用预构建镜像（推荐）
+### Using Pre-built Image (Recommended)
 
-我们提供了预构建的Docker镜像，支持arm64和amd64平台，可以直接使用：
+We provide pre-built Docker images supporting arm64 and amd64 platforms:
 
 ```bash
 docker run -p 7860:7860 \
@@ -89,9 +95,9 @@ docker run -p 7860:7860 \
   arnocher/videogen
 ```
 
-### 自行构建镜像
+### Build Your Own Image
 
-也可以自行构建镜像：
+You can also build the image yourself:
 
 ```bash
 docker build -t videogen .
@@ -105,17 +111,26 @@ docker run -p 7860:7860 \
   videogen
 ```
 
-## 注意事项
+## Notes
+
+- Make sure you have all required API keys and access permissions
+- Uploaded images must comply with the requirements of each platform (recommended resolution ≥300px, aspect ratio between 0.4-2.5)
+- Generated video content should comply with relevant laws, regulations, and platform policies
+- If the Tebi.io configuration is missing, image-to-video features will be disabled
+- The application requires translation files in the i18n directory to run
+
+## Internationalization
 
-- 请确保您拥有所有必需的API密钥和访问权限
-- 上传的图片需符合各平台的要求（建议分辨率≥300px，宽高比在0.4-2.5之间）
-- 生成的视频内容应遵守相关法律法规和平台政策
+The application supports multiple languages through the i18n system:
+- Translation files are stored in the `i18n` directory as JSON files
+- Currently supported languages: English (`en.json`) and Chinese (`zh.json`)
+- To add a new language, create a new JSON file in the i18n directory following the same format
 
-## 鸣谢
+## Acknowledgements
 
-本项目使用了以下开源库和API服务：
+This project uses the following open-source libraries and API services:
 - Gradio
 - Volcengine SDK
 - DashScope SDK
 - ZhipuAI SDK
-- Boto3 (用于S3兼容存储) 
+- Boto3 (for S3-compatible storage) 

README_zh.md

@@ -0,0 +1,136 @@
+# Videogen - AI视频生成工具
+
+![Videogen](assets/creation-light.svg)
+
+*[English](README.md) | 中文*
+
+Videogen是一个基于多种AI模型的视频生成工具，支持火山引擎、阿里云百炼和智谱AI三大平台，可以通过文本提示词或图片+文本组合生成高质量视频。
+
+## 功能特点
+
+- **多平台支持**：
+  - 火山引擎视频生成
+  - 阿里云百炼通义万相系列模型
+  - 智谱AI的CogVideoX模型系列
+- **多种生成模式**：
+  - 文生视频：仅通过文本描述生成视频
+  - 图生视频：通过图片和文本联合生成视频
+- **高度定制化**：
+  - 支持设置宽高比、分辨率、时长、帧率等参数
+  - 支持AI音效生成（智谱AI）
+- **便捷文件管理**：
+  - 集成Tebi.io云存储服务，自动上传和管理图片素材
+- **多语言支持**：
+  - 支持中英文双语界面
+  - 自动检测系统语言并设置默认语言
+  - 界面内轻松切换语言
+
+## 环境变量配置
+
+使用本项目需要配置以下环境变量：
+
+```bash
+# 火山引擎配置
+ARK_API_KEY=your_ark_api_key    # 火山引擎的API KEY
+ARK_ENDPOINT=your_ark_endpoint  # 火山引擎的推理接入点
+
+# 阿里云百炼配置
+DASHSCOPE_API_KEY=your_dashscope_api_key  # 阿里云百炼的API KEY
+
+# 智谱AI配置
+ZHIPUAI_API_KEY=your_zhipuai_api_key  # 智谱AI的API KEY
+
+# Tebi.io云存储配置
+TEBI_ACCESS_KEY=your_tebi_access_key  # Tebi.io的ACCESS KEY
+TEBI_SECRET_KEY=your_tebi_secret_key  # Tebi.io的SECRET KEY
+```
+
+## 安装说明
+
+1. 克隆仓库：
+
+```bash
+git clone https://github.com/yourusername/videogen.git
+cd videogen
+```
+
+2. 安装依赖：
+
+```bash
+pip install -r requirements.txt
+```
+
+3. 设置环境变量（可使用.env文件或系统环境变量）
+
+4. 启动应用：
+
+```bash
+python main.py
+```
+
+## 使用方法
+
+1. 在浏览器中访问 `http://localhost:7860`（默认端口）
+2. 选择目标AI平台（火山引擎/阿里云百炼/智谱AI）
+3. 输入详细的提示词描述您想要生成的视频内容
+4. 如果需要图生视频，请上传参考图片
+5. 根据您选择的平台，设置合适的参数（宽高比、分辨率、时长等）
+6. 点击"生成视频"按钮，等待生成完成
+7. 生成完成后，可以直接在界面中预览或下载视频
+
+## Docker部署
+
+### 使用预构建镜像（推荐）
+
+我们提供了预构建的Docker镜像，支持arm64和amd64平台，可以直接使用：
+
+```bash
+docker run -p 7860:7860 \
+  -e ARK_API_KEY=your_ark_api_key \
+  -e ARK_ENDPOINT=your_ark_endpoint \
+  -e DASHSCOPE_API_KEY=your_dashscope_api_key \
+  -e ZHIPUAI_API_KEY=your_zhipuai_api_key \
+  -e TEBI_ACCESS_KEY=your_tebi_access_key \
+  -e TEBI_SECRET_KEY=your_tebi_secret_key \
+  arnocher/videogen
+```
+
+### 自行构建镜像
+
+也可以自行构建镜像：
+
+```bash
+docker build -t videogen .
+docker run -p 7860:7860 \
+  -e ARK_API_KEY=your_ark_api_key \
+  -e ARK_ENDPOINT=your_ark_endpoint \
+  -e DASHSCOPE_API_KEY=your_dashscope_api_key \
+  -e ZHIPUAI_API_KEY=your_zhipuai_api_key \
+  -e TEBI_ACCESS_KEY=your_tebi_access_key \
+  -e TEBI_SECRET_KEY=your_tebi_secret_key \
+  videogen
+```
+
+## 注意事项
+
+- 请确保您拥有所有必需的API密钥和访问权限
+- 上传的图片需符合各平台的要求（建议分辨率≥300px，宽高比在0.4-2.5之间）
+- 生成的视频内容应遵守相关法律法规和平台政策
+- 如果缺少Tebi.io配置，图生视频功能将被禁用
+- 应用程序需要i18n目录中的翻译文件才能运行
+
+## 国际化支持
+
+应用程序通过i18n系统支持多种语言：
+- 翻译文件以JSON格式存储在`i18n`目录中
+- 当前支持的语言：中文（`zh.json`）和英文（`en.json`）
+- 要添加新语言，只需在i18n目录中按照相同的格式创建新的JSON文件
+
+## 鸣谢
+
+本项目使用了以下开源库和API服务：
+- Gradio
+- Volcengine SDK
+- DashScope SDK
+- ZhipuAI SDK
+- Boto3 (用于S3兼容存储) 

api/ark.py

@@ -1,42 +1,58 @@
 import os
 import re
 import time
+import logging
 
 from volcenginesdkarkruntime import Ark
 
-# 初始化火山引擎客户端
-volc_client = Ark()
+# Set up logging
+logging.basicConfig(level=logging.WARNING)
+logger = logging.getLogger(__name__)
+
+# Initialize Volcengine client and check environment variables
+volc_client = None
+try:
+    api_key = os.getenv('ARK_API_KEY')
+    endpoint = os.getenv('ARK_ENDPOINT')
+    if api_key and endpoint:
+        volc_client = Ark()
+except Exception as e:
+    logger.error(f"Failed to initialize Volcengine client: {str(e)}")
 
 
 def generate_volcengine(prompt, image_url=None, ratio="16:9", duration=5):
+    # If client initialization failed, return error
+    if not volc_client:
+        return None, "Volcengine API not configured or initialization failed"
+        
     content = [{"type": "text", "text": f"{prompt} --ratio {ratio} --dur {str(duration)}"}]
     if image_url:
         content.append({"type": "image_url", "image_url": {"url": image_url}})
 
     try:
-        # 创建任务，传入额外参数
+        # Create task with additional parameters
         create_result = volc_client.content_generation.tasks.create(
             model=os.getenv('ARK_ENDPOINT'),
             content=content
         )
-        print(f"任务创建成功: {create_result.id}")
+        logger.info(f"Task created successfully: {create_result.id}")
 
-        # 轮询任务状态
+        # Poll task status
         while True:
             get_result = volc_client.content_generation.tasks.get(task_id=create_result.id)
-            print(f"任务状态: {get_result.status}")
+            logger.info(f"Task status: {get_result.status}")
             if get_result.status in ["succeeded", "failed"]:
                 break
             time.sleep(2)
 
-        # 提取 video_url
+        # Extract video_url
         if get_result.status == "succeeded":
             content_str = str(get_result.content)
             video_url_match = re.search(r"video_url='(.*?)'", content_str)
             if video_url_match:
-                return video_url_match.group(1), "任务完成"
-            return None, "任务成功但未找到视频 URL"
+                return video_url_match.group(1), "Task completed"
+            return None, "Task successful but video URL not found"
         else:
-            return None, f"任务失败: {get_result.status}"
+            return None, f"Task failed: {get_result.status}"
     except Exception as e:
-        return None, f"火山引擎生成失败: {str(e)}"
+        return None, f"Volcengine generation failed: {str(e)}"