Update app.py

app.py

@@ -4,28 +4,25 @@ import uvicorn
 from fastapi import FastAPI
 from huggingface_hub import hf_hub_download
 
-app = FastAPI(title="Midnight Syndicate OSINT API")
-
-# Dataset details
-DATASET_ID = "Watchhrr/Telegram"
+app = FastAPI()
 
 @app.get("/")
-def home():
-    return {"message": "API is Running! Use /search/ID to find data."}
+def read_root():
+    return {"status": "Midnight API Online"}
 
 @app.get("/search/{user_id}")
 def search(user_id: str):
     token = os.getenv("HF_TOKEN")
     results = []
     
-    # Dono Databases
+    # Dono DBs ki list
     dbs = ["Midnight_Master_v2.db", "Telegram_part_1_DB.db"]
 
     for db_name in dbs:
         try:
-            # Download/Stream from Dataset
+            # Dataset se file link uthana
             path = hf_hub_download(
-                repo_id=DATASET_ID,
+                repo_id="Watchhrr/Telegram",
                 filename=db_name,
                 repo_type="dataset",
                 token=token
@@ -34,8 +31,8 @@ def search(user_id: str):
             conn = sqlite3.connect(path)
             cursor = conn.cursor()
             
+            # DB Type ke hisab se query
             if "Midnight" in db_name:
-                # 27Cr DB search
                 cursor.execute("SELECT user_id FROM telegram_users WHERE user_id LIKE ?", (f"{user_id}%",))
                 for r in cursor.fetchall():
                     parts = r[0].split(';')
@@ -46,15 +43,10 @@ def search(user_id: str):
                             "name": parts[2] if len(parts)>2 else "N/A"
                         })
             else:
-                # 5Cr DB search
                 cursor.execute("SELECT * FROM users WHERE user_id = ?", (user_id,))
                 for r in cursor.fetchall():
-                    results.append({
-                        "db": "Part_1",
-                        "phone": r[1],
-                        "username": r[2],
-                        "name": f"{r[3]} {r[4] or ''}".strip()
-                    })
+                    results.append({"db": "Part_1", "phone": r[1], "username": r[2], "name": f"{r[3]} {r[4] or ''}".strip()})
+            
             conn.close()
         except Exception as e:
             print(f"Error: {e}")
@@ -62,6 +54,5 @@ def search(user_id: str):
     return {"status": "success", "user_id": user_id, "data": results}
 
 if __name__ == "__main__":
-    # Hugging Face default port 7860 use karna zaroori hai
     uvicorn.run(app, host="0.0.0.0", port=7860)