Update app.py

app.py

@@ -3,16 +3,11 @@ 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. تنظیمات لاگینگ برای نمایش پیام ها در کنسول اسپیس
+# 1. تنظیمات لاگینگ
 logging.basicConfig(
     format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
     level=logging.INFO
@@ -23,26 +18,20 @@ 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"  # <-- آدرس دیتاسنت شما
+HF_REPO_ID = "Asrasahar/Ezmary-uploads"
 
-# 3. بررسی وجود توکن ها
+# 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}")
+logger.info(f"Public File Upload API Initialized. Target repo: {HF_REPO_ID}")
 
 # --- توابع کمکی (بدون تغییر) ---
 def get_folder_by_mimetype(mimetype):
-    """بر اساس نوع MIME، نام پوشه مقصد را برمی‌گرداند."""
-    if not mimetype:
-        return "others"
+    if not mimetype: return "others"
     main_type = mimetype.split('/')[0]
     if main_type == "image": return "images"
     elif main_type == "video": return "videos"
@@ -54,37 +43,20 @@ def get_folder_by_mimetype(mimetype):
     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",
-    )
-
+    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 (برای وب سایت) ---
-
+# --- بخش API ---
 @app.route('/')
 def serve_html():
     return send_from_directory(app.static_folder, 'index.html')
@@ -92,138 +64,20 @@ def serve_html():
 @app.route('/upload', methods=['POST'])
 def upload_file_endpoint():
     try:
-        # حالت اول: آپلود از کامپیوتر
+        # آپلود از کامپیوتر یا ربات (حالت FormData)
         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
+            return jsonify({"error": "Invalid request. 'file' part is missing."}), 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)