Update app.py

app.py

@@ -4,11 +4,10 @@ import pandas as pd
 import os
 from huggingface_hub import hf_hub_download
 
-# Fast download enable karne ke liye
+# Fast download enable
 os.environ["HF_HUB_ENABLE_HF_TRANSFER"] = "1"
 
 # --- CONFIGURATION ---
-# Token ko Space Settings > Secrets mein HF_TOKEN naam se save karna
 TOKEN = os.getenv("HF_TOKEN") 
 REPO = "Watchhrr/Telegram"
 DB1_NAME = "Telegram_Master_DB.db"
@@ -17,11 +16,11 @@ DB2_NAME = "Telegram_Part2.db"
 # --- DOWNLOAD DATABASES ---
 def load_data():
     if not TOKEN:
+        print("❌ HF_TOKEN not found in Secrets!")
         return None, None
         
     print("⏳ Downloading Databases (Large Files)...")
     try:
-        # Dono DB files ko download karke local path lena
         path1 = hf_hub_download(repo_id=REPO, filename=DB1_NAME, repo_type="dataset", token=TOKEN)
         path2 = hf_hub_download(repo_id=REPO, filename=DB2_NAME, repo_type="dataset", token=TOKEN)
         return path1, path2
@@ -29,7 +28,6 @@ def load_data():
         print(f"Download Error: {e}")
         return None, None
 
-# Initialize paths at startup
 db1_path, db2_path = load_data()
 
 def hitech_search(search_query):
@@ -40,15 +38,11 @@ def hitech_search(search_query):
         return "⚠️ Please enter at least 3 characters."
     
     try:
-        # SQLite Connection
         conn = sqlite3.connect(db1_path)
-        # Dusre DB ko attach karna taaki ek sath search ho sake
         conn.execute(f"ATTACH DATABASE '{db2_path}' AS db2")
         
-        # Phone search ke liye sirf numbers extract karna
         clean_phone = ''.join(filter(str.isdigit, str(search_query)))
         
-        # Optimized SQL Query with UNION
         sql = """
         SELECT 'Master' as source, user_id, phone, username, first_name, last_name FROM users 
         WHERE phone = ? OR user_id = ? OR username = ?
@@ -70,9 +64,9 @@ def hitech_search(search_query):
         return f"❌ Search Error: {str(e)}"
 
 # --- GRADIO UI ---
-with gr.Blocks(theme=gr.themes.Soft()) as demo:
+with gr.Blocks() as app: # Yahan 'demo' ki jagah 'app' kar diya
     gr.Markdown("# 🚀 Hitech Telegram Search Engine")
-    gr.Markdown("Searching across **251 Million+** records in real-time.")
+    gr.Markdown("Searching across **251 Million+** records.")
     
     with gr.Row():
         input_text = gr.Textbox(
@@ -84,9 +78,9 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
     search_btn = gr.Button("🔍 Deep Search", variant="primary")
     output_table = gr.DataFrame(label="Results")
     
-    # Click aur Enter dono pe kaam karega
     search_btn.click(fn=hitech_search, inputs=input_text, outputs=output_table)
     input_text.submit(fn=hitech_search, inputs=input_text, outputs=output_table)
 
+# Gradio 6.0 fix: theme yahan launch mein aayega
 if __name__ == "__main__":
-    demo.launch()
+    app.launch(theme=gr.themes.Soft())