Update app.py

app.py

@@ -129,66 +129,87 @@ def generate_phrases_for_words(wordlist, mode):
                 yield " ".join(combo)
 
 # ============================================================
-# Gradio functions
+# Gradio functions — YIELD for live updates
 # ============================================================
 
 def generate_random_phrase():
     return _get_mnemonic().generate(strength=128)
 
 
-def run_multi_phrase_scan(num_phrases, addresses_file, phrase_mode,
+def run_multi_phrase_scan(my_phrase, num_phrases, addresses_file, phrase_mode,
                           start_index, end_index, passphrase):
     """
-    User se poocha gaya number of phrases generate karta hai,
-    phir ek ek karke scan karta hai.
+    Generator — har step pe yield, taake UI live update ho.
     """
     if not os.path.exists(addresses_file):
-        return f"❌ File nahi mili: {addresses_file}", ""
+        yield f"❌ File nahi mili: {addresses_file}", ""
+        return
 
     num_phrases = int(num_phrases)
-    if num_phrases < 1:
-        return "❌ Kam az kam 1 phrase chahiye", ""
-
     start_index = int(start_index)
     end_index = int(end_index)
-    if start_index > end_index:
-        return "❌ Start index end index se bada nahi ho sakta", ""
+
+    yield "Addresses load ho rahe hain...", ""
 
     try:
         known = load_addresses_cached(addresses_file)
     except Exception as e:
-        return f"❌ Addresses load fail: {e}", ""
+        yield f"❌ Addresses load fail: {e}", ""
+        return
 
     if not known:
-        return "❌ Koi valid address file mein nahi mila", ""
+        yield "❌ Koi valid address file mein nahi mila", ""
+        return
 
-    # Generate phrases
     mnemo = _get_mnemonic()
-    phrases = [mnemo.generate(strength=128) for _ in range(num_phrases)]
+    phrases = []
+
+    # Apni phrase pehle
+    my_words = my_phrase.strip().split()
+    if len(my_words) == 12:
+        phrases.append(" ".join(my_words))
+
+    for _ in range(num_phrases):
+        phrases.append(mnemo.generate(strength=128))
 
     log_lines = [
         f"✅ Loaded {len(known):,} addresses (cached)",
         f"🔧 Mode: {phrase_mode}",
         f"🔑 Index range: {start_index}–{end_index}",
-        f"📝 Phrases to check: {num_phrases}",
+        f"📝 Total phrases: {len(phrases)}",
         "",
-        "Generated phrases:",
+        "Phrases:",
     ]
     for i, p in enumerate(phrases, 1):
-        log_lines.append(f"  {i}. {p}")
+        tag = " (AAPKI)" if i == 1 and len(my_words) == 12 else ""
+        log_lines.append(f"  {i}. {p}{tag}")
     log_lines.append("")
 
+    # Initial yield — phrases dikh jayein
+    yield "\n".join(log_lines), "Scan shuru ho raha hai..."
+
     matches = []
     start = time.time()
+    total_phrases = len(phrases)
 
     for p_idx, phrase in enumerate(phrases, 1):
-        log_lines.append(f"🔍 [{p_idx}/{num_phrases}] Checking: {phrase}")
+        log_lines.append(f"🔍 [{p_idx}/{total_phrases}] Checking: {phrase}")
+        yield "\n".join(log_lines), "\n".join(
+            f"Phrase: {p}\nIndex: {i}\nKey: {k}\nAddr: {a}\n" for p, i, k, a in matches
+        ) if matches else "Abhi tak koi match nahi"
+
         words = phrase.strip().split()
         if len(words) != 12:
             log_lines.append(f"   ⚠️ Skipped (not 12 words)")
+            yield "\n".join(log_lines), "Koi match nahi mila"
             continue
 
         for idx in range(start_index, end_index + 1):
+            log_lines.append(f"   Index {idx}...")
+            yield "\n".join(log_lines), "\n".join(
+                f"Phrase: {p}\nIndex: {i}\nKey: {k}\nAddr: {a}\n" for p, i, k, a in matches
+            ) if matches else "Abhi tak koi match nahi"
+
             for phrase_str in generate_phrases_for_words(words, phrase_mode):
                 if not is_valid_bip39(phrase_str):
                     continue
@@ -209,6 +230,10 @@ def run_multi_phrase_scan(num_phrases, addresses_file, phrase_mode,
                     )
                     log_lines.append(match_line)
                     matches.append((phrase_str, idx, priv.hex(), addr))
+                    # Match mile to turant dikhayein
+                    yield "\n".join(log_lines), "\n".join(
+                        f"Phrase: {p}\nIndex: {i}\nKey: {k}\nAddr: {a}\n" for p, i, k, a in matches
+                    )
 
     elapsed = time.time() - start
     log_lines.append(f"=== DONE ===")
@@ -219,7 +244,7 @@ def run_multi_phrase_scan(num_phrases, addresses_file, phrase_mode,
         f"Phrase: {p}\nIndex: {i}\nKey: {k}\nAddr: {a}\n" for p, i, k, a in matches
     ) if matches else "Koi match nahi mila"
 
-    return "\n".join(log_lines), matches_text
+    yield "\n".join(log_lines), matches_text
 
 
 # ============================================================
@@ -227,14 +252,19 @@ def run_multi_phrase_scan(num_phrases, addresses_file, phrase_mode,
 # ============================================================
 with gr.Blocks(title="Phrase Scanner") as demo:
     gr.Markdown("# 🔍 Phrase Scanner")
-    gr.Markdown("Kitni random phrases check karni hain? Phir index range ke saath scan karein.")
+    gr.Markdown("Apni phrase ya random phrases check karein, index range ke saath.")
 
     with gr.Row():
         with gr.Column(scale=1):
             gr.Markdown("### Settings")
+            my_phrase = gr.Textbox(
+                label="🔑 Apni Phrase (optional)",
+                value="lesson cushion crane affair federal decline limb spend reunion chief cart zebra",
+                lines=2,
+            )
             num_phrases = gr.Number(
                 label="Kitni Random Phrases Check Karein?",
-                value=10, precision=0, minimum=1,
+                value=10, precision=0, minimum=0,
             )
             addresses_file = gr.Textbox(
                 label="Addresses File",
@@ -279,7 +309,7 @@ with gr.Blocks(title="Phrase Scanner") as demo:
 
     run_btn.click(
         fn=run_multi_phrase_scan,
-        inputs=[num_phrases, addresses_file, phrase_mode,
+        inputs=[my_phrase, num_phrases, addresses_file, phrase_mode,
                 start_index, end_index, passphrase],
         outputs=[log_output, matches_output],
     )