Update main.py

main.py

@@ -10,11 +10,14 @@ from contextlib import asynccontextmanager
 # --- CONFIG ---
 API_ID = 21934109
 API_HASH = 'e7e8c554b9ff88d180983996c33bdf27'
-BOT_USERNAME = '@OSINTXERA_bOt'
+
+# Extracted Peer ID from your web link (-1003794439741)
+# Telethon requires the -100 prefix dropped for entity lookups via ID integer
+TARGET_GROUP_ID = -1003794439741 
 
 # Dynamically bind to Hugging Face's required routing port
 PORT = int(os.environ.get("PORT", 7860))
-state = {"my_id": None, "bot_entity": None}
+state = {"my_id": None, "group_entity": None}
 
 # Pointing explicitly to the absolute path in the HF container
 client = TelegramClient('/code/num_lookup_session', API_ID, API_HASH)
@@ -83,11 +86,12 @@ async def lifespan(app: FastAPI):
         state["my_id"] = me.id
         
         try:
-            print(f"[STARTUP] Caching entity profile for: {BOT_USERNAME}")
-            state["bot_entity"] = await client.get_input_entity(BOT_USERNAME)
-            print("[STARTUP] Bot entity caching completed successfully.")
+            print(f"[STARTUP] Fetching and caching entity for Group ID: {TARGET_GROUP_ID}")
+            # Group must be in the account's dialog history for this to find it instantly
+            state["group_entity"] = await client.get_input_entity(TARGET_GROUP_ID)
+            print("[STARTUP] Group entity caching completed successfully.")
         except Exception as e:
-            print(f"[CRITICAL STARTUP ERROR] Could not cache bot: {e}")
+            print(f"[CRITICAL STARTUP ERROR] Could not cache Group Entity: {e}")
         
     yield
     print("[SHUTDOWN] Disconnecting Telegram Client...")
@@ -97,7 +101,7 @@ app = FastAPI(lifespan=lifespan)
 
 app.add_middleware(
     CORSMiddleware,
-    allow_origins=["https://crm.gudmed.in"],  # Secured for your production frontend CRM environment
+    allow_origins=["https://crm.gudmed.in"],
     allow_credentials=True,
     allow_methods=["*"],
     allow_headers=["*"],
@@ -105,47 +109,52 @@ app.add_middleware(
 
 @app.get("/")
 async def root_index():
-    return {"status": "healthy", "service": "Telegram Truecaller Polling API"}
+    return {"status": "healthy", "service": "Telegram Truecaller Group Polling API"}
 
 @app.get("/lookup")
 async def lookup(phone: str):
     clean_phone = str(phone)[-10:].strip()
     await ensure_connected()
     
-    if not state["my_id"] or not state["bot_entity"]:
+    if not state["my_id"] or not state["group_entity"]:
         raise HTTPException(status_code=503, detail="Telegram client components are uninitialized.")
     
     try:
+        # Determine the latest message ID in the group before tracking
         try:
-            last_messages = await client.get_messages(state["bot_entity"], limit=1)
+            last_messages = await client.get_messages(state["group_entity"], limit=1)
             start_msg_id = last_messages[0].id if last_messages else 0
         except Exception:
             await ensure_connected()
-            last_messages = await client.get_messages(state["bot_entity"], limit=1)
+            last_messages = await client.get_messages(state["group_entity"], limit=1)
             start_msg_id = last_messages[0].id if last_messages else 0
 
         command_payload = f"/num {clean_phone}"
-        print(f"[API] Dispatching search request: {command_payload}")
-        await client.send_message(state["bot_entity"], command_payload)
+        print(f"[API] Dispatching search request to Group: {command_payload}")
+        await client.send_message(state["group_entity"], command_payload)
         
-        for i in range(40):
-            await asyncio.sleep(0.4) 
+        # Poll up to 50 iterations for group environments due to latency differences
+        for i in range(50):
+            await asyncio.sleep(0.5) 
             
             try:
-                messages = await client.get_messages(state["bot_entity"], limit=2)
+                # Fetch more messages (limit=5) because group chatter can pass quickly
+                messages = await client.get_messages(state["group_entity"], limit=5)
             except Exception as loop_err:
                 print(f"[API WARNING] Polling read failed on step {i}: {loop_err}. Reconnecting...")
                 await ensure_connected()
                 continue
             
             for m in messages:
+                # Target messages that appeared AFTER our request, and are NOT sent by us
                 if m.id > start_msg_id and m.sender_id != state["my_id"]:
                     if not m.text or any(x in m.text for x in ["Fetching", "Please wait", "Searching", "Typing", "/num"]):
                         continue
                     
+                    # Validate that this payload matches our tracked query sequence
                     query_match = re.search(r'Query:\s*`?(\d+)`?', m.text, re.IGNORECASE)
                     if query_match and query_match.group(1).endswith(clean_phone):
-                        print(f"[API] Targeted response packet found on iteration step {i}!")
+                        print(f"[API] Targeted response packet found in group on iteration step {i}!")
                         total_records_match = re.search(r'Total Records:\s*(\d+)', m.text, re.IGNORECASE)
                         total_count = int(total_records_match.group(1)) if total_records_match else 0
                         
@@ -157,7 +166,7 @@ async def lookup(phone: str):
                             "full_text": m.text
                         }
 
-        raise HTTPException(status_code=504, detail="Timeout: Bot didn't respond with matched query number in time.")
+        raise HTTPException(status_code=504, detail="Timeout: Target bot in group didn't answer with query verification.")
         
     except HTTPException:
         raise
@@ -166,5 +175,4 @@ async def lookup(phone: str):
         raise HTTPException(status_code=500, detail=str(e))
 
 if __name__ == "__main__":
-    # Point directly to "main:app" for proper cloud environment worker execution
     uvicorn.run("main:app", host="0.0.0.0", port=PORT, loop="asyncio")