Update app.py

app.py

@@ -4,36 +4,48 @@ import os
 
 app = FastAPI()
 
-# Database ka path (Hugging Face Dataset se connect karne ke liye)
+# Database ka path (Hugging Face Dataset se link hai ya upload hai)
 # Agar aapne database isi Space mein upload kiya hai toh:
 DB_PATH = "Telegram_Master_DB.db" 
 
+@app.get("/")
+def home():
+    return {"message": "Hitech DB API is Running!", "usage": "/search?num=PHONE or /search?id=ID"}
+
 @app.get("/search")
 def search(num: str = None, id: str = None):
     if not os.path.exists(DB_PATH):
-        return {"error": "Database file nahi mili bhai!"}
-    
-    conn = sqlite3.connect(DB_PATH)
-    conn.row_factory = sqlite3.Row
-    cursor = conn.cursor()
+        return {"error": "Bhai, Database file 'Telegram_Master_DB.db' nahi mili!"}
     
-    result = None
-    if num:
-        cursor.execute("SELECT * FROM users WHERE phone = ?", (num,))
-        result = cursor.fetchone()
-    elif id:
-        cursor.execute("SELECT * FROM users WHERE user_id = ?", (id,))
-        result = cursor.fetchone()
+    try:
+        conn = sqlite3.connect(DB_PATH)
+        conn.row_factory = sqlite3.Row
+        cursor = conn.cursor()
         
-    conn.close()
-    
-    if result:
-        return {
-            "status": "success",
-            "data": dict(result)
-        }
-    return {"status": "error", "message": "Data nahi mila!"}
+        result = None
+        if num:
+            # Phone search
+            cursor.execute("SELECT * FROM users WHERE phone = ?", (num,))
+            result = cursor.fetchone()
+        elif id:
+            # ID search
+            cursor.execute("SELECT * FROM users WHERE user_id = ?", (id,))
+            result = cursor.fetchone()
+            
+        conn.close()
+        
+        if result:
+            return {
+                "status": "success",
+                "data": dict(result)
+            }
+        else:
+            return {"status": "error", "message": "Data nahi mila!"}
+            
+    except Exception as e:
+        return {"status": "error", "message": str(e)}
 
 if __name__ == "__main__":
     import uvicorn
+    # Hugging Face Spaces 7860 port mangta hai
     uvicorn.run(app, host="0.0.0.0", port=7860)