Create app.py

app.py

@@ -0,0 +1,229 @@
+import os
+import uuid
+import requests
+import io
+import logging
+import threading # برای اجرای همزمان ربات و وب اپلیکیشن
+from flask import Flask, request, jsonify, send_from_directory
+from huggingface_hub import HfApi
+import mimetypes
+
+# کتابخانه های جدید برای ربات تلگرام
+from telegram import Update, File as TelegramFile
+from telegram.ext import Updater, CommandHandler, MessageHandler, Filters, CallbackContext
+
+# 1. تنظیمات لاگینگ برای نمایش پیام ها در کنسول اسپیس
+logging.basicConfig(
+    format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
+    level=logging.INFO
+)
+logger = logging.getLogger(__name__)
+
+app = Flask(__name__, static_folder='static', static_url_path='/static')
+
+# 2. خواندن متغیرهای محیطی
+HF_TOKEN = os.getenv("HF_TOKEN")
+TELEGRAM_TOKEN = os.getenv("TELEGRAM_TOKEN") # <-- توکن ربات تلگرام
+HF_REPO_ID = "Asrasahar/Ezmary-uploads"  # <-- آدرس دیتاسنت شما
+
+# 3. بررسی وجود توکن ها
+if not HF_TOKEN:
+    logger.error("FATAL ERROR: HF_TOKEN environment variable not set!")
+    exit()
+if not TELEGRAM_TOKEN:
+    logger.error("FATAL ERROR: TELEGRAM_TOKEN environment variable not set! Please set it in your Space's 'Repository secrets'.")
+    exit()
+
+# 4. ایجاد یک نمونه از HfApi
+hf_api = HfApi(token=HF_TOKEN)
+logger.info(f"Flask app initialized. Target dataset repository: {HF_REPO_ID}")
+
+# --- توابع کمکی (بدون تغییر) ---
+def get_folder_by_mimetype(mimetype):
+    """بر اساس نوع MIME، نام پوشه مقصد را برمی‌گرداند."""
+    if not mimetype:
+        return "others"
+    main_type = mimetype.split('/')[0]
+    if main_type == "image": return "images"
+    elif main_type == "video": return "videos"
+    elif main_type == "audio": return "audios"
+    elif main_type == "application":
+        if 'pdf' in mimetype: return "documents/pdfs"
+        elif 'zip' in mimetype or 'rar' in mimetype or '7z' in mimetype: return "archives"
+        else: return "documents/applications"
+    elif main_type == "text": return "documents/texts"
+    else: return "others"
+
+# --- تابع اصلی آپلود (بازنویسی شده برای استفاده مشترک) ---
+def process_and_upload(file_content: bytes, original_filename: str, mimetype: str) -> str:
+    """
+    محتوای فایل را گرفته، پردازش کرده و به Hugging Face آپلود می‌کند.
+    لینک مستقیم فایل آپلود شده را برمی‌گرداند.
+    """
+    # ایجاد نام فایل منحصر به فرد
+    file_extension = os.path.splitext(original_filename)[1] or mimetypes.guess_extension(mimetype) or ""
+    filename = f"{uuid.uuid4().hex}{file_extension}"
+    
+    # تعیین پوشه مقصد
+    target_folder = get_folder_by_mimetype(mimetype)
+    path_in_repo = f"{target_folder}/{filename}"
+    
+    # آپلود به Hugging Face Hub
+    file_stream = io.BytesIO(file_content)
+    logger.info(f"Uploading to Hugging Face: {path_in_repo}")
+    hf_api.upload_file(
+        path_or_fileobj=file_stream,
+        path_in_repo=path_in_repo,
+        repo_id=HF_REPO_ID,
+        repo_type="dataset",
+    )
+
+    direct_url = f"https://huggingface.co/datasets/{HF_REPO_ID}/resolve/main/{path_in_repo}"
+    logger.info(f"File uploaded successfully! URL: {direct_url}")
+    
+    return direct_url
+
+# --- بخش Flask (برای وب سایت) ---
+
+@app.route('/')
+def serve_html():
+    return send_from_directory(app.static_folder, 'index.html')
+
+@app.route('/upload', methods=['POST'])
+def upload_file_endpoint():
+    try:
+        # حالت اول: آپلود از کامپیوتر
+        if 'file' in request.files:
+            uploaded_file = request.files['file']
+            if uploaded_file.filename == '':
+                return jsonify({"error": "No selected file"}), 400
+
+            file_content = uploaded_file.read()
+            direct_url = process_and_upload(file_content, uploaded_file.filename, uploaded_file.mimetype)
+            return jsonify({"message": "File uploaded successfully!", "hf_url": direct_url}), 200
+
+        # حالت دوم: آپلود از طریق لینک
+        elif request.is_json and 'url' in request.json:
+            file_url = request.json['url']
+            if not file_url:
+                return jsonify({"error": "No URL provided"}), 400
+
+            logger.info(f"Downloading file from URL: {file_url}")
+            response = requests.get(file_url, stream=True, timeout=30)
+            response.raise_for_status()
+            
+            file_content = response.content
+            mimetype = response.headers.get('Content-Type')
+            original_filename = file_url.split('/')[-1].split('?')[0]
+
+            direct_url = process_and_upload(file_content, original_filename, mimetype)
+            return jsonify({"message": "File uploaded successfully!", "hf_url": direct_url}), 200
+            
+        else:
+            return jsonify({"error": "Invalid request"}), 400
+
+    except requests.exceptions.RequestException as e:
+        logger.error(f"Failed to download from URL: {e}", exc_info=True)
+        return jsonify({"error": f"Failed to download from URL: {str(e)}"}), 400
+    except Exception as e:
+        logger.error(f"An internal server error occurred: {e}", exc_info=True)
+        return jsonify({"error": f"An internal server error occurred: {str(e)}"}), 500
+
+# --- بخش ربات تلگرام ---
+
+def start(update: Update, context: CallbackContext) -> None:
+    """پاسخ به دستور /start"""
+    welcome_message = "سلام! من ربات آپلودر فایل هستم.\n\n" \
+                      "فایل (عکس، ویدیو، داکیومنت و...) یا لینک مستقیم آن را برای من بفرستید تا آن را برای شما به یک لینک مستقیم دائمی تبدیل کنم."
+    update.message.reply_text(welcome_message)
+
+def handle_file(update: Update, context: CallbackContext) -> None:
+    """پردازش فایل های ارسالی (داکیومنت، ویدیو، عکس و...)"""
+    message = update.message
+    file_to_process = message.document or message.video or message.audio or (message.photo[-1] if message.photo else None)
+    
+    if not file_to_process:
+        message.reply_text("نوع فایل ارسالی پشتیبانی نمی‌شود.")
+        return
+
+    # ارسال پیام "در حال پردازش" به کاربر
+    status_message = message.reply_text("فایل دریافت شد، در حال پردازش و آپلود...")
+    
+    try:
+        tg_file: TelegramFile = file_to_process.get_file()
+        file_content = tg_file.download_as_bytearray()
+        
+        # نام فایل و نوع آن را از تلگرام می گیریم
+        original_filename = file_to_process.file_name if hasattr(file_to_process, 'file_name') else tg_file.file_path.split('/')[-1]
+        mimetype = file_to_process.mime_type or mimetypes.guess_type(original_filename)[0]
+        
+        logger.info(f"Received file from Telegram: {original_filename} (MIME: {mimetype})")
+        
+        # آپلود فایل با استفاده از تابع مشترک
+        direct_url = process_and_upload(bytes(file_content), original_filename, mimetype)
+        
+        # ویرایش پیام "در حال پردازش" و ارسال لینک نهایی
+        status_message.edit_text(f"✅ آپلود با موفقیت انجام شد!\n\nلینک مستقیم شما:\n{direct_url}")
+        
+    except Exception as e:
+        logger.error(f"Error processing Telegram file: {e}", exc_info=True)
+        status_message.edit_text(f"❌ خطا در پردازش فایل: {e}")
+
+def handle_text(update: Update, context: CallbackContext) -> None:
+    """پردازش پیام های متنی (برای لینک ها)"""
+    message = update.message
+    url = message.text.strip()
+    
+    # بررسی اینکه آیا متن یک لینک معتبر است یا نه
+    if url.startswith(('http://', 'https://')):
+        status_message = message.reply_text("لینک دریافت شد، در حال دانلود و آپلود...")
+        try:
+            logger.info(f"Downloading file from URL via Telegram: {url}")
+            response = requests.get(url, stream=True, timeout=30)
+            response.raise_for_status()
+            
+            file_content = response.content
+            mimetype = response.headers.get('Content-Type')
+            original_filename = url.split('/')[-1].split('?')[0]
+
+            # آپلود فایل با استفاده از تابع مشترک
+            direct_url = process_and_upload(file_content, original_filename, mimetype)
+            status_message.edit_text(f"✅ آپلود از لینک با موفقیت انجام شد!\n\nلینک مستقیم شما:\n{direct_url}")
+            
+        except requests.exceptions.RequestException as e:
+            logger.error(f"Failed to download from URL (Telegram): {e}", exc_info=True)
+            status_message.edit_text(f"❌ خطا در دانلود از لینک: {e}")
+        except Exception as e:
+            logger.error(f"An internal error occurred (Telegram URL): {e}", exc_info=True)
+            status_message.edit_text(f"❌ خطا در پردازش لینک: {e}")
+    else:
+        message.reply_text("لطفا یک فایل یا یک لینک معتبر ارسال کنید.")
+
+
+def run_bot():
+    """تابع اصلی برای راه اندازی و اجرای ربات تلگرام"""
+    updater = Updater(TELEGRAM_TOKEN)
+    dispatcher = updater.dispatcher
+
+    # تعریف دستورات و پاسخ ها
+    dispatcher.add_handler(CommandHandler("start", start))
+    dispatcher.add_handler(MessageHandler(Filters.text & ~Filters.command, handle_text))
+    dispatcher.add_handler(MessageHandler(Filters.document | Filters.video | Filters.photo | Filters.audio, handle_file))
+
+    # راه اندازی ربات
+    updater.start_polling()
+    logger.info("Telegram bot started polling...")
+    # updater.idle() # این خط لازم نیست چون وب سرور برنامه را فعال نگه می دارد
+
+# --- راه اندازی برنامه ---
+
+# اجرای ربات در یک ترد (نخ) جداگانه تا با وب سرور تداخل نداشته باشد
+bot_thread = threading.Thread(target=run_bot)
+bot_thread.daemon = True # با بستن برنامه اصلی، این ترد هم بسته می شود
+bot_thread.start()
+
+# این بخش فقط برای اجرای محلی (local) است
+# در Hugging Face Spaces، از gunicorn برای اجرای برنامه استفاده می شود
+if __name__ == '__main__':
+    port = int(os.environ.get('PORT', 7860))
+    app.run(host='0.0.0.0', port=port, debug=False)