Update app.py

app.py

@@ -11,7 +11,6 @@ import random
 
 
 load_dotenv()
-HF_API_KEY = os.environ["HF_API_KEY"]
 
 """
 For more information on `huggingface_hub` Inference API support, please check the docs: https://huggingface.co/docs/huggingface_hub/v0.22.2/en/guides/inference
@@ -132,14 +131,17 @@ def chat_with_musician(user_input, history, artist):
     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": gemma_response})
+    history.append({"role": "assistant", "content": lyric_response})
 
-    lyric_response = artist_response(gemma_response, artist)
     #if artist == "Google Gemma":
     #    lyric_response = gemma_response
     return lyric_response
 
+HF_API_KEY = os.environ["HF_API_KEY"]
+
 radiohead_lyrics = load_lyrics("radiohead_lyrics.txt")
 kendrick_lyrics = load_lyrics("kendrick_lamar_lyrics.txt")
 grateful_dead_lyrics = load_lyrics('grateful_dead_lyrics.txt')
@@ -169,14 +171,14 @@ grateful_dead_embeddings = grateful_dead_embeddings.astype(np.float32)
 
 
 
-size = 128 #256
+size = 225 #256
 
 # Initialize Hugging Face Inference Client
 client = InferenceClient(#provider="hf-inference",
                           token=HF_API_KEY)
 
 #system_message = "Please limit your response to only a few sentences."
-system_message = "Be creative and imprecise with your word choice. Please limit your respone 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."
 #Function to generate chatbot responses