Update app.py

app.py

@@ -1,20 +1,30 @@
-import os, shutil
+# app.py
+import os
+import shutil
 import gradio as gr
 from runner import AutomationRunner
 
-# đảm bảo thư mục output luôn sạch
-def clean_outputs():
-    if os.path.exists("outputs"):
-        shutil.rmtree("outputs")
-    os.makedirs("outputs", exist_ok=True)
-    os.makedirs("images", exist_ok=True)
+COOKIES_FILE = "cookies.json"
 
-def run_automation(prompts, avatar, headless, wait_time, mode):
-    clean_outputs()
-    # lưu prompt
+def clean_workspace():
+    for d in ("outputs", "images"):
+        if os.path.exists(d):
+            shutil.rmtree(d)
+        os.makedirs(d, exist_ok=True)
+
+def run_automation(prompts, avatar, cookies_file, headless, wait_time, mode):
+    clean_workspace()
+
+    # 1) Lưu cookies.json nếu user upload
+    if cookies_file is not None:
+        # cookies_file.name là đường dẫn tạm tới file upload
+        shutil.copy(cookies_file.name, COOKIES_FILE)
+
+    # 2) Ghi prompt vào input.txt
     with open("input.txt", "w", encoding="utf-8") as f:
         f.write(prompts)
 
+    # 3) Lưu ảnh nhân vật nếu có
     avatar_path = None
     if avatar is not None:
         avatar.save(os.path.join("images", avatar.name))
@@ -24,6 +34,7 @@ def run_automation(prompts, avatar, headless, wait_time, mode):
     def log_cb(msg):
         logs.append(msg)
 
+    # 4) Run automation
     runner = AutomationRunner(
         headless=headless,
         wait_time=wait_time,
@@ -34,27 +45,30 @@ def run_automation(prompts, avatar, headless, wait_time, mode):
     )
     runner.run()
 
-    # collect kết quả
-    videos = sorted([os.path.join("outputs",f) for f in os.listdir("outputs") if f.endswith(".mp4")])
-    return logs, videos
+    # 5) Thu thập video đầu ra
+    videos = sorted(
+        [os.path.join("outputs", f) for f in os.listdir("outputs") if f.endswith(".mp4")]
+    )
+    return "\n".join(logs), videos
 
 with gr.Blocks() as demo:
     gr.Markdown("## HaiLuo AI Video Automation")
     with gr.Row():
-        with gr.Column():
-            prompts = gr.Textbox(lines=8, label="Nhập các Scene (mỗi dòng 'Scene X: ...')", value="Scene 1: ")
-            avatar = gr.File(label="Ảnh nhân vật (tuỳ chọn)")
+        with gr.Column(scale=2):
+            prompts = gr.Textbox(lines=8, label="Nhập các Scene (mỗi dòng `Scene X: ...`)", value="Scene 1: ")
+            avatar = gr.File(label="Ảnh nhân vật (tuỳ chọn)", file_count=1, file_types=["image"])
+            cookies_file = gr.File(label="Upload cookies.json (bắt buộc lần đầu)", file_count=1, file_types=[".json"])
             mode = gr.Radio(["subject","text"], label="Chế độ", value="subject")
             headless = gr.Checkbox(label="Chạy headless", value=True)
             wait_time = gr.Slider(60, 1800, value=600, step=30, label="Thời gian chờ (giây)")
             run_btn = gr.Button("🚀 Chạy Automation")
-        with gr.Column():
+        with gr.Column(scale=1):
             log_output = gr.Textbox(lines=15, label="Logs", interactive=False)
-            video_files = gr.File(label="Video đầu ra", file_count="multiple")
+            video_out = gr.File(label="Download Videos", file_count="multiple")
     run_btn.click(
         run_automation,
-        inputs=[prompts, avatar, headless, wait_time, mode],
-        outputs=[log_output, video_files]
+        inputs=[prompts, avatar, cookies_file, headless, wait_time, mode],
+        outputs=[log_output, video_out]
     )
 
 if __name__ == "__main__":