Update app.py

app.py

@@ -48,20 +48,23 @@ def load_lyrics(filename):
     #return [line for line in lyrics]
 
 def artist_response(gemma_response, artist):
-    encoded_gemma = encoder_model.encode(gemma_response)
     if artist == "Radiohead":
         artist_embeddings = radiohead_embeddings
+        lyric_list = stitched_radiohead_lyrics
     if artist == "Kendrick Lamar":
         artist_embeddings = kendrick_embeddings
+        lyric_list = stitched_kendrick_lyrics
     if artist == "Grateful Dead":
         artist_embeddings = grateful_dead_embeddings
+        lyric_list = stitched_grateful_dead_lyrics
 
+    encoded_gemma = encoder_model.encode(gemma_response)
     similarity_result = encoder_model.similarity(
             encoded_gemma,
             artist_embeddings,
         )
     result_max_index = np.argmax(similarity_result)
-    lyric_response = stitched_kendrick_lyrics[result_max_index]
+    lyric_response = lyric_list[result_max_index]
     return lyric_response
 
 radiohead_lyrics = load_lyrics("radiohead_lyrics.txt")
@@ -115,10 +118,7 @@ def chat_with_musician(user_input, history, artist):
     
     history.append({"role": "user", "content": user_input})
     history.append({"role": "assistant", "content": gemma_response})
-    if artist == "Radiohead":
-        lyric_response = find_most_relevant_lyric(stitched_radiohead_lyrics, 
-                        gemma_response)
-        return lyric_response
+
     lyric_response = artist_response(gemma_response, artist)
     if artist == "Google Gemma":
         lyric_response = gemma_response