Update app.py

app.py

@@ -8,7 +8,7 @@ import random
 import gradio as gr
 import shutil
 
-# Hàm lấy gợi ý tìm kiếm từ Google
+# Các hàm phụ trợ giữ nguyên
 def get_google_suggestions(query):
     url = f"http://suggestqueries.google.com/complete/search?client=firefox&q={query}"
     headers = {"User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/58.0.3029.110 Safari/537.3"}
@@ -18,7 +18,6 @@ def get_google_suggestions(query):
     except:
         return []
 
-# Hàm tải ảnh
 def download_images(search_term, num_images_per_term, save_folder, status_callback):
     if not os.path.exists(save_folder):
         os.makedirs(save_folder)
@@ -54,7 +53,6 @@ def download_images(search_term, num_images_per_term, save_folder, status_callba
     
     return count
 
-# Hàm nén thư mục
 def zip_folder(folder_path):
     output_zip = os.path.join(os.path.dirname(folder_path), "downloaded_images")
     try:
@@ -63,7 +61,6 @@ def zip_folder(folder_path):
     except Exception as 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, zip_files):
     status_log = []
     
@@ -108,79 +105,136 @@ def start_download(initial_query, target_images, max_per_term, save_folder, zip_
     
     return "\n".join(status_log), zip_file_path
 
-# Giao diện Gradio
+# Giao diện Gradio cải tiến
 def create_interface():
     css = """
     .container {
-        background: linear-gradient(135deg, #e0eafc, #cfdef3);
-        padding: 20px;
-        border-radius: 10px;
-        box-shadow: 0 4px 8px rgba(0,0,0,0.1);
+        max-width: 1000px;
+        margin: 0 auto;
+        background: #ffffff;
+        padding: 30px;
+        border-radius: 15px;
+        box-shadow: 0 5px 15px rgba(0,0,0,0.1);
     }
     h1 {
-        color: #2c3e50;
+        color: #1a73e8;
         text-align: center;
-        font-family: Arial, sans-serif;
+        font-family: 'Segoe UI', sans-serif;
+        margin-bottom: 20px;
     }
-    .input-box {
-        background: white;
-        border-radius: 5px;
-        padding: 10px;
-        margin: 5px 0;
+    .input-group {
+        background: #f8f9fa;
+        padding: 15px;
+        border-radius: 10px;
+        margin-bottom: 15px;
     }
     .status-box {
-        background: #f9f9f9;
-        border: 1px solid #ddd;
-        border-radius: 5px;
-        padding: 10px;
-        height: 300px;
-        overflow-y: auto;
+        background: #f1f3f5;
+        border: 1px solid #e9ecef;
+        border-radius: 10px;
+        padding: 15px;
+        height: 350px;
+        font-family: 'Courier New', monospace;
+        font-size: 14px;
     }
-    .button {
-        background: #3498db !important;
+    .button-primary {
+        background: #1a73e8 !important;
         color: white !important;
-        border-radius: 5px !important;
-        padding: 10px 20px !important;
+        border: none !important;
+        border-radius: 8px !important;
+        padding: 12px 25px !important;
+        font-weight: bold !important;
+        transition: all 0.3s !important;
+    }
+    .button-primary:hover {
+        background: #1557b0 !important;
+        transform: translateY(-2px) !important;
     }
-    .button:hover {
-        background: #2980b9 !important;
+    .footer {
+        text-align: center;
+        color: #666;
+        font-size: 12px;
+        margin-top: 20px;
     }
     """
-    
-    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.Blocks(css=css, title="Image Downloader Pro") as demo:
+        gr.Markdown("<h1>📷 Image Downloader Pro</h1>")
+        gr.Markdown("<p style='text-align: center; color: #666;'>Tải ảnh từ Google Images dễ dàng với giao diện thân thiện</p>")
         
         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(scale=1, min_width=300):
+                with gr.Group(elem_classes="input-group"):
+                    gr.Markdown("#### Cài đặt tải")
+                    initial_query = gr.Textbox(
+                        label="Từ khóa ban đầu",
+                        value="free images",
+                        placeholder="Nhập từ khóa tìm kiếm..."
+                    )
+                    target_images = gr.Slider(
+                        label="Số lượng ảnh",
+                        minimum=1,
+                        maximum=10000,
+                        value=100,
+                        step=1
+                    )
+                    max_per_term = gr.Slider(
+                        label="Ảnh tối đa mỗi từ khóa",
+                        minimum=1,
+                        maximum=50,
+                        value=20,
+                        step=1
+                    )
+                    save_folder = gr.Textbox(
+                        label="Thư mục lưu",
+                        value=os.path.join(os.getcwd(), "free_images"),
+                        placeholder="Đường dẫn thư mục..."
+                    )
+                    folder_button = gr.Button("Chọn thư mục", variant="secondary")
+                    zip_files = gr.Checkbox(label="Nén thành ZIP sau khi tải", value=True)
+                    submit_btn = gr.Button("Bắt đầu tải", elem_classes="button-primary")
             
             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)
+                with gr.Group(elem_classes="input-group"):
+                    gr.Markdown("#### Trạng thái")
+                    output_status = gr.Textbox(
+                        label="Nhật ký tải",
+                        lines=15,
+                        interactive=False,
+                        elem_classes="status-box"
+                    )
+                    output_file = gr.File(label="File ZIP (nếu có)", visible=False)
         
+        gr.Markdown("<p class='footer'>Powered by Gradio & xAI</p>")
+
+        # Xử lý chọn thư mục
+        def update_folder_path():
+            folder = gr.FileExplorer(glob="*/", root_dir=os.getcwd())
+            return folder
+
+        folder_button.click(
+            fn=update_folder_path,
+            outputs=save_folder
+        )
+
+        # Xử lý tải
         def run_download(query, target, max_term, folder, zip_opt):
+            if not folder:
+                return "Vui lòng chọn thư mục lưu!", gr.File.update(visible=False)
+                
             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, zip_files],
             outputs=[output_status, output_file]
         )
-    
+
     return demo
 
-# Chạy ứng dụng
 if __name__ == "__main__":
     interface = create_interface()
     interface.launch()