Update app.py

app.py

@@ -14,12 +14,23 @@ from itertools import permutations, combinations
 from mnemonic import Mnemonic
 import coincurve
 from eth_utils import keccak, to_checksum_address
-import spaces
 
-@spaces.GPU
+# ============================================================
+# Spaces GPU Fallback
+# ============================================================
+try:
+    import spaces
+    gpu_decorator = spaces.GPU
+except ImportError:
+    def gpu_decorator(func):
+        return func
+
+@gpu_decorator
 def _noop():
     return "ok"
 
+# ============================================================
+# Backend Logic (Unchanged)
 # ============================================================
 _MNEMONIC = None
 def _get_mnemonic():
@@ -28,7 +39,6 @@ def _get_mnemonic():
         _MNEMONIC = Mnemonic("english")
     return _MNEMONIC
 
-# ---------- BIP32 manual ----------
 _SECP256K1_N = 0xFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFEBAAEDCE6AF48A03BBFD25E8CD0364141
 _HARDENED    = 0x80000000
 
@@ -51,7 +61,6 @@ def seed_to_privkey_hd(seed, address_index=0):
         key, chain = _bip32_derive_child_priv(key, chain, idx)
     return key
 
-# ---------- BIP39 ----------
 try:
     import bip39 as _bip39_rust
     _USE_RUST_BIP39 = True
@@ -76,9 +85,7 @@ def phrase_to_seed(phrase, passphrase=""):
         return _seed_fn(phrase, passphrase)
     return _get_mnemonic().to_seed(phrase, passphrase=passphrase)
 
-# ---------- Address Cache ----------
 _ADDRESS_CACHE = {}
-
 def load_addresses_cached(path):
     if path in _ADDRESS_CACHE:
         cached_mtime, cached_set = _ADDRESS_CACHE[path]
@@ -88,61 +95,51 @@ def load_addresses_cached(path):
                 return cached_set
         except OSError:
             pass
-
     s = set()
     with open(path, "r", encoding="utf-8") as f:
         for line in f:
             a = line.strip().strip('"').strip("'").lstrip("\ufeff").lower()
-            if a.startswith("0x"):
-                a = a[2:]
-            if a and len(a) == 40:
-                s.add(a)
-
-    try:
-        mtime = os.path.getmtime(path)
-    except OSError:
-        mtime = 0
+            if a.startswith("0x"): a = a[2:]
+            if a and len(a) == 40: s.add(a)
+    try: mtime = os.path.getmtime(path)
+    except OSError: mtime = 0
     _ADDRESS_CACHE[path] = (mtime, s)
     return s
 
 def normalize(addr):
     a = addr.strip().lower()
-    if a.startswith("0x"):
-        a = a[2:]
+    if a.startswith("0x"): a = a[2:]
     return a
 
-# ---------- Generator ----------
 def generate_phrases_for_words(wordlist, mode):
     n = len(wordlist)
     if mode == "order":
-        for perm in permutations(wordlist):
-            yield " ".join(perm)
+        for perm in permutations(wordlist): yield " ".join(perm)
     elif mode == "replace":
         for i in range(n):
             for w in wordlist:
-                new = wordlist.copy()
-                new[i] = w
-                yield " ".join(new)
+                new = wordlist.copy(); new[i] = w; yield " ".join(new)
     elif mode == "subset":
         for r in range(1, n + 1):
-            for combo in combinations(wordlist, r):
-                yield " ".join(combo)
+            for combo in combinations(wordlist, r): yield " ".join(combo)
 
 # ============================================================
 # Scan Function
 # ============================================================
-
+@gpu_decorator
 def run_scan(my_phrase, num_phrases, addresses_file, phrase_mode,
              start_index, end_index, passphrase, progress=gr.Progress()):
     
+    num_phrases = int(num_phrases or 0)
+    start_index = int(start_index or 0)
+    end_index = int(end_index or 0)
+    my_phrase = my_phrase or ""
+    passphrase = passphrase or ""
+    
     if not os.path.exists(addresses_file):
         return f"❌ File nahi mili: {addresses_file}", ""
 
