Update app.py

app.py

@@ -47,22 +47,22 @@ if "messages" not in st.session_state:
 init_alr = False
 
 def init_model(user_input):
-    # os.environ["API_KEY"] = "AIzaSyAgqODyVqcqlsDC1Gzuy8TkpiWnQZVGovw"
-    # genai.configure(api_key=os.environ.get("API_KEY"))
-    # model = genai.GenerativeModel('gemini-pro')
+    os.environ["API_KEY"] = "AIzaSyAgqODyVqcqlsDC1Gzuy8TkpiWnQZVGovw"
+    genai.configure(api_key=os.environ.get("API_KEY"))
+    model = genai.GenerativeModel('gemini-pro')
     
-    # userInput = user_input #"Give me an afrofunk beat along with a cool melody and catchy drums, along with an opera style"
-    # prompt = userInput + ". Output ONLY the specific music genre that the user would like that fits their requirements, with no excess words. NO APOSTROPHES. Add more than one genre to best fit the user's prompt. Desired Output: <all_lowercase_comma_separated_genres>"
-    # response = model.generate_content(prompt)
-    # genre = response.text
+    userInput = user_input #"Give me an afrofunk beat along with a cool melody and catchy drums, along with an opera style"
+    prompt = userInput + ". Output ONLY the specific music genre that the user would like that fits their requirements, with no excess words. NO APOSTROPHES. Add more than one genre to best fit the user's prompt. Desired Output: <all_lowercase_comma_separated_genres>"
+    response = model.generate_content(prompt)
+    genre = response.text
     
-    # userInput = "Genre: " + genre + ". Generate lyrics that would best fit into the required genre. Generate around 2 minutes worth of lyrics. Separate verses using [Verse #] and NO APOSTROPHES AT ALL. Desired Format: <seperate_each_verse_no_asteriks_or_apostrophes>"
-    # response = model.generate_content(userInput)
-    # lyrics = response.text
+    userInput = "Genre: " + genre + ". Generate lyrics that would best fit into the required genre. Generate around 2 minutes worth of lyrics. Separate verses using [Verse #] and NO APOSTROPHES AT ALL. Desired Format: <seperate_each_verse_no_asteriks_or_apostrophes>"
+    response = model.generate_content(userInput)
+    lyrics = response.text
     
-    # lyricInput = "Lyrics: " + lyrics[:25] + ". Based upon the following lyrics, please generate a title for this song. Desired Format: <title_case_without_genre>"
-    # response = model.generate_content(lyricInput)
-    # title = response.text
+    lyricInput = "Lyrics: " + lyrics[:25] + ". Based upon the following lyrics, please generate a title for this song. Desired Format: <title_case_without_genre>"
+    response = model.generate_content(lyricInput)
+    title = response.text
 
     # userInput = "Lyrics: " + lyrics + ". Split the following lyrics into ONLY 10 slices with each slice enhanced to describe a music video. You may slightlly alter the lyrics to best cater to this requirment. Each section is separated by a |. Use only 10 |'s. Delete ALL \n characters and Verse seperators as well. Desired Output: <|_separated>"
     # response = model.generate_content(userInput)
@@ -70,18 +70,10 @@ def init_model(user_input):
     # whollySplit = sepTen.split("|")
 
 
-    #generate_audio(lyrics, genre, title)
+    generate_audio(lyrics, genre, title)
     
     #make API request
-    #audiourl = getSongUrl(title)
-
-    title = 'Realm of Whispers'
-    genre = 'afrofunk, opera'
-
-    lyrics = "[Verse 1]\nIn the realm where rhythms soar\nWhere funky beats ignite the core\nLike a cosmic dance, our spirits unite\nAs ancestral drums carry us through the night\n[Verse 2]\nAfrofunk's siren, a call so divine\nA symphony of souls, a musical shrine\nFrom Yoruba chants to Congolese grooves\nWe traverse the realms of mystical moves\n[Verse 3]\nVoices erupt in operatic splendor\nA timeless tapestry, a transcendent render\nEchoes of Verdi meet Fela's bold fire\nOpera's grandeur melds with funk's primal desire\n[Verse 4]\nIn this fusion of worlds, a cosmic delight\nWhere melodies soar to an ethereal height\nWe become vessels of musical ecstasy\nAfrofunk and opera, a symphony of majesty"
-    
-
-    audiourl = "https://cdn1.suno.ai/c94802b4-5624-48ee-8e85-44837ae69e4a.mp4"
+    audiourl = getSongUrl(title)
     
     #call webscraping script
     #get_mp3("Samba Kickoff")