Update app.py

app.py

@@ -7,6 +7,8 @@ import time
 import random
 import gradio as gr
 import shutil
+import tkinter as tk
+from tkinter import filedialog
 
 # Các hàm phụ trợ giữ nguyên
 def get_google_suggestions(query):
@@ -26,7 +28,7 @@ def download_images(search_term, num_images_per_term, save_folder, status_callba
     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"}
     
     try:
-        response = requests.get(search_url, headers=headers, timeout=10)
+        response = requests.get(url, headers=headers, timeout=10)
         response.raise_for_status()
     except:
         status_callback(f"Lỗi truy cập {search_term}")
@@ -105,54 +107,80 @@ 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 cải tiến
+# Hàm chọn thư mục bằng tkinter
+def select_folder():
+    root = tk.Tk()
+    root.withdraw()  # Ẩn cửa sổ chính của tkinter
+    folder = filedialog.askdirectory(initialdir=os.getcwd(), title="Chọn thư mục lưu")
+    root.destroy()
+    return folder if folder else os.path.join(os.getcwd(), "free_images")
+
+# Giao diện Gradio
 def create_interface():
     css = """
+    body {
+        background: #2b2b2b;
+        font-family: 'Segoe UI', sans-serif;
+    }
     .container {
         max-width: 1000px;
-        margin: 0 auto;
-        background: #ffffff;
-        padding: 30px;
-        border-radius: 15px;
-        box-shadow: 0 5px 15px rgba(0,0,0,0.1);
+        margin: 20px auto;
+        background: #36393f;
+        padding: 25px;
+        border-radius: 10px;
+        box-shadow: 0 2px 10px rgba(0,0,0,0.3);
     }
     h1 {
-        color: #1a73e8;
+        color: #7289da;
         text-align: center;
-        font-family: 'Segoe UI', sans-serif;
+        margin-bottom: 15px;
+    }
+    .description {
+        color: #b9bbbe;
+        text-align: center;
+        font-size: 14px;
         margin-bottom: 20px;
     }
     .input-group {
-        background: #f8f9fa;
+        background: #40444b;
         padding: 15px;
-        border-radius: 10px;
+        border-radius: 8px;
         margin-bottom: 15px;
     }
     .status-box {
-        background: #f1f3f5;
-        border: 1px solid #e9ecef;
-        border-radius: 10px;
+        background: #2f3136;
+        border: 1px solid #202225;
+        border-radius: 8px;
         padding: 15px;
         height: 350px;
+        color: #dcddde;
         font-family: 'Courier New', monospace;
         font-size: 14px;
     }
     .button-primary {
-        background: #1a73e8 !important;
+        background: #7289da !important;
         color: white !important;
         border: none !important;
-        border-radius: 8px !important;
-        padding: 12px 25px !important;
+        border-radius: 5px !important;
+        padding: 10px 20px !important;
         font-weight: bold !important;
         transition: all 0.3s !important;
     }
     .button-primary:hover {
-        background: #1557b0 !important;
-        transform: translateY(-2px) !important;
+        background: #677bc4 !important;
+    }
+    .button-secondary {
+        background: #4f545c !important;
+        color: #dcddde !important;
+        border-radius: 5px !important;
+        padding: 8px 15px !important;
+    }
+    .button-secondary:hover {
+        background: #5c6169 !important;
     }
     .footer {
         text-align: center;
-        color: #666;
+        color: #72767d;
         font-size: 12px;
         margin-top: 20px;
     }
@@ -160,7 +188,7 @@ def create_interface():
 
     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>")
+        gr.Markdown("<p class='description'>Tải ảnh từ Google Images với giao diện tối giản và thân thiện</p>")
         
         with gr.Row(elem_classes="container"):
             with gr.Column(scale=1, min_width=300):
@@ -190,7 +218,7 @@ def create_interface():
                         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")
+                    folder_button = gr.Button("Chọn thư mục", elem_classes="button-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")
             
@@ -208,12 +236,9 @@ def create_interface():
         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,
+            fn=select_folder,
+            inputs=[],
             outputs=save_folder
         )