Update app.py

app.py

@@ -27,13 +27,13 @@ def return_image(artist):
         return "gemma.png"
     return "radiohead.png"
     
-def find_most_relevant_lyric(lyrics, user_input):
-    user_doc = nlp(user_input)
-    best_match = max(lyrics, key=lambda lyric: user_doc.similarity(nlp(lyric)))
-    return best_match
-
-def stitch_lyrics(lyrics, line_number=1):
-    return [lyrics[i] + " " + lyrics[i + line_number] for i in range(len(lyrics) - line_number)]
+#def find_most_relevant_lyric(lyrics, user_input):
+#    user_doc = nlp(user_input)
+#    best_match = max(lyrics, key=lambda lyric: user_doc.similarity(nlp(lyric)))
+#    return best_match
+#
+#def stitch_lyrics(lyrics, line_number=1):
+#    return [lyrics[i] + " " + lyrics[i + line_number] for i in range(len(lyrics) - line_number)]
 
 # Load lyrics from a text file
 def load_lyrics(filename):
@@ -137,14 +137,26 @@ def chat_with_musician(user_input, history, artist):
         return f"Error: {str(e)}"
     
     lyric_response = artist_response(gemma_response, artist)
+    if lyric_response == messages[-2]["content"]:
+        # If the response is the same as the last one, get a new one:
+        messages[-1]={"role": "user", "content": system_message_repeated + "\n\n" + user_input}
+        try:
+            response = client.chat_completion(
+                messages=messages,
+                model="google/gemma-2-2b-it",
+                max_tokens=256,
+                temperature=0.75,
+                #top_p=0.9
+            )
+            gemma_response= response["choices"][0]["message"]["content"]
+        except Exception as e:
+            return f"Error: {str(e)}"
+    lyric_response = artist_response(gemma_response, artist)
 
     history.append({"role": "user", "content": user_input})
     history.append({"role": "assistant", "content": lyric_response})
     artist_history.append(artist)  # Store the previous artist for the next call
     artist_history[:] = artist_history[-10:]  # Keep only last 10 entries
-
-    #if artist == "Google Gemma":
-    #    lyric_response = gemma_response
     return lyric_response
 
 HF_API_KEY = os.environ["HF_API_KEY"]
@@ -188,6 +200,7 @@ client = InferenceClient(#provider="hf-inference",
 #system_message = "Please limit your response to only a few sentences."
 #system_message = "Be creative and imprecise with your word choice. And don't be too repetitive. Please limit your respone to only a few sentences."
 system_message = "Don't be too repetitive. Please limit your respone to only a few sentences."
+system_message_repeated = "Be creative and imprecise with your word choice. Limit your response to a few sentences"
 #Function to generate chatbot responses
 
 artist_history = [""]  # Variable to keep track of the previous artist