Update bot.py

bot.py

@@ -4,60 +4,31 @@ import asyncio
 import aiosqlite
 import time
 import shutil
-import urllib.request
 import random
-from threading import Thread
-from http.server import BaseHTTPRequestHandler, HTTPServer
+import threading
 from pyrogram import Client, filters, idle
 from pyrogram.errors import FloodWait
 from pyrogram.types import InputMediaPhoto, InputMediaVideo, InputMediaDocument, InputMediaAudio
 from pyrogram.enums import ChatAction
+import gradio as gr
 
-# --- 1. الإعدادات الأساسية ---
+# ============ الإعدادات ============
 DATA_DIR = "./data"
 DOWNLOAD_DIR = os.path.join(DATA_DIR, "downloads")
 DB_PATH = os.path.join(DATA_DIR, 'bot_archiver.db')
 os.makedirs(DOWNLOAD_DIR, exist_ok=True)
 
-HEALTHCHECK_URL = os.environ.get("HEALTHCHECK_URL", "")
-MAX_FILE_SIZE = 2000 * 1024 * 1024 # 2GB
-BASE_UPLOAD_DELAY = 4 # زيادة الأمان الافتراضي للرفع
+MAX_FILE_SIZE = 2000 * 1024 * 1024  # 2GB
+BASE_UPLOAD_DELAY = 4
 
-download_queue = asyncio.Queue()
-
-# --- 2. سيرفر الحفاظ على التشغيل ---
-class RequestHandler(BaseHTTPRequestHandler):
-    def do_GET(self):
-        self.send_response(200)
-        self.end_headers()
-        self.wfile.write(b"Ultimate Radar Archiver Engine Active!")
-    def log_message(self, format, *args): pass
-
-Thread(target=lambda: HTTPServer(('0.0.0.0', 7860), RequestHandler).serve_forever(), daemon=True).start()
-
-# --- 3. إعدادات الحساب (تم إصلاح المعاملات وتعديل البصمة) ---
 API_ID = int(os.environ.get("API_ID", 0))
 API_HASH = os.environ.get("API_HASH", "")
 SESSION_STRING = os.environ.get("SESSION_STRING", "")
 TARGET_CHAT = int(os.environ.get("TARGET_CHAT", 0))
 
-app = Client(
-    "memory_session",
-    api_id=API_ID,
-    api_hash=API_HASH,
-    session_string=SESSION_STRING,
-    in_memory=True,
-    device_model="Samsung Galaxy S23 Ultra",
-    system_version="Android 14.0",
-    app_version="10.14.1",
-    lang_code="ar"
-)
-
-# --- 4. أدوات النظام وقاعدة البيانات ---
-def check_disk_space():
-    total, used, free = shutil.disk_usage(DATA_DIR)
-    return (free / (1024 * 1024)) > 500
+download_queue = asyncio.Queue()
 
+# ============ قاعدة البيانات ============
 async def init_db():
     async with aiosqlite.connect(DB_PATH) as db:
         await db.execute("PRAGMA journal_mode=WAL;")
@@ -80,7 +51,8 @@ async def init_db():
 
 def get_media_details(message):
     media = message.photo or message.video or message.document or message.audio or message.voice or message.video_note or message.animation
-    if not media: return None, None, 0
+    if not media:
+        return None, None, 0
     
     m_type = "doc"
     if message.photo: m_type = "photo"
@@ -94,12 +66,12 @@ def get_media_details(message):
     file_size = getattr(media, "file_size", 0)
     return unique_id, m_type, file_size
 
-# --- 5. محرك التنزيل المتوازي ---
+# ============ محرك التنزيل ============
 async def download_worker_task(worker_id):
     while True:
         message = await download_queue.get()
         try:
-            while not check_disk_space():
+            while shutil.disk_usage(DATA_DIR).free / (1024 * 1024) <= 500:
                 await asyncio.sleep(15)
 
             unique_id, m_type, file_size = get_media_details(message)
@@ -128,10 +100,11 @@ async def download_worker_task(worker_id):
                                 (message.chat.id, message.id, unique_id, group_id, m_type, final_caption, file_path, file_size, time.time())
                             )
                             await db.commit()
-                        print(f"⚡ [Worker-{worker_id}] تم تنزيل رسالة بنجاح: {message.id}")
+                        print(f"⚡ [Worker-{worker_id}] تم تنزيل: {message.id}")
                         break
                     else:
-                        if file_path and os.path.exists(file_path): os.remove(file_path)
+                        if file_path and os.path.exists(file_path):
+                            os.remove(file_path)
                 except Exception:
                     await asyncio.sleep(2)
         except Exception as e:
@@ -139,13 +112,24 @@ async def download_worker_task(worker_id):
         finally:
             download_queue.task_done()
 
-# --- 6. صائد الرسائل الحية ---
+# ============ صائد الرسائل ============
+app = Client(
+    "memory_session",
+    api_id=API_ID,
+    api_hash=API_HASH,
+    session_string=SESSION_STRING,
+    in_memory=True,
+    device_model="Samsung Galaxy S23 Ultra",
+    system_version="Android 14.0",
+    app_version="10.14.1",
+    lang_code="ar"
+)
+
 @app.on_message(filters.incoming & ~filters.me & (filters.channel | filters.group) & filters.media)
 async def catch_incoming(client, message):
     if message.chat.id != TARGET_CHAT:
         await download_queue.put(message)
 
-# --- 7. لوحة المراقبة ---
 @app.on_message(filters.me & filters.command("status"))
 async def status_command(client, message):
     async with aiosqlite.connect(DB_PATH) as db:
@@ -158,21 +142,23 @@ async def status_command(client, message):
     free_mb = round(free / (1024 * 1024), 2)
     
     report = (
-        f"👑 **حالة رادار الأرشفة المحترف:**\n\n"
+        f"👑 **حالة الأرشفة:**\n\n"
         f"📥 جاهز للرفع: `{pending}`\n"
-        f"📤 تم رفعه بأمان: `{uploaded}`\n"
+        f"📤 تم رفعه: `{uploaded}`\n"
         f"🔄 قيد التنزيل: `{download_queue.qsize()}`\n"
-        f"💾 المساحة المتاحة: `{free_mb} MB`"
+        f"💾 مساحة متاحة: `{free_mb} MB`"
     )
     await message.reply_text(report)
 
-# --- 8. محرك الرفع الذكي العالي الأمان ---
+# ============ محرك الرفع ============
 async def upload_worker():
     global BASE_UPLOAD_DELAY
     while True:
         try:
             async with aiosqlite.connect(DB_PATH) as db:
-                async with db.execute("SELECT id, chat_id, msg_id, media_group_id, media_type, caption, file_path, file_size, timestamp FROM queue WHERE status='DOWNLOADED' ORDER BY id ASC LIMIT 1") as cursor:
+                async with db.execute(
+                    "SELECT id, chat_id, msg_id, media_group_id, media_type, caption, file_path, file_size, timestamp FROM queue WHERE status='DOWNLOADED' ORDER BY id ASC LIMIT 1"
+                ) as cursor:
                     row = await cursor.fetchone()
 
             if not row:
@@ -188,16 +174,22 @@ async def upload_worker():
                 continue
 
             try:
-                if m_type == "photo": await app.send_chat_action(TARGET_CHAT, ChatAction.UPLOAD_PHOTO)
-                elif m_type in ["video", "animation"]: await app.send_chat_action(TARGET_CHAT, ChatAction.UPLOAD_VIDEO)
-                elif m_type == "doc": await app.send_chat_action(TARGET_CHAT, ChatAction.UPLOAD_DOCUMENT)
-                elif m_type in ["audio", "voice"]: await app.send_chat_action(TARGET_CHAT, ChatAction.UPLOAD_AUDIO)
+                if m_type == "photo":
+                    await app.send_chat_action(TARGET_CHAT, ChatAction.UPLOAD_PHOTO)
+                elif m_type in ["video", "animation"]:
+                    await app.send_chat_action(TARGET_CHAT, ChatAction.UPLOAD_VIDEO)
+                elif m_type == "doc":
+                    await app.send_chat_action(TARGET_CHAT, ChatAction.UPLOAD_DOCUMENT)
+                elif m_type in ["audio", "voice"]:
+                    await app.send_chat_action(TARGET_CHAT, ChatAction.UPLOAD_AUDIO)
                 await asyncio.sleep(random.uniform(1.0, 2.0))
