update settings

Signed-off-by: Zhang Jun <[email protected]>

.gitignore

@@ -3,4 +3,13 @@ __pycache__/
 .env
 .DS_Store
 venv/
-.ipynb_checkpoints/
+.ipynb_checkpoints/
+
+# Environment variables
+.env
+.env.local
+.env.development.local
+.env.test.local
+.env.production.local
+
+generated_website*

app.py

@@ -1,14 +1,22 @@
 import os
 import cv2
 import time
+import uuid
 import base64
 import re
 import gradio as gr
 from openai import OpenAI
 from concurrent.futures import ThreadPoolExecutor, as_completed
+from dotenv import load_dotenv
+
+# Load environment variables from .env file
+load_dotenv()
 
 # --- Configuration ---
-BASE_URL = "https://aistudio.baidu.com/llm/lmapi/v3"
+QIANFAN_URL = os.getenv("QIANFAN_URL", "")
+QIANFAN_TOKEN = os.getenv("QIANFAN_TOKEN", "")
+
+
 MODEL_NAME = "ernie-4.5-turbo-vl"
 MAX_CONCURRENT_REQUESTS = 4
 MAX_VIDEO_DURATION_SEC = 1800 
@@ -99,7 +107,7 @@ def main_process(video_file, progress=gr.Progress()):
     # 1. 初始化状态
     yield "⏳ Initializing...", None, None, None
     
-    api_key = os.environ.get("ERNIE_API_KEY")
+    api_key = QIANFAN_TOKEN
     if not api_key: raise gr.Error("Server Config Error: API KEY missing.")
     if not video_file: raise gr.Error("Please upload a video.")
 
@@ -112,7 +120,7 @@ def main_process(video_file, progress=gr.Progress()):
     if duration > MAX_VIDEO_DURATION_SEC:
         raise gr.Error(f"Video too long ({duration/60:.1f}m). Limit is 30m.")
 
-    client = OpenAI(api_key=api_key, base_url=BASE_URL)
+    client = OpenAI(api_key=api_key, base_url=QIANFAN_URL)
     
     # 2. 抽帧阶段
     yield "🎞️ Step 1/3: Extracting video frames...", None, None, None
@@ -144,7 +152,7 @@ def main_process(video_file, progress=gr.Progress()):
     progress(0.85, desc="Synthesizing code...")
     html_code = aggregate_and_generate_webpage(client, chunk_summaries)
     
-    output_path = "generated_website.html"
+    output_path = f"generated_website_{uuid.uuid4().hex}.html"
     with open(output_path, "w", encoding="utf-8") as f:
         f.write(html_code)
     

requirements.txt

@@ -1,3 +1,4 @@
 opencv-python-headless
 openai
-numpy
+numpy
+python-dotenv