Update app.py

app.py

@@ -5,8 +5,7 @@ import urllib.request
 import json
 import time
 import random
-import tkinter as tk
-from tkinter import ttk, messagebox, filedialog
+import gradio as gr
 import shutil
 import threading
 
@@ -56,132 +55,86 @@ def download_images(search_term, num_images_per_term, save_folder, status_callba
     
     return count
 
-# Hàm nén thư mục thành file zip
-def zip_folder(folder_path, output_zip):
+# 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:
         shutil.make_archive(output_zip, 'zip', folder_path)
-        return True
+        return f"Đã nén thành công: {output_zip}.zip"
     except Exception as e:
-        return str(e)
+        return f"Lỗi khi nén: {str(e)}"
 
 # Hàm chính tải ảnh
-def start_download(initial_query, target_images, save_folder, max_per_term, status_text):
-    def download_thread():
-        total_downloaded = 0
-        current_query = initial_query
-        used_queries = set()
-        
-        while total_downloaded < target_images:
-            suggestions = get_google_suggestions(current_query)
-            if not suggestions:
-                status_text(f"Hết gợi ý, dừng lại.")
-                break
-            
-            available_suggestions = [s for s in suggestions if s not in used_queries]
-            if not available_suggestions:
-                status_text("Hết gợi ý mới, dừng lại.")
-                break
-            
-            current_query = random.choice(available_suggestions)
-            used_queries.add(current_query)
-            
-            remaining = target_images - total_downloaded
-            images_to_download = min(max_per_term, remaining)
-            
-            status_text(f"Tìm kiếm: {current_query}")
-            downloaded = download_images(current_query, images_to_download, save_folder, status_text)
-            total_downloaded += downloaded
-            
-            status_text(f"Tổng: {total_downloaded}/{target_images}")
-            time.sleep(random.uniform(2, 5))
-        
-        status_text(f"Hoàn tất! Đã tải {total_downloaded} ảnh.")
-        
-        # Nén thư mục
-        status_text("Đang nén thư mục...")
-        zip_result = zip_folder(save_folder, os.path.join(os.path.dirname(save_folder), "downloaded_images"))
-        if zip_result is True:
-            status_text("Đã nén thành công: downloaded_images.zip")
-        else:
-            status_text(f"Lỗi khi nén: {zip_result}")
+def start_download(initial_query, target_images, max_per_term, save_folder):
+    status_log = []
     
-    thread = threading.Thread(target=download_thread)
-    thread.start()
-
-# Giao diện GUI
-class ImageDownloaderApp:
-    def __init__(self, root):
-        self.root = root
-        self.root.title("Image Downloader")
-        self.root.geometry("600x500")
-        
-        # Nhãn và ô nhập từ khóa
-        tk.Label(root, text="Từ khóa ban đầu:").pack(pady=5)
-        self.query_entry = tk.Entry(root, width=40)
-        self.query_entry.insert(0, "free images")
-        self.query_entry.pack()
-        
-        # Số lượng ảnh mục tiêu
-        tk.Label(root, text="Số lượng ảnh cần tải:").pack(pady=5)
-        self.target_entry = tk.Entry(root, width=10)
-        self.target_entry.insert(0, "10000")
-        self.target_entry.pack()
+    def status_callback(message):
+        status_log.append(message)
+    
+    total_downloaded = 0
+    current_query = initial_query
+    used_queries = set()
+    
+    while total_downloaded < target_images:
+        suggestions = get_google_suggestions(current_query)
+        if not suggestions:
+            status_callback("Hết gợi ý, dừng lại.")
+            break
         
-        # Số ảnh tối đa mỗi từ khóa
-        tk.Label(root, text="Số ảnh tối đa mỗi từ khóa:").pack(pady=5)
-        self.max_per_term_entry = tk.Entry(root, width=10)
-        self.max_per_term_entry.insert(0, "20")
-        self.max_per_term_entry.pack()
+        available_suggestions = [s for s in suggestions if s not in used_queries]
+        if not available_suggestions:
+            status_callback("Hết gợi ý mới, dừng lại.")
+            break
         
-        # Thư mục lưu
-        tk.Label(root, text="Thư mục lưu:").pack(pady=5)
-        self.folder_var = tk.StringVar(value="free_images")
-        folder_frame = tk.Frame(root)
-        folder_frame.pack()
-        tk.Entry(folder_frame, textvariable=self.folder_var, width=30).pack(side=tk.LEFT)
-        tk.Button(folder_frame, text="Chọn", command=self.choose_folder).pack(side=tk.LEFT, padx=5)
+        current_query = random.choice(available_suggestions)
+        used_queries.add(current_query)
         
-        # Nút bắt đầu
-        self.start_button = tk.Button(root, text="Bắt đầu tải", command=self.start_download)
-        self.start_button.pack(pady=10)
+        remaining = target_images - total_downloaded
+        images_to_download = min(max_per_term, remaining)
         
-        # Khu vực hiển thị trạng thái
-        tk.Label(root, text="Trạng thái:").pack(pady=5)
-        self.status_text = tk.Text(root, height=15, width=70)
-        self.status_text.pack()
+        status_callback(f"Tìm kiếm: {current_query}")
+        downloaded = download_images(current_query, images_to_download, save_folder, status_callback)
+        total_downloaded += downloaded
         
-    def choose_folder(self):
-        folder = filedialog.askdirectory()
-        if folder:
-            self.folder_var.set(folder)
+        status_callback(f"Tổng: {total_downloaded}/{target_images}")
+        time.sleep(random.uniform(2, 5))
     
-    def update_status(self, message):
-        self.status_text.insert(tk.END, f"{message}\n")
-        self.status_text.see(tk.END)
-        self.root.update_idletasks()
+    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)
     
-    def start_download(self):
-        try:
-            initial_query = self.query_entry.get()
-            target_images = int(self.target_entry.get())
-            max_per_term = int(self.max_per_term_entry.get())
-            save_folder = self.folder_var.get()
-            
-            if not initial_query or target_images <= 0 or max_per_term <= 0 or not save_folder:
-                messagebox.showerror("Lỗi", "Vui lòng nhập đầy đủ và hợp lệ các thông tin!")
-                return
-            
-            self.start_button.config(state="disabled")
-            self.status_text.delete(1.0, tk.END)
-            start_download(initial_query, target_images, save_folder, max_per_term, self.update_status)
+    return "\n".join(status_log)
+
+# 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.")
+        
+        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")
             
-        except ValueError:
-            messagebox.showerror("Lỗi", "Số lượng phải là số nguyên!")
-        finally:
-            self.start_button.config(state="normal")
+            with gr.Column():
+                output = gr.Textbox(label="Trạng thái", lines=20, interactive=False)
+                submit_btn = gr.Button("Bắt đầu tải")
+        
+        def run_download(query, target, max_term, folder):
+            return start_download(query, int(target), int(max_term), folder)
+        
+        submit_btn.click(
+            fn=run_download,
+            inputs=[initial_query, target_images, max_per_term, save_folder],
+            outputs=output
+        )
+    
+    return demo
 
 # Chạy ứng dụng
 if __name__ == "__main__":
-    root = tk.Tk()
-    app = ImageDownloaderApp(root)
-    root.mainloop()
+    interface = create_interface()
+    interface.launch()