-    num_phrases = int(num_phrases)
-    start_index = int(start_index)
-    end_index = int(end_index)
-
-    progress(0, desc="Loading addresses...")
+    progress(0, desc="Initializing scanner...")
     try:
         known = load_addresses_cached(addresses_file)
     except Exception as e:
@@ -163,85 +160,217 @@ def run_scan(my_phrase, num_phrases, addresses_file, phrase_mode,
 
     log_text = f"✅ Loaded {len(known):,} addresses\n"
     log_text += f"🔧 Mode: {phrase_mode} | 🔑 Index: {start_index}-{end_index}\n"
-    log_text += f"📝 Total phrases: {len(phrases)}\n\n"
+    log_text += f"📝 Total phrases to check: {len(phrases)}\n\n"
     
-    log_text += "Phrases:\n"
+    log_text += "📜 Phrases:\n"
     for i, p in enumerate(phrases, 1):
-        tag = " (AAPKI)" if i == 1 and len(my_words) == 12 else ""
-        log_text += f"  {i}. {p}{tag}\n"
-    log_text += "\n--- Scanning ---\n"
+        tag = " (YOUR PHRASE)" if i == 1 and len(my_words) == 12 else ""
+        log_text += f"  [{i}] {p}{tag}\n"
+    log_text += "\n" + "="*40 + "\n🚀 STARTING SCAN...\n" + "="*40 + "\n"
 
     matches = []
     start = time.time()
     total_phrases = len(phrases)
 
     for p_idx, phrase in enumerate(phrases, 1):
-        progress((p_idx - 1) / total_phrases, desc=f"Phrase {p_idx}/{total_phrases}...")
-        log_text += f"\n🔍 [{p_idx}/{total_phrases}] Checking: {phrase}\n"
+        progress((p_idx - 1) / total_phrases, desc=f"Scanning Phrase {p_idx}/{total_phrases}...")
+        log_text += f"\n🔍 [{p_idx}/{total_phrases}] {phrase}\n"
         words = phrase.strip().split()
         
         if len(words) != 12:
-            log_text += "   ⚠️ Skipped (not 12 words)\n"
+            log_text += "   ⚠️ Skipped (invalid length)\n"
             continue
 
         for idx in range(start_index, end_index + 1):
             for phrase_str in generate_phrases_for_words(words, phrase_mode):
-                if not is_valid_bip39(phrase_str):
-                    continue
+                if not is_valid_bip39(phrase_str): continue
                 try:
                     seed = phrase_to_seed(phrase_str, passphrase)
                     priv = seed_to_privkey_hd(seed, idx)
                     pub = coincurve.PrivateKey(priv).public_key.format(compressed=False)[1:]
                     addr = to_checksum_address(keccak(pub)[-20:])
-                except Exception:
-                    continue
+                except Exception: continue
                 
                 if normalize(addr) in known:
-                    match_line = f"🎉 MATCH!\nPhrase: {phrase_str}\nIndex: {idx}\nKey: {priv.hex()}\nAddr: {addr}\n"
-                    log_text += match_line
+                    match_line = f"🎉 MATCH FOUND!\nPhrase: {phrase_str}\nIndex: {idx}\nKey: {priv.hex()}\nAddr: {addr}\n"
+                    log_text += f"   >>> {match_line}"
                     matches.append(match_line)
 
     elapsed = time.time() - start
-    log_text += f"\n=== DONE ===\nTotal time: {elapsed:.1f}s\nTotal matches: {len(matches)}\n"
-
-    matches_text = "\n".join(matches) if matches else "Koi match nahi mila"
+    log_text += f"\n{'='*40}\n✅ SCAN COMPLETE\nTime: {elapsed:.1f}s | Matches: {len(matches)}\n{'='*40}"
+    matches_text = "\n".join(matches) if matches else "No matches found in this run."
 
     return log_text, matches_text
 
 # ============================================================
-# Bulletproof UI
+# 🎨 PREMIUM UI DESIGN
 # ============================================================
-with gr.Blocks() as demo:
-    gr.Markdown("# 🔍 Phrase Scanner")
+
+custom_css = """
+/* Container & Typography */
+.gradio-container {
+    max-width: 1100px !important;
+    margin: auto !important;
+    font-family: 'Inter', 'Segoe UI', system-ui, sans-serif !important;
+}
+h1 {
+    text-align: center;
+    background: linear-gradient(90deg, #10b981, #3b82f6);
+    -webkit-background-clip: text;
+    -webkit-text-fill-color: transparent;
+    font-weight: 800 !important;
+    margin-bottom: 5px !important;
+}
+h3.sub-header {
+    color: #64748b !important;
+    text-align: center;
+    margin-top: 0 !important;
+    margin-bottom: 30px !important;
+    font-weight: 400 !important;
+}
+h4.section-title {
+    border-bottom: 2px solid #1e293b;
+    padding-bottom: 8px;
+    margin-bottom: 15px !important;
+    color: #e2e8f0 !important;
+    font-weight: 600 !important;
+}
+
+/* Terminal / Log Boxes */
+.terminal-log textarea, .match-log textarea {
+    font-family: 'JetBrains Mono', 'Fira Code', 'Courier New', monospace !important;
+    font-size: 13px !important;
+    border-radius: 10px !important;
+    padding: 15px !important;
+    line-height: 1.6 !important;
+    border: 1px solid #1e293b !important;
+}
+.terminal-log textarea {
+    background-color: #0f172a !important;
+    color: #cbd5e1 !important;
+    box-shadow: inset 0 0 20px rgba(0,0,0,0.5) !important;
+}
+.match-log textarea {
+    background-color: #022c22 !important;
+    color: #6ee7b7 !important;
+    border: 1px solid #065f46 !important;
+    font-weight: bold !important;
+    font-size: 14px !important;
+}
+
+/* Buttons */
+.primary-btn button {
+    background: linear-gradient(90deg, #10b981, #059669) !important;
+    border: none !important;
+    font-weight: 700 !important;
+    font-size: 16px !important;
+    padding: 14px 24px !important;
+    border-radius: 10px !important;
+    box-shadow: 0 4px 15px rgba(16, 185, 129, 0.3) !important;
+    transition: all 0.3s ease !important;
+    letter-spacing: 0.5px !important;
+}
+.primary-btn button:hover {
+    transform: translateY(-2px) !important;
+    box-shadow: 0 8px 20px rgba(16, 185, 129, 0.5) !important;
+}
+
+/* Hide default Gradio footer */
+footer { display: none !important; }
+
+/* Tabs styling */
+.tabs > .tab-nav > button {
+    font-weight: 600 !important;
+    font-size: 15px !important;
+}
+"""
+
+theme = gr.themes.Soft(
+    primary_hue="emerald",
+    secondary_hue="blue",
+    neutral_hue="slate",
+).set(
+    body_background_fill_dark="#020617",
+    block_background_fill_dark="#0f172a",
+    block_border_color_dark="#1e293b",
+    input_border_color_dark="#334155",
+)
+
+with gr.Blocks(theme=theme, css=custom_css, title="Crypto Phrase Scanner") as demo:
     
+    gr.Markdown("# 🚀 Crypto Phrase Scanner")
+    gr.Markdown("### Advanced BIP39 / BIP32 Address Recovery & Scanning Tool", elem_classes=["sub-header"])
+
+    with gr.Row():
+        # Left Column: Target Settings
+        with gr.Column(scale=2):
+            gr.Markdown("#### 🎯 Target Configuration", elem_classes=["section-title"])
+            addresses_file = gr.Textbox(
+                label="Target Addresses File (CSV/TXT)", 
+                value="12730802_addresses.csv",
+                info="Path to the file containing known Ethereum addresses."
+            )
+            with gr.Row():
+                start_index = gr.Number(label="Start Derivation Index", value=0, precision=0, minimum=0, scale=1)
+                end_index = gr.Number(label="End Derivation Index", value=20, precision=0, minimum=0, scale=1)
+
+        # Right Column: Scan Settings
+        with gr.Column(scale=1):
+            gr.Markdown("#### ⚙️ Scan Parameters", elem_classes=["section-title"])
+            phrase_mode = gr.Radio(
+                choices=["replace", "order", "subset"], 
+                value="replace", 
+                label="Mutation Mode",
+                info="How to modify the base words."
+            )
+            num_phrases = gr.Number(label="Random Phrases to Generate", value=10, precision=0, minimum=0)
+            passphrase = gr.Textbox(
+                label="BIP39 Passphrase (Optional)", 
+                value="", 
+                type="password",
+                info="Leave blank if no 25th word/passphrase is used."
+            )
+
+    gr.Markdown("#### 🔑 Your Base Phrase (Optional)", elem_classes=["section-title"])
     my_phrase = gr.Textbox(
-        label="🔑 Apni Phrase (optional)",
+        label="Enter your 12-word seed phrase",
         value="lesson cushion crane affair federal decline limb spend reunion chief cart zebra",
         lines=2,
+        placeholder="word1 word2 word3 ... word12",
+        info="If left blank, the scanner will only use randomly generated phrases."
     )
-        
-    num_phrases = gr.Number(label="Kitni Random Phrases Check Karein?", value=10, precision=0, minimum=0)
-    addresses_file = gr.Textbox(label="Addresses File", value="12730802_addresses.csv")
-    
-    phrase_mode = gr.Radio(choices=["replace", "order", "subset"], value="replace", label="Phrase Mode")
-    start_index = gr.Number(label="Start Index", value=0, precision=0, minimum=0)
-    end_index = gr.Number(label="End Index", value=20, precision=0, minimum=0)
-        
-    passphrase = gr.Textbox(label="BIP39 Passphrase (khali chhod dein)", value="", type="password")
-        
-    run_btn = gr.Button("🚀 Scan Shuru Karein", variant="primary")
 
-    gr.Markdown("### Output")
-    log_output = gr.Textbox(lines=15, interactive=False, show_label=False)
-    
-    gr.Markdown("### Matches")
-    matches_output = gr.Textbox(lines=8, interactive=False, show_label=False)
+    with gr.Row():
+        with gr.Column():
+            run_btn = gr.Button("🚀 START SCANNING", variant="primary", elem_classes=["primary-btn"], size="lg")
+
+    gr.Markdown("---")
+
+    with gr.Tabs():
+        with gr.Tab("📊 Live Execution Log"):
+            log_output = gr.Textbox(
+                lines=20, 
+                interactive=False, 
+                show_label=False,
+                elem_classes=["terminal-log"],
+                value="Waiting for scan to begin..."
+            )
+        
+        with gr.Tab("🎉 Matches Found"):
+            matches_output = gr.Textbox(
+                lines=10, 
+                interactive=False, 
+                show_label=False,
+                elem_classes=["match-log"],
+                value="No matches found yet."
+            )
 
     run_btn.click(
         fn=run_scan,
         inputs=[my_phrase, num_phrases, addresses_file, phrase_mode,
                 start_index, end_index, passphrase],
         outputs=[log_output, matches_output],
+        show_progress="full"
     )
 
 # ZeroGPU requirement
@@ -249,7 +378,6 @@ _noop()
 
 _original_launch = demo.launch
 def _patched_launch(**kwargs):
-    kwargs.setdefault("ssr_mode", False)
     kwargs.setdefault("share", True)
     kwargs.setdefault("server_name", "0.0.0.0")
     kwargs.setdefault("server_port", 7860)
@@ -257,4 +385,4 @@ def _patched_launch(**kwargs):
 demo.launch = _patched_launch
 
 if __name__ == "__main__":
-    demo.launch(theme=gr.themes.Soft())
+    demo.launch()