Update app.py

app.py

@@ -35,8 +35,12 @@ st.set_page_config(page_title="Hugging Face Space Tools", page_icon="🛠️")
 
 st.title("🛠️ ابزارهای مدیریت اسپیس هاگینگ‌فیس")
 
-# ایجاد دو تب
-tab1, tab2 = st.tabs(["🔄 انتقال به گیت‌هاب", "📦 دانلود اسپیس (فایل ZIP)"])
+# ایجاد سه تب (اضافه شدن تب انتقال به هاگینگ‌فیس)
+tab1, tab2, tab3 = st.tabs([
+    "🔄 انتقال به گیت‌هاب", 
+    "📤 انتقال به هاگینگ‌فیس", 
+    "📦 دانلود اسپیس (فایل ZIP)"
+])
 
 # ---------------------------------------------------------
 # تب اول: انتقال به گیت‌هاب (پشتیبانی از لینک و فایل زیپ + حل مشکل پوشه اضافه)
@@ -98,17 +102,14 @@ with tab1:
                         zip_ref.extractall(temp_dir)
                     
                     # بررسی و رفع مشکل پوشه اضافه (انتقال محتویات به روت اصلی)
-                    # پوشه‌های سیستمی مک را نادیده می‌گیریم تا اختلال ایجاد نکنند
                     extracted_items = [item for item in os.listdir(temp_dir) if item != '__MACOSX']
                     
                     # اگر فقط یک پوشه در مسیر وجود داشت
                     if len(extracted_items) == 1:
                         single_folder_path = os.path.join(temp_dir, extracted_items[0])
                         if os.path.isdir(single_folder_path):
-                            # تمام فایل‌های داخل آن پوشه را یک سطح بالاتر (به روت) منتقل می‌کنیم
                             for item in os.listdir(single_folder_path):
                                 shutil.move(os.path.join(single_folder_path, item), temp_dir)
-                            # پوشه خالی باقیمانده را حذف می‌کنیم
                             os.rmdir(single_folder_path)
                 
                 # مرحله 2: عملیات گیت و پوش
@@ -149,9 +150,116 @@ with tab1:
                     shutil.rmtree(temp_dir, ignore_errors=True)
 
 # ---------------------------------------------------------
-# تب دوم: دانلود فایل‌ها به صورت زیپ
+# تب دوم: انتقال به هاگینگ‌فیس (پشتیبانی از لینک ریپازیتوری و فایل ZIP)
 # ---------------------------------------------------------
 with tab2:
