Update app.py

app.py

@@ -34,8 +34,9 @@ def add_random_noise(prompt, noise_level=0.07):
     noise_indices = random.sample(range(len(prompt)), num_noise_chars)
     # Add noise to the selected characters
     prompt_list = list(prompt)
+    noise_chars = string.ascii_letters + string.punctuation + ' '
     for index in noise_indices:
-        prompt_list[index] = random.choice(string.ascii_letters + string.punctuation)
+        prompt_list[index] = random.choice(noise_chars)
     return "".join(prompt_list)