Update app.py

app.py

@@ -7,7 +7,6 @@ import time
 import random
 import gradio as gr
 import shutil
-import threading
 
 # Hàm lấy gợi ý tìm kiếm từ Google
 def get_google_suggestions(query):
@@ -60,12 +59,12 @@ def zip_folder(folder_path):
     output_zip = os.path.join(os.path.dirname(folder_path), "downloaded_images")
     try:
         shutil.make_archive(output_zip, 'zip', folder_path)
-        return f"Đã nén thành công: {output_zip}.zip"
+        return output_zip + ".zip", "Đã nén thành công"
     except Exception as e:
-        return f"Lỗi khi nén: {str(e)}"
+        return None, f"Lỗi khi nén: {str(e)}"
 
 # Hàm chính tải ảnh
-def start_download(initial_query, target_images, max_per_term, save_folder):
+def start_download(initial_query, target_images, max_per_term, save_folder, zip_files):
     status_log = []
     
     def status_callback(message):
@@ -100,36 +99,83 @@ def start_download(initial_query, target_images, max_per_term, save_folder):
         time.sleep(random.uniform(2, 5))
     
     status_callback(f"Hoàn tất! Đã tải {total_downloaded} ảnh.")
-    status_callback("Đang nén thư mục...")
-    zip_result = zip_folder(save_folder)
-    status_callback(zip_result)
     
-    return "\n".join(status_log)
+    zip_file_path = None
+    if zip_files:
+        status_callback("Đang nén thư mục...")
+        zip_file_path, zip_message = zip_folder(save_folder)
+        status_callback(zip_message)
+    
+    return "\n".join(status_log), zip_file_path
 
 # Giao diện Gradio
 def create_interface():
-    with gr.Blocks(title="Image Downloader") as demo:
-        gr.Markdown("# Image Downloader")
-        gr.Markdown("Tải ảnh từ Google Images và nén thành file zip.")
+    css = """
+    .container {
+        background: linear-gradient(135deg, #e0eafc, #cfdef3);
+        padding: 20px;
+        border-radius: 10px;
+        box-shadow: 0 4px 8px rgba(0,0,0,0.1);
+    }
+    h1 {
+        color: #2c3e50;
+        text-align: center;
+        font-family: Arial, sans-serif;
+    }
+    .input-box {
+        background: white;
+        border-radius: 5px;
+        padding: 10px;
+        margin: 5px 0;
+    }
+    .status-box {
+        background: #f9f9f9;
+        border: 1px solid #ddd;
+        border-radius: 5px;
+        padding: 10px;
+        height: 300px;
+        overflow-y: auto;
+    }
+    .button {
+        background: #3498db !important;
+        color: white !important;
+        border-radius: 5px !important;
+        padding: 10px 20px !important;
+    }
+    .button:hover {
+        background: #2980b9 !important;
+    }
+    """
+    
+    with gr.Blocks(css=css, title="Image Downloader") as demo:
+        gr.Markdown("<h1>📸 Image Downloader</h1>")
+        gr.Markdown("Tải ảnh từ Google Images và tùy chọn nén thành file zip.")
         
-        with gr.Row():
-            with gr.Column():
-                initial_query = gr.Textbox(label="Từ khóa ban đầu", value="free images")
-                target_images = gr.Number(label="Số lượng ảnh cần tải", value=10000, precision=0)
-                max_per_term = gr.Number(label="Số ảnh tối đa mỗi từ khóa", value=20, precision=0)
-                save_folder = gr.Textbox(label="Thư mục lưu", value="free_images")
+        with gr.Row(elem_classes="container"):
+            with gr.Column(scale=1):
+                gr.Markdown("### Cài đặt", elem_classes="input-box")
+                initial_query = gr.Textbox(label="Từ khóa ban đầu", value="free images", elem_classes="input-box")
+                target_images = gr.Number(label="Số lượng ảnh", value=10000, precision=0, elem_classes="input-box")
+                max_per_term = gr.Number(label="Ảnh tối đa mỗi từ khóa", value=20, precision=0, elem_classes="input-box")
+                save_folder = gr.Textbox(label="Thư mục lưu", value="free_images", elem_classes="input-box")
+                zip_files = gr.Checkbox(label="Nén thành file zip", value=True, elem_classes="input-box")
+                submit_btn = gr.Button("Bắt đầu tải", elem_classes="button")
             
-            with gr.Column():
-                output = gr.Textbox(label="Trạng thái", lines=20, interactive=False)
-                submit_btn = gr.Button("Bắt đầu tải")
+            with gr.Column(scale=2):
+                gr.Markdown("### Trạng thái", elem_classes="input-box")
+                output_status = gr.Textbox(label="", lines=15, interactive=False, elem_classes="status-box")
+                output_file = gr.File(label="Tải file zip (nếu có)", visible=False)
         
-        def run_download(query, target, max_term, folder):
-            return start_download(query, int(target), int(max_term), folder)
+        def run_download(query, target, max_term, folder, zip_opt):
+            status, zip_path = start_download(query, int(target), int(max_term), folder, zip_opt)
+            if zip_path:
+                return status, gr.File.update(value=zip_path, visible=True)
+            return status, gr.File.update(visible=False)
         
         submit_btn.click(
             fn=run_download,
-            inputs=[initial_query, target_images, max_per_term, save_folder],
-            outputs=output
+            inputs=[initial_query, target_images, max_per_term, save_folder, zip_files],
+            outputs=[output_status, output_file]
         )
     
     return demo