Update app.py

app.py

@@ -174,7 +174,7 @@ def cosine_similarity_int8(query, embeddings):
 
     return dots / (emb_norms * query_norm + 1e-8)
 
-#HF_API_KEY = os.environ["HF_API_KEY"]
+HF_API_KEY = os.environ["HF_API_KEY"]
 
 
 
@@ -208,8 +208,8 @@ all_phrases_grateful_dead = generate_cumulative_phrases(songs_from_text(grateful
 size = 350 #256
 
 # Initialize Hugging Face Inference Client
-client = InferenceClient(#token=HF_API_KEY)
-)
+client = InferenceClient(token=HF_API_KEY)
+
 
 system_message = "Don't be too repetitive. Please limit your response to only a few sentences."
 
@@ -220,71 +220,71 @@ size = 150
 
 
 
-# def update_artist_image(artist):
-#     # Call your existing function to get the image path or PIL.Image
-#     return return_image(artist)
-
-# with gr.Blocks() as demo:
-#     with gr.Row():
-#         with gr.Column(scale=1):
-#             artist_dropdown = gr.Dropdown(
-#                 choices=["Radiohead", "Kendrick Lamar", "Grateful Dead", "Google Gemma"],
-#                 value="Radiohead",
-#                 label="Select artist",
-#                 interactive=True,
-#             )
-#             artist_image = gr.Image(
-#                 value=return_image("Radiohead"),
-#                 label="Thumbnail",
-#                 height=size,
-#                 width=size,
-#                 show_label=False,
-#                 show_fullscreen_button=False,
-#                 show_download_button=False,
-#                 show_share_button=False,
-#             )
-#         with gr.Column(scale=3):
-#             chatbot = gr.Chatbot(height=400, type='messages')
-
-#             message_input = gr.Textbox(
-#                 label="Your message",
-#                 placeholder="Enter a message and press Enter",
-#                 lines=2,
-#                 interactive=True,
-#             )
-
-#     # Update image when artist changes
-#     artist_dropdown.change(fn=update_artist_image, inputs=artist_dropdown, outputs=artist_image)
-
-#     # Define how chat works
-#     def chatbot_response(message, artist, chat_history):
-#         if message is None or message.strip() == "":
-#             return chat_history
-#         # Call your chat_with_musician function, which should return response text
-#         response = chat_with_musician(message, artist)
-#         chat_history = chat_history or []
-#         chat_history.append((message, response))
-#         return chat_history
-
-#     # Wire message input submit to update chatbot
-#     message_input.submit(
-#         fn=chatbot_response,
-#         inputs=[message_input, artist_dropdown, chatbot],
-#         outputs=chatbot,
-#         queue=False,
-#     ).then(lambda: "", None, message_input)  # Clear input after submit
+def update_artist_image(artist):
+    # Call your existing function to get the image path or PIL.Image
+    return return_image(artist)
 
-def return_image(artist):
-# For test: return a colored image based on artist
-    colors = {
-        "Radiohead": (255, 0, 0),
-        "Kendrick Lamar": (0, 255, 0),
-        "Grateful Dead": (0, 0, 255),
-        "Google Gemma": (255, 255, 0)
-    }
-    color = colors.get(artist, (128, 128, 128))
-    img = Image.new('RGB', (150, 150), color=color)
-    return img
+with gr.Blocks() as demo:
+    with gr.Row():
+        with gr.Column(scale=1):
+            artist_dropdown = gr.Dropdown(
+                choices=["Radiohead", "Kendrick Lamar", "Grateful Dead", "Google Gemma"],
+                value="Radiohead",
+                label="Select artist",
+                interactive=True,
+            )
+            artist_image = gr.Image(
+                value=return_image("Radiohead"),
+                label="Thumbnail",
+                height=size,
+                width=size,
+                show_label=False,
+                show_fullscreen_button=False,
+                show_download_button=False,
+                show_share_button=False,
+            )
+        with gr.Column(scale=1):
+            chatbot = gr.Chatbot(height=400, type='messages')
+
+            message_input = gr.Textbox(
+                label="Your message",
+                placeholder="Enter a message and press Enter",
+                lines=2,
+                interactive=True,
+            )
+
+    # Update image when artist changes
+    artist_dropdown.change(fn=update_artist_image, inputs=artist_dropdown, outputs=artist_image)
+
+    # Define how chat works
+    def chatbot_response(message, artist, chat_history):
+        if message is None or message.strip() == "":
+            return chat_history
+        # Call your chat_with_musician function, which should return response text
+        response = chat_with_musician(message, artist)
+        chat_history = chat_history or []
+        chat_history.append((message, response))
+        return chat_history
+
+    # Wire message input submit to update chatbot
+    message_input.submit(
+        fn=chatbot_response,
+        inputs=[message_input, artist_dropdown, chatbot],
+        outputs=chatbot,
+        queue=False,
+    ).then(lambda: "", None, message_input)  # Clear input after submit
+
+# def return_image(artist):
+# # For test: return a colored image based on artist
+#     colors = {
+#         "Radiohead": (255, 0, 0),
+#         "Kendrick Lamar": (0, 255, 0),
+#         "Grateful Dead": (0, 0, 255),
+#         "Google Gemma": (255, 255, 0)
+#     }
+#     color = colors.get(artist, (128, 128, 128))
+#     img = Image.new('RGB', (150, 150), color=color)
+#     return img
 
 def respond(message, artist, chat_history):
     if not message: