Update app.py

app.py

@@ -12,21 +12,23 @@ from flask import Flask
 # ==========================================
 # 🛑 APNI DETAILS YAHAN DAALEIN 🛑
 # ==========================================
+
 API_ID = "35985614"
 API_HASH = "6a0df17414daf6935f1f0a71b8af1ee9"
 STRING_SESSION = "BQIlGM4AkncApFEGc9JNMRHYJCqi-IbbXCuSEw7yrNIcloArm-xwbEM8ksoMYQEK4_uD1NEgro-CIDXnc6yEuaPkm91FGhHVpHJGoUnvv19UeilrIhopSY8dboZ0emUF15IhIgZD6Qqp5Nv2gJobdCrZQuitAS6Xelmm14rn-3kWynpUo3vD4GpeRWh1b8ydh8Rlqi1ALoWOHC5uGyK_zJBAnxDUea2FJUvgX2Q93TuejnVuj4o84M_DqMtHowvf5EGkE4HSZfjPW6c5Qb2VkFtRwFmt1-6hZGoABJjvrJpZQmc94LLApSd45Hme-YXpPrxGWck4mFF0t8SRYrPGMxyyVJRgAAAAHK_kURAA"
-TARGET_CHAT = "-1002545525249"
+# YAHAN INT() LAGANA ZAROORI HAI:
+TARGET_CHAT = int("-1002545525249") 
 
 BOT_TOKEN = "8830608789:AAF7J6I-1H1mzRkpccgCOuTh2wR_yOSOMNA"
 
-# ✅ YAHAN CHANGE KIYA HAI (Taki error na aaye)
-HF_TOKEN = os.environ.get("HF_TOKEN") 
+HF_TOKEN = os.environ.get("HF_TOKEN")
 HF_REPO_ID = "Jitendra55566/pdf-notes"
 FIREBASE_URL = "https://logo-ncc-default-rtdb.asia-southeast1.firebasedatabase.app"
 
 # ==========================================
 # 1. DAILY STATS TRACKER
 # ==========================================
+
 def get_ist_date():
     ist_time = datetime.utcnow() + timedelta(hours=5, minutes=30)
     return ist_time.strftime("%d-%m-%Y")
@@ -47,8 +49,9 @@ def add_to_stats(subject, chapter, pdf_name):
     daily_stats["details"].append(f"✅ {subject} ➔ {chapter} ({pdf_name})")
 
 # ==========================================
-# 2. FLASK SERVER
+# 2. FLASK SERVER (Keep Awake)
 # ==========================================
+
 web_app = Flask(__name__)
 
 @web_app.route('/')
@@ -61,6 +64,7 @@ def run_flask():
 # ==========================================
 # 3. TELEGRAM CLIENTS SETUP
 # ==========================================
+
 app = Client("user_account", session_string=STRING_SESSION, api_id=API_ID, api_hash=API_HASH)
 bot_app = Client("report_bot", bot_token=BOT_TOKEN, api_id=API_ID, api_hash=API_HASH)
 hf_api = HfApi()
@@ -68,6 +72,7 @@ hf_api = HfApi()
 # ==========================================
 # 4. REPORT BOT HANDLERS (@BotFather bot)
 # ==========================================
+
 @bot_app.on_message(filters.command("start"))
 async def start_cmd(client, message):
     text = "👋 Hello Sir!\n\nMain auto-upload bot hu.\n📈 Aaj ki Report dekhne ke liye 👉 /today"
@@ -80,7 +85,7 @@ async def today_cmd(client, message):
         daily_stats["date"] = today
         daily_stats["count"] = 0
         daily_stats["details"] = []
-        
+
     if daily_stats["count"] == 0:
         await message.reply_text(f"📅 Date: {today}\n\n😴 Aaj abhi tak koi PDF upload nahi hui.")
     else:
@@ -91,6 +96,7 @@ async def today_cmd(client, message):
 # ==========================================
 # 5. USER ACCOUNT LOGIC (PDF DOWNLOAD & UPLOAD)
 # ==========================================
