Update app.py

app.py

@@ -2,6 +2,8 @@ import os
 import requests
 import subprocess
 import gradio as gr
+from datetime import datetime
+from huggingface_hub import HfApi, Repository
 
 # Token Hugging Face từ biến môi trường
 hf_token = os.getenv("HF_TOKEN")
@@ -31,6 +33,28 @@ if not os.path.exists(app_path):
 if not os.path.exists(model_path):
     download_file(model_url, model_path, hf_token)
 
+# Hàm lưu kết quả lên Hugging Face Dataset
+def save_to_huggingface(image_path, hf_token):
+    # Tạo tên thư mục và tên file dựa trên ngày giờ hiện tại
+    now = datetime.now()
+    folder_name = f"SetImages+{now.strftime('%d/%m/%Y')}"
+    file_name = f"Image+{now.strftime('%S/%M/%H')}.png"
+    
+    # Khởi tạo API Hugging Face
+    api = HfApi(token=hf_token)
+    
+    # Tải ảnh lên dataset
+    try:
+        api.upload_file(
+            path_or_fileobj=image_path,
+            path_in_repo=f"{folder_name}/{file_name}",
+            repo_id="ArrcttacsrjksX/Deffusion",
+            repo_type="dataset"
+        )
+        print(f"Uploaded: {folder_name}/{file_name}")
+    except Exception as e:
+        print(f"Failed to upload: {e}")
+
 # Hàm xử lý chạy ứng dụng
 def run_command(
     prompt, mode, height, width, steps, seed, cfg_scale, strength, sampling_method,
@@ -95,7 +119,12 @@ def run_command(
         # Kiểm tra kết quả và trả về
         if process.returncode == 0:
             output_path = "./output.png"  # Đường dẫn ảnh đầu ra mặc định
-            return output_path if os.path.exists(output_path) else None, "\n".join(logs)
+            if os.path.exists(output_path):
+                # Lưu ảnh lên Hugging Face Dataset
+                save_to_huggingface(output_path, hf_token)
+                return output_path, "\n".join(logs)
+            else:
+                return None, "\n".join(logs)
         else:
             error_log = process.stderr.read()  # Đọc lỗi
             logs.append(error_log)