Update app.py

app.py

@@ -1,20 +1,26 @@
-import gradio as gr
+import re
 import numpy as np
 import wave
-
-# Assuming the classes and methods from animalese.js are translated into Python
+import gradio as gr
 
 class Animalese:
     def __init__(self, letters_file, onload):
         with open(letters_file, 'rb') as f:
             self.letter_library = np.frombuffer(f.read(), dtype=np.uint8)
         onload()
-        
+
     def synthesize(self, script, shorten=False, pitch=1.0):
         def shorten_word(word):
             return word[0] + word[-1] if len(word) > 1 else word
 
-        processed_script = "".join(map(shorten_word, script.replace(/[^a-z]/gi, ' ').split())) if shorten else script
+        if shorten:
+            # Replace all non-alphabet characters with spaces and split the script into words
+            words = re.sub(r'[^a-zA-Z]', ' ', script).split()
+            # Shorten each word and join them back into a single string
+            processed_script = "".join(map(shorten_word, words))
+        else:
+            processed_script = script
+
         data = []
 
         sample_freq = 44100
@@ -36,12 +42,13 @@ class Animalese:
         return self.create_wave(data, sample_freq)
     
     def create_wave(self, data, sample_rate):
-        with wave.open("output.wav", "wb") as f:
+        output_file = "output.wav"
+        with wave.open(output_file, "wb") as f:
             f.setnchannels(1)
             f.setsampwidth(1)
             f.setframerate(sample_rate)
             f.writeframes(data.tobytes())
-        return "output.wav"
+        return output_file
 
 # Initialize the synthesizer
 synth = Animalese('animalese.wav', lambda: print("Loaded"))
@@ -56,7 +63,7 @@ def preview_audio(audio_file):
 # Gradio UI
 with gr.Blocks() as demo:
     gr.Markdown("# Animalese.js Demo in Gradio")
-    gr.Markdown("vist also [web version](https://huggingface.co/spaces/Blane187/animalese-js)")
+    
     text_input = gr.Textbox(label="Input Text", placeholder="Enter text to convert to Animalese")
     shorten_input = gr.Checkbox(label="Shorten Words")
     pitch_input = gr.Slider(minimum=0.2, maximum=2.0, step=0.1, value=1.0, label="Pitch")