Update app.py

app.py

@@ -1,63 +1,36 @@
 import streamlit as st
 import numpy as np
 import soundfile as sf
+import io
 
-def text_to_melody(text, sample_rate=22050, duration=5):
-    notes_map = {
-        'a': 220, 'b': 246.94, 'c': 261.63, 'd': 293.66, 'e': 329.63, 'f': 349.23, 'g': 392, 'h': 440,
-        'i': 493.88, 'j': 523.25, 'k': 587.33, 'l': 659.25, 'm': 698.46, 'n': 783.99, 'o': 880,
-        'p': 987.77, 'q': 1046.5, 'r': 1174.66, 's': 1318.51, 't': 1396.91, 'u': 1567.98, 'v': 1760,
-        'w': 1975.53, 'x': 2093, 'y': 2349.32, 'z': 2637, ' ': 0,
-        'а': 277.18, 'б': 311.13, 'в': 329.63, 'г': 369.99, 'д': 415.30, 'е': 440, 'ё': 466.16, 'ж': 493.88,
-        'з': 554.37, 'и': 622.25, 'й': 659.25, 'к': 739.99, 'л': 830.61, 'м': 932.33, 'н': 987.77, 'о': 1046.5,
-        'п': 1108.73, 'р': 1244.51, 'с': 1318.51, 'т': 1479.98, 'у': 1661.22, 'ф': 1864.66, 'х': 2217.46,
-        'ц': 2489.02, 'ч': 2793.83, 'ш': 2959.96, 'щ': 3322.44, 'ъ': 3729.31, 'ы': 4186.01, 'ь': 4698.63,
-        'э': 4978.03, 'ю': 5587.65, 'я': 5919.91
-        # Add more characters as needed
-    }
+notes_map = {
+    'a': 220, 'b': 246.94, 'c': 261.3, 'd': 293.66, 'e': 329.63, 'f': 349.23, 'g': 392, 'h': 440,
+    'i': 493.88, 'j': 523.25, 'k': 587.33, 'l': 659.25, 'm': 698.46, 'n': 783.99, 'o': 880,
+    'p': 987.77, 'q': 1046.5, 'r': 1147.66, 's': 1318, 't': 1396, 'u': 1567.98, 'v': 1760.66,
+    'w': 1975.7, 'x': 2095, 'y': 2349.32, 'z': 2607, '!': 5078, ' ': 0
+}
 
-    # Russian word melodies
-    word_melodies = {
-        'christmas': [329.63, 392, 493.88, 587.33, 659.25, 783.99, 880, 987.77],
-        'celebration': [440, 493.88, 587.33, 659.25, 783.99, 880, 987.77, 1046.5],
-        'holiday': [659.25, 698.46, 783.99, 659.25, 698.46, 783.99, 659.25, 329.63, 659.25, 697],
-        'joyful': [392, 440, 392, 440, 392, 440, 392, 330, 392],
-        'party': [783.99, 880, 783.99, 880, 783.99, 698.46, 659.25, 587.33, 659.25, 698.46, 783.99, 987.77],
-        'stars': [261.63, 293.66, 349.23, 392, 440, 493.88, 523.25, 587.33],
-        'дружба': [277.18, 329.63, 369.99, 415.30, 493.88, 622.25],
-        'солнце': [440, 493.88, 554.37, 659.25, 739.99, 830.61],
-        'путешествие': [277.18, 311.13, 349.23, 392, 440, 493.88, 554.37, 622.25, 739.99, 830.61, 932.33, 1046.5]
-        # Add more words with associated melodies as needed
-    }
-
-    text = text.lower()
-    prompt = text
-
-    audio_data = []
+def generate_music(prompt, duration=1.0):
+    audio_data = np.array([])
     for char in prompt:
         if char in notes_map:
             freq = notes_map[char]
-            duration = 1  # You can adjust the duration as needed
             t = np.linspace(0, duration, int(44100 * duration), False)
             note = np.sin(freq * t * 2 * np.pi)
             audio_data = np.append(audio_data, note)
 
-    t = np.linspace(0, duration, int(sample_rate * duration), endpoint=False)
-    audio = sum(np.sin(2 * np.pi * note * t) for note in audio_data)
-    audio /= np.max(np.abs(audio))
+    return audio_data
 
-    return audio, sample_rate
+st.sidebar.title("Music Generator")
+st.sidebar.write("Music generator based on your input")
+prompt = st.sidebar.text_input("Enter prompt...")
 
-def main():
-    st.title('Text to Melody Streamlit App')
-    text_input = st.text_area("Enter text for melody generation", value="дружба")
+if st.sidebar.button("Generate"):
+    audio = generate_music(prompt)
 
-    if st.button("Generate Melody from Text"):
-        st.write("Generating melody based on the provided text prompt...")
-        audio_data, sample_rate = text_to_melody(text_input)
-        with st.empty():
-            sf.write('output_melody.wav', audio_data, sample_rate, format='wav')
-            st.audio('output_melody.wav', format='audio/wav')
+    bytes_io = io.BytesIO()
+    sf.write(bytes_io, audio, 44100, format ='WAV')
+    bytes_io.seek(0)
 
-if __name__ == '__main__':
-    main()
+    st.audio(bytes_io, format="audio/wav")
+    st.write("Audio generation finished successfully")