-            except: pass
+            except:
+                pass
 
             if group_id:
                 time_passed = time.time() - ts
-                if time_passed < 10: await asyncio.sleep(10 - time_passed)
+                if time_passed < 10:
+                    await asyncio.sleep(10 - time_passed)
 
                 async with aiosqlite.connect(DB_PATH) as db:
                     async with db.execute(
@@ -215,10 +207,14 @@ async def upload_worker():
                             g_id, g_type, g_cap, g_path = g_item
                             item_caption = g_cap if idx == 0 else ""
 
-                            if g_type == "photo": media_list.append(InputMediaPhoto(media=g_path, caption=item_caption))
-                            elif g_type == "video": media_list.append(InputMediaVideo(media=g_path, caption=item_caption, supports_streaming=True))
-                            elif g_type == "doc": media_list.append(InputMediaDocument(media=g_path, caption=item_caption))
-                            elif g_type == "audio": media_list.append(InputMediaAudio(media=g_path, caption=item_caption))
+                            if g_type == "photo":
+                                media_list.append(InputMediaPhoto(media=g_path, caption=item_caption))
+                            elif g_type == "video":
+                                media_list.append(InputMediaVideo(media=g_path, caption=item_caption, supports_streaming=True))
+                            elif g_type == "doc":
+                                media_list.append(InputMediaDocument(media=g_path, caption=item_caption))
+                            elif g_type == "audio":
+                                media_list.append(InputMediaAudio(media=g_path, caption=item_caption))
 
                             ids_to_update.append(g_id)
                             files_to_delete.append(g_path)
@@ -231,25 +227,34 @@ async def upload_worker():
                                 await db.commit()
 
                             for f_path in files_to_delete:
-                                if os.path.exists(f_path): os.remove(f_path)
+                                if os.path.exists(f_path):
+                                    os.remove(f_path)
                             
                             gc.collect()
                             await asyncio.sleep(BASE_UPLOAD_DELAY + random.uniform(2.0, 4.0))
                     continue
 
-            if m_type == "photo": await app.send_photo(TARGET_CHAT, photo=file_path, caption=caption)
-            elif m_type == "video": await app.send_video(TARGET_CHAT, video=file_path, caption=caption, supports_streaming=True)
-            elif m_type == "doc": await app.send_document(TARGET_CHAT, document=file_path, caption=caption)
-            elif m_type == "audio": await app.send_audio(TARGET_CHAT, audio=file_path, caption=caption)
-            elif m_type == "voice": await app.send_voice(TARGET_CHAT, voice=file_path, caption=caption)
-            elif m_type == "video_note": await app.send_video_note(TARGET_CHAT, video_note=file_path)
-            elif m_type == "animation": await app.send_animation(TARGET_CHAT, animation=file_path, caption=caption)
+            if m_type == "photo":
+                await app.send_photo(TARGET_CHAT, photo=file_path, caption=caption)
+            elif m_type == "video":
+                await app.send_video(TARGET_CHAT, video=file_path, caption=caption, supports_streaming=True)
+            elif m_type == "doc":
+                await app.send_document(TARGET_CHAT, document=file_path, caption=caption)
+            elif m_type == "audio":
+                await app.send_audio(TARGET_CHAT, audio=file_path, caption=caption)
+            elif m_type == "voice":
+                await app.send_voice(TARGET_CHAT, voice=file_path, caption=caption)
+            elif m_type == "video_note":
+                await app.send_video_note(TARGET_CHAT, video_note=file_path)
+            elif m_type == "animation":
+                await app.send_animation(TARGET_CHAT, animation=file_path, caption=caption)
 
             async with aiosqlite.connect(DB_PATH) as db:
                 await db.execute("UPDATE queue SET status='UPLOADED' WHERE id=?", (q_id,))
                 await db.commit()
 
-            if os.path.exists(file_path): os.remove(file_path)
+            if os.path.exists(file_path):
+                os.remove(file_path)
             
             size_factor = min(file_size / (10 * 1024 * 1024), 10)
             safe_sleep = BASE_UPLOAD_DELAY + size_factor + random.uniform(1.0, 3.0)
@@ -258,52 +263,74 @@ async def upload_worker():
             await asyncio.sleep(safe_sleep)
 
         except FloodWait as e:
-            print(f"⚠️ تليجرام فرض حماية FloodWait لمدة {e.value} ثانية. جاري التهدئة...")
+            print(f"⚠️ FloodWait: {e.value} ثانية")
             BASE_UPLOAD_DELAY += 5
             await asyncio.sleep(e.value + random.uniform(10, 20))
         except Exception as e:
-            print(f"⚠️ خطأ غير متوقع، تجاوز الملف: {e}")
+            print(f"⚠️ خطأ: {e}")
             async with aiosqlite.connect(DB_PATH) as db:
                 await db.execute("UPDATE queue SET status='FAILED' WHERE id=?", (q_id,))
                 await db.commit()
-            if os.path.exists(file_path): os.remove(file_path)
+            if os.path.exists(file_path):
+                os.remove(file_path)
             await asyncio.sleep(5)
 
-async def healthcheck_ping():
-    while True:
-        await asyncio.sleep(300)
-        if HEALTHCHECK_URL:
-            try: urllib.request.urlopen(HEALTHCHECK_URL, timeout=10)
-            except: pass
-
+# ============ التشغيل الرئيسي ============
 async def main():
     await init_db()
     await app.start()
     
-    print("🔄 جاري مزامنة بصمة الهاتف مع مراكز بيانات تليجرام...")
+    print("🔄 جاري الاتصال...")
     try:
         me = await app.get_me()
-        print(f"✅ تم الاتصال بنجاح بالحساب: {me.first_name} (معرف الجلسة مستقر)")
+        print(f"✅ متصل: {me.first_name}")
         async for _ in app.get_dialogs(limit=150):
             pass
     except Exception as e:
-        print(f"⚠️ تحذير الاتصال: {e}")
+        print(f"⚠️ تحذير: {e}")
 
-    print("🚀 تم تشغيل الرادار الحي بنسخة الأمان القصوى والمحاكاة التامة!")
+    print("🚀 البوت يعمل!")
     
     for i in range(2):
         asyncio.create_task(download_worker_task(i + 1))
         
     asyncio.create_task(upload_worker())
-    asyncio.create_task(healthcheck_ping())
     
     await idle()
     await app.stop()
 
-if __name__ == "__main__":
+def run_bot():
+    """تشغيل البوت في thread منفصلة"""
+    loop = asyncio.new_event_loop()
+    asyncio.set_event_loop(loop)
     while True:
         try:
-            app.run(main())
-        except Exception as crash_error:
-            print(f"⚠️ إعادة تشغيل ذكية: {crash_error}")
-            time.sleep(10)
+            loop.run_until_complete(main())
+        except Exception as e:
+            print(f"💥 إعادة تشغيل: {e}")
+            time.sleep(10)
+
+# ============ Gradio Interface ============
+bot_status = {"running": False, "logs": "⏳ جاري التشغيل..."}
+
+def start_bot():
+    if not bot_status["running"]:
+        bot_status["running"] = True
+        bot_status["logs"] = "🟢 البوت يعمل في الخلفية"
+        thread = threading.Thread(target=run_bot, daemon=True)
+        thread.start()
+        return "✅ تم تشغيل البوت!"
+    return "⚠️ البوت يعمل بالفعل!"
+
+# تشغيل البوت تلقائياً
+start_bot()
+
+with gr.Blocks(title="Telegram Archiver") as demo:
+    gr.Markdown("# 📡 Telegram Archiver Bot")
+    gr.Markdown("البوت يعمل في الخلفية ويقوم بأرشفة الملفات تلقائياً")
+    
+    status_box = gr.Textbox(label="الحالة", value="🟢 البوت يعمل", interactive=False)
+    gr.Markdown("### المتغيرات المطلوبة في Settings:")
+    gr.Markdown("- `API_ID`\n- `API_HASH`\n- `SESSION_STRING`\n- `TARGET_CHAT`")
+
+demo.launch(server_name="0.0.0.0", server_port=7860)