Update app.py

app.py

@@ -110,9 +110,10 @@ def artist_response(gemma_response, artist):
 
     return lyric_response
 
-def chat_with_musician(user_input, history, artist, previous_artist=None):
+def chat_with_musician(user_input, history, artist):
     if history is None:
         history = []
+    previous_artist = artist_history[-1]
     if artist != previous_artist:
         # Reset history if the artist changes
         history = []
@@ -138,7 +139,7 @@ def chat_with_musician(user_input, history, artist, previous_artist=None):
 
     history.append({"role": "user", "content": user_input})
     history.append({"role": "assistant", "content": lyric_response})
-    previous_artist = artist  # Store the previous artist for the next call
+    artist_history.append(artist)  # Store the previous artist for the next call
     #if artist == "Google Gemma":
     #    lyric_response = gemma_response
     return lyric_response
@@ -186,7 +187,7 @@ client = InferenceClient(#provider="hf-inference",
 system_message = "Don't be too repetitive. Please limit your respone to only a few sentences."
 #Function to generate chatbot responses
 
-previous_artist = None  # Variable to keep track of the previous artist
+artist_history = [""]  # Variable to keep track of the previous artist
 Artist_dropdown = gr.Dropdown(choices=["Radiohead", "Kendrick Lamar","Grateful Dead","Google Gemma"],
                         value="Radiohead",
                         label="Select artist",