Update app.py

app.py

@@ -37,38 +37,38 @@ 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. Simply the genre, NO APOSTROPHES. If required, add more than one genre to best fit the user's prompt by adding a comma and space and a another genre. 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. Simply the genre, NO APOSTROPHES. If required, add more than one genre to best fit the user's prompt by adding a comma and space and a another genre. 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. The lyrics should not be about the genre, but would best fit into the genre. Generate around 1 minutes worth of lyrics. Separate verses using [Verse #] and NO APOSTROPHES AT ALL. If you must use them, please remove them from the word, even if it mispells the word. 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. The lyrics should not be about the genre, but would best fit into the genre. Generate around 1 minutes worth of lyrics. Separate verses using [Verse #] and NO APOSTROPHES AT ALL. If you must use them, please remove them from the word, even if it mispells the word. Desired Format: <seperate_each_verse_no_asteriks_or_apostrophes>"
+    response = model.generate_content(userInput)
+    lyrics = response.text
     
-    # lyricInput = "Lyrics: " + lyrics + ". 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 + ". 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)
-    # sepTen = response.text
-    # whollySplit = sepTen.split("|")
+    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)
+    sepTen = response.text
+    whollySplit = sepTen.split("|")
 
-    #Add code for list of frame
-    #call separate script for frame generation
-    #gen_movie(whollySplit[:10])
+    Add code for list of frame
+    call separate script for frame generation
+    gen_movie(whollySplit[:10])
 
     #make API request
 
-    #TODO
+    
     
     #call webscraping script
-    get_mp3("Samba Kickoff")
+    #get_mp3("Samba Kickoff")
 
     return genre