+    st.info("منبع خود را برای انتقال به اسپیس هاگینگ‌فیس انتخاب کنید:")
+    
+    # انتخاب نوع منبع خارج از فرم برای رندر داینامیک
+    source_type_hf = st.radio("نوع منبع (برای هاگینگ‌فیس):", ["لینک ریپازیتوری (گیت‌هاب و غیره)", "آپلود فایل ZIP"], key="src_type_hf")
+    
+    with st.form("transfer_hf_form"):
+        src_url_hf = ""
+        uploaded_zip_hf = None
+        
+        if source_type_hf == "لینک ریپازیتوری (گیت‌هاب و غیره)":
+            src_url_hf = st.text_input("لینک ریپازیتوری منبع (مثلاً گیت‌هاب)", 
+                                       placeholder="https://github.com/user/repo.git", key="src_url_hf_input")
+        else:
+            uploaded_zip_hf = st.file_uploader("آپلود فایل ZIP (منبع)", type=["zip"], key="zip_hf_input")
+            
+        st.markdown("---")
+        hf_target_url = st.text_input("لینک کامل اسپیس هاگینگ‌فیس (مقصد)", 
+                                      placeholder="https://huggingface.co/spaces/user/space-name", key="target_hf_url")
+        
+        hf_token = st.text_input("توکن هاگینگ‌فیس (با دسترسی Write)", type="password", key="hf_token_input")
+        
+        submitted_hf = st.form_submit_button("شروع عملیات انتقال به هاگینگ‌فیس 🚀")
+
+    if submitted_hf:
+        if not hf_target_url or not hf_token:
+            st.error("لطفاً لینک اسپیس مقصد و توکن هاگینگ‌فیس را وارد کنید.")
+        elif source_type_hf == "لینک ریپازیتوری (گیت‌هاب و غیره)" and not src_url_hf:
+            st.error("لطفاً لینک ریپازیتوری منبع را وارد کنید.")
+        elif source_type_hf == "آپلود فایل ZIP" and not uploaded_zip_hf:
+            st.error("لطفاً فایل ZIP را آپلود کنید.")
+        elif "huggingface.co/spaces/" not in hf_target_url:
+            st.error("❌ لینک اسپیس مقصد اشتباه است! لینک باید با ساختار huggingface.co/spaces/... باشد.")
+        else:
+            status_area_hf = st.empty()
+            temp_dir_hf = "temp_repo_hf"
+            
+            if os.path.exists(temp_dir_hf):
+                shutil.rmtree(temp_dir_hf, ignore_errors=True)
+                
+            try:
+                # فرمت مناسب لینک برای احراز هویت توکن هاگینگ‌فیس
+                auth_hf_url = hf_target_url.replace("https://", f"https://__token__:{hf_token}@")
+                
+                # مرحله 1: آماده‌سازی فایل‌ها (دانلود یا استخراج زیپ)
+                if source_type_hf == "لینک ریپازیتوری (گیت‌هاب و غیره)":
+                    status_area_hf.info("📥 در حال دانلود (کلون) ریپازیتوری منبع...")
+                    success, msg = run_command(f"git clone {src_url_hf} {temp_dir_hf}")
+                    if not success:
+                        st.error(f"خطا در کلون کردن ریپازیتوری منبع: {msg}")
+                        st.stop()
+                else:
+                    status_area_hf.info("📥 در حال استخراج فایل ZIP...")
+                    os.makedirs(temp_dir_hf, exist_ok=True)
+                    with zipfile.ZipFile(uploaded_zip_hf, 'r') as zip_ref:
+                        zip_ref.extractall(temp_dir_hf)
+                    
+                    # رفع مشکل پوشه اضافه درون فایل ZIP
+                    extracted_items = [item for item in os.listdir(temp_dir_hf) if item != '__MACOSX']
+                    if len(extracted_items) == 1:
+                        single_folder_path = os.path.join(temp_dir_hf, extracted_items[0])
+                        if os.path.isdir(single_folder_path):
+                            for item in os.listdir(single_folder_path):
+                                shutil.move(os.path.join(single_folder_path, item), temp_dir_hf)
+                            os.rmdir(single_folder_path)
+                
+                # مرحله 2: ساخت مخزن تمیز گیت و پوش به هاگینگ‌فیس
+                os.chdir(temp_dir_hf)
+                status_area_hf.info("⚙️ در حال تنظیمات گیت...")
+                
+                if os.path.exists(".git"):
+                    shutil.rmtree(".git", ignore_errors=True)
+                
+                run_command("git init")
+                run_command("git add .")
+                run_command('git config user.email "bot@transfer.com"')
+                run_command('git config user.name "Transfer Bot"')
+                run_command('git commit -m "Moved to Hugging Face automatically"')
+                run_command("git branch -M main")
+                
+                status_area_hf.info("📤 در حال آپلود فایل‌ها به اسپیس هاگینگ‌فیس...")
+                push_success, push_msg = run_command(f"git push -f {auth_hf_url} main")
+                
+                if push_success:
+                    st.success("✅ انتقال موفقیت‌آمیز بود! فرآیند بیلد (Build) اسپیس شما در هاگینگ‌فیس آغاز شد.")
+                    st.balloons()
+                else:
+                    # مخفی‌سازی توکن جهت مسائل امنیتی در لاگ خطا
+                    clean_error = push_msg.replace(hf_token, "***")
+                    st.error(f"خطا در آپلود به هاگینگ‌فیس: {clean_error}")
+                
+                os.chdir("..")
+                    
+            except Exception as e:
+                st.error(f"خطای سیستمی: {e}")
+            finally:
+                try:
+                    if os.getcwd().endswith(temp_dir_hf):
+                        os.chdir("..")
+                except:
+                    pass
+                if os.path.exists(temp_dir_hf):
+                    shutil.rmtree(temp_dir_hf, ignore_errors=True)
+
+# ---------------------------------------------------------
+# تب سوم: دانلود فایل‌ها به صورت زیپ
+# ---------------------------------------------------------
+with tab3:
     st.info("با استفاده از این بخش می‌توانید کل فایل‌های یک اسپیس عمومی را به صورت ZIP دانلود کنید.")
     
     hf_url_zip = st.text_input("لینک صفحه اسپیس هاگینگ فیس", 
@@ -202,4 +310,4 @@ with tab2:
                 st.error(f"خطای سیستمی: {e}")
             finally:
                 if os.path.exists(temp_dir_zip):
-                    shutil.rmtree(temp_dir_zip, ignore_errors=True)
+                    shutil.rmtree(temp_dir_zip, ignore_errors=True)