Update app.py

app.py

@@ -7,8 +7,6 @@ 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):
@@ -28,7 +26,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(url, headers=headers, timeout=10)
+        response = requests.get(search_url, headers=headers, timeout=10)
         response.raise_for_status()
     except:
         status_callback(f"Lỗi truy cập {search_term}")
@@ -107,20 +105,6 @@ def start_download(initial_query, target_images, max_per_term, save_folder, zip_
     
     return "\n".join(status_log), zip_file_path
 
-# Hàm lấy danh sách thư mục con
-def get_subfolders(root_dir):
-    if not os.path.exists(root_dir):
-        return [root_dir]
-    return [os.path.join(root_dir, d) for d in os.listdir(root_dir) if os.path.isdir(os.path.join(root_dir, d))]
-
-# Hàm chọn thư mục bằng tkinter
-def select_folder():
-    root = tk.Tk()
-    root.withdraw()
-    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 = """
@@ -190,13 +174,6 @@ def create_interface():
         font-size: 12px;
         margin-top: 20px;
     }
-    .folder-explorer {
-        background: #2f3136;
-        padding: 10px;
-        border-radius: 5px;
-        max-height: 150px;
-        overflow-y: auto;
-    }
     """
 
     with gr.Blocks(css=css, title="Image Downloader Pro") as demo:
@@ -226,21 +203,11 @@ def create_interface():
                         value=20,
                         step=1
                     )
-                    # Ô khám phá thư mục
-                    root_folder = gr.Textbox(
-                        label="Thư mục gốc",
-                        value=os.getcwd(),
-                        placeholder="Nhập thư mục gốc..."
-                    )
-                    folder_dropdown = gr.Dropdown(
-                        label="Chọn thư mục con",
-                        choices=get_subfolders(os.getcwd()),
+                    save_folder = gr.Textbox(
+                        label="Thư mục lưu",
                         value=os.path.join(os.getcwd(), "free_images"),
-                        elem_classes="folder-explorer"
+                        placeholder="Đường dẫn thư mục..."
                     )
-                    with gr.Row():
-                        refresh_button = gr.Button("Làm mới", elem_classes="button-secondary")
-                        tk_folder_button = gr.Button("Chọn bằng hộp thoại", 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")
             
@@ -257,29 +224,10 @@ def create_interface():
         
         gr.Markdown("<p class='footer'>Powered by Gradio & xAI</p>")
 
-        # Hàm cập nhật danh sách thư mục con
-        def update_folders(root_dir):
-            subfolders = get_subfolders(root_dir)
-            return gr.Dropdown.update(choices=subfolders, value=subfolders[0] if subfolders else root_dir)
-
-        # Sự kiện làm mới danh sách thư mục
-        refresh_button.click(
-            fn=update_folders,
-            inputs=root_folder,
-            outputs=folder_dropdown
-        )
-
-        # Sự kiện chọn thư mục bằng tkinter
-        tk_folder_button.click(
-            fn=select_folder,
-            inputs=[],
-            outputs=folder_dropdown
-        )
-
         # 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)
+                return "Vui lòng nhập 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:
@@ -288,7 +236,7 @@ def create_interface():
 
         submit_btn.click(
             fn=run_download,
-            inputs=[initial_query, target_images, max_per_term, folder_dropdown, zip_files],
+            inputs=[initial_query, target_images, max_per_term, save_folder, zip_files],
             outputs=[output_status, output_file]
         )