+
 def upload_to_huggingface(file_path, file_name):
     hf_api.upload_file(
         path_or_fileobj=file_path,
@@ -111,14 +117,15 @@ def find_or_create_node(base_url, title, create_data=None):
     res = requests.post(f"{base_url}.json", json=create_data)
     return res.json()['name']
 
+# 🔴 FIX: Added "async" here
 @app.on_message(filters.chat(TARGET_CHAT) & filters.document)
-def handle_new_pdf(client, message):
+async def handle_new_pdf(client, message):
     text = message.caption if message.caption else ""
     if not text:
         return
 
     try:
-        # 1. EXTRACT DATA FROM TELEGRAM TEXT
+        # EXTRACT DATA
         file_name_match = re.search(r"File Name\s*:\s*(.*?)\.pdf", text, re.IGNORECASE)
         file_name = file_name_match.group(1).strip() if file_name_match else "Unknown"
 
@@ -131,62 +138,51 @@ def handle_new_pdf(client, message):
         if not batch_name or not subject_name:
             return
 
-        # Chapter (ex: वर्तनी) extraction
         chapter_match = re.search(r"\((.*?)\)(?!\s*वीर बैच)", file_name)
         chapter_name = chapter_match.group(1).strip() if chapter_match else "General"
         
-        # Clean Video Name extraction
         parts = file_name.split(')')
         class_str = parts[-1] if len(parts) > 1 else file_name
-        class_str = re.sub(r'(?i)PDF.*', '', class_str).strip() # Remove "PDF [Date]..."
-        video_name = re.sub(r'\s*\d+$', '', class_str).strip() # Remove "1" or "2" from end
+        class_str = re.sub(r'(?i)PDF.*', '', class_str).strip()
+        video_name = re.sub(r'\s*\d+$', '', class_str).strip()
 
-        # ==========================================
-        # 2. FIREBASE PANEL CONNECTION
-        # ==========================================
+        # FIREBASE (We keep it synchronous for simplicity, but it's fast)
         cat_id = find_or_create_node(f"{FIREBASE_URL}/categories", batch_name, {"title": batch_name})
         sub_id = find_or_create_node(f"{FIREBASE_URL}/categories/{cat_id}/subcategories", subject_name, {"title": subject_name})
         chap_id = find_or_create_node(f"{FIREBASE_URL}/categories/{cat_id}/subcategories/{sub_id}/subcategories", chapter_name, {"title": chapter_name})
         
-        # VIDEO (Class) banega YAHAN DUMMY LINK HAI
         vid_id = find_or_create_node(f"{FIREBASE_URL}/categories/{cat_id}/subcategories/{sub_id}/subcategories/{chap_id}/videos", video_name, {
             "title": video_name, 
-            "url": "https://dummy-video-url.com", # <--- DUMMY LINK YAHAN HAI
+            "url": "https://dummy-video-url.com",
             "date": int(time.time() * 1000)
         })
 
-        # ==========================================
-        # 3. PDF DUPLICATE CHECK & AUTO-NAMING (PDF 1 / PDF 2)
-        # ==========================================
         pdf_folder_url = f"{FIREBASE_URL}/categories/{cat_id}/subcategories/{sub_id}/subcategories/{chap_id}/videos/{vid_id}/pdfs"
         existing_pdfs = requests.get(f"{pdf_folder_url}.json").json() or {}
         
         pdf_count = 0
         for pid, pdata in existing_pdfs.items():
             pdf_count += 1
-            # Check duplicate based on original file name
             if pdata.get('original_name') == file_name:
                 print(f"Skipping! '{file_name}' pehle se maujud hai.")
                 return
 
-        # Naya PDF ka Button Name set karna (PDF 1, PDF 2...)
         pdf_button_name = f"PDF {pdf_count + 1}"
 
-        # ==========================================
-        # 4. DOWNLOAD & UPLOAD TO HUGGING FACE
-        # ==========================================
         print(f"Downloading: {file_name}.pdf")
-        local_path = message.download()
-        hf_pdf_url = upload_to_huggingface(local_path, f"{int(time.time())}_{file_name}.pdf")
+        
+        # 🔴 FIX: Added 'await' before message.download()
+        local_path = await message.download()
+        
+        # 🔴 FIX: Run HuggingFace upload in background thread so Bot doesn't freeze!
+        hf_pdf_url = await asyncio.to_thread(upload_to_huggingface, local_path, f"{int(time.time())}_{file_name}.pdf")
+        
         os.remove(local_path)
         
-        # ==========================================
-        # 5. UPLOAD TO FIREBASE PANEL
-        # ==========================================
         pdf_data = {
-            "title": pdf_button_name, # App me "PDF 1" / "PDF 2" dikhega
+            "title": pdf_button_name,
             "url": hf_pdf_url,
-            "original_name": file_name, # Bot ko yaad rakhne ke liye ki ye upload ho chuka h
+            "original_name": file_name,
             "show_blocker": True
         }
         requests.post(f"{pdf_folder_url}.json", json=pdf_data)
@@ -200,9 +196,11 @@ def handle_new_pdf(client, message):
 # ==========================================
 # 6. RUN ALL
 # ==========================================
+
+# 🔴 FIX: Corrected __main__ condition syntax
 if __name__ == "__main__":
     threading.Thread(target=run_flask).start()
-    
+
     async def main():
         print("🚀 Systems Online: Pyrogram + Report Bot + Auto-Uploader")
         await compose([app, bot_app])