Upload folder using huggingface_hub

Dockerfile

@@ -0,0 +1,19 @@
+# 使用官方 Python 运行时作为父镜像
+FROM python:3.9-slim
+
+# 先更新包列表，然后安装 git。
+# --no-install-recommends 可以减少不必要的包安装，保持镜像体积较小。
+# 最后清理 apt 缓存，这是减小镜像大小的好习惯。
+RUN apt-get update && \
+    apt-get install -y git --no-install-recommends && \
+    rm -rf /var/lib/apt/lists/*
+
+# 设置工作目录
+WORKDIR /app
+COPY . .
+
+RUN  pip install --no-cache-dir -r requirements.txt 
+
+
+# 容器启动时运行 main.py
+CMD ["python", "main.py"]

README.md

@@ -1,24 +1,23 @@
----
-title: Simple Substitution Cipher Decryptor
-emoji: 🕵️
-colorFrom: blue
-colorTo: green
-sdk: gradio
-app_file: app.py
-pinned: false
-
----
-
-# Simple Substitution Cipher Decryptor
-
-This is a simple tool to automatically decrypt text that has been encrypted with a simple substitution cipher.
-
-**How it works:**
-The backend uses n-gram frequency analysis (specifically quadgrams) to score possible decryptions and find the most likely plaintext.
-
-**How to use:**
-1.  Paste your ciphertext into the "Ciphertext" box.
-2.  (Optional) If you know any letter mappings (e.g., you know 'a' in the ciphertext is 'T' in the plaintext), you can provide them in the "Known Key Mappings" box. The format is `a=T b=E`.
-3.  The decrypted plaintext will appear in the "Plaintext" box.
-
-*Note: The decryption process is heuristic and may not always produce a perfect result, especially for short ciphertexts.*
+---
+title: Simple Substitution Cipher Decryptor
+emoji: 🕵️
+colorFrom: blue
+colorTo: green
+sdk: gradio
+app_file: main.py
+pinned: false
+---
+
+# Simple Substitution Cipher Decryptor
+
+This is a simple tool to automatically decrypt text that has been encrypted with a simple substitution cipher.
+
+**How it works:**
+The backend uses n-gram frequency analysis (specifically quadgrams) to score possible decryptions and find the most likely plaintext.
+
+**How to use:**
+1.  Paste your ciphertext into the "Ciphertext" box.
+2.  (Optional) If you know any letter mappings (e.g., you know 'a' in the ciphertext is 'T' in the plaintext), you can provide them in the "Known Key Mappings" box. The format is `a=T b=E`.
+3.  The decrypted plaintext will appear in the "Plaintext" box.
+
+*Note: The decryption process is heuristic and may not always produce a perfect result, especially for short ciphertexts.*

main.py

@@ -0,0 +1,118 @@
+# 导入所需的库
+from pycipher import SimpleSubstitution as SimpleSub
+import random
+import re
+from ngram_score import ngram_score
+import proability
+import gradio as gr
+
+# 全局变量，避免重复加载大文件
+_fitness = None
+
+def get_fitness():
+    global _fitness
+    if _fitness is None:
+        print("Loading quadgrams data...")
+        _fitness = ngram_score('quadgrams.txt')
+        print("Quadgrams data loaded successfully.")
+    return _fitness
+
+def decrypt_text_internal(ciphertext):
+    fitness = get_fitness()
+    ctext = re.sub('[^A-Z]', '', ciphertext.upper())
+    maxkey = list('ABCDEFGHIJKLMNOPQRSTUVWXYZ')
+    maxscore = -99e9
+    parentscore, parentkey = maxscore, maxkey[:]
+    
+    i = 0
+    # 进一步减少迭代次数以加快响应速度
+    while i < 500:  # 减少迭代次数
+        i = i + 1
+        random.shuffle(parentkey)
+        deciphered = SimpleSub(parentkey).decipher(ctext)
+        parentscore = fitness.score(deciphered)
+        count = 0
+        while count < 500:  # 减少内部迭代次数
+            a = random.randint(0, 25)
+            b = random.randint(0, 25)
+            child = parentkey[:]
+            child[a], child[b] = child[b], child[a]
+            deciphered = SimpleSub(child).decipher(ctext)
+            score = fitness.score(deciphered)
+            if score > parentscore:
+                parentscore = score
+                parentkey = child[:]
+                count = 0
+            count = count + 1
+        if parentscore > maxscore:
+            maxscore, maxkey = parentscore, parentkey[:]
+            ss = SimpleSub(maxkey)
+            plaintext = ss.decipher(ctext)
+            plaintext1 = add_punctuation_and_spaces(ciphertext, plaintext)
+            # In a web context, we return the first good result.
+            # The original loop was infinite, which is not suitable for a server.
+            return plaintext1
+    # Fallback if no good solution is found within the iteration limit
+    ss = SimpleSub(maxkey)
+    plaintext = ss.decipher(ctext)
+    return add_punctuation_and_spaces(ciphertext, plaintext)
+
+
+def output(string1, dic, string2):
+    modified_string1 = list(string1)
+    modified_string2 = list(string2)
+    for i in range(len(string1)):
+        if modified_string1[i] in dic and modified_string2[i] != ' ':
+            modified_string2[i] = dic[modified_string1[i]]
+    modified_string2 = ''.join(modified_string2)
+    return modified_string2
+
+def add_punctuation_and_spaces(ciphertext, plaintext):
+    result = ""
+    j = 0
+    for i in range(len(ciphertext)):
+        if not ciphertext[i].isalpha():
+            result += ciphertext[i]
+        else:
+            if ciphertext[i].islower():
+                result += plaintext[j].lower()
+            else:
+                # The original code had a bug here, always making it lowercase.
+                # This is a guess at the intended behavior.
+                result += plaintext[j] 
+            j += 1
+    return result
+
+def decrypt_interface(ciphertext, key):
+    """
+    This is the main function that will be exposed through the Gradio interface.
+    """
+    if not ciphertext:
+        return "Please enter some ciphertext."
+        
+    plaintext = decrypt_text_internal(ciphertext)
+    
+    if key:
+        try:
+            key_dic = proability.read_key(key)
+            plaintext = output(ciphertext, key_dic, plaintext)
+        except Exception as e:
+            return f"Error processing key: {e}. Please check the key format (e.g., a=B c=D)."
+            
+    return plaintext
+
+# Create the Gradio interface
+iface = gr.Interface(
+    fn=decrypt_interface,
+    inputs=[
+        gr.Textbox(lines=10, label="Ciphertext", placeholder="Enter the text to decrypt..."),
+        gr.Textbox(lines=2, label="Known Key Mappings (Optional)", placeholder="e.g., a=B c=D")
+    ],
+    outputs=gr.Textbox(lines=10, label="Plaintext"),
+    title="Simple Substitution Cipher Decryptor",
+    description="An automatic decryption tool for simple substitution ciphers. You can optionally provide known letter mappings to improve accuracy."
+)
+
+# Launch the app
+if __name__ == "__main__":
+    iface.launch(server_name="0.0.0.0", server_port=7860)