Update app.py

app.py

@@ -219,96 +219,99 @@ size = 150
 
 
 
-def update_artist_image(artist):
-    # Call your existing function to get the image path or PIL.Image
-    return return_image(artist)
+# 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
 
-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))
+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:
         return chat_history
+    reply = f"Echo ({artist}): {message}"
+    chat_history = chat_history or []
+    chat_history.append((message, reply))
+    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
+with gr.Blocks() as demo:
+    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=150, width=150)
 
-# Launch app
-if __name__ == "__main__":
-    #demo.launch()
-    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:
-            return chat_history
-        reply = f"Echo ({artist}): {message}"
-        chat_history = chat_history or []
-        chat_history.append((message, reply))
-        return chat_history
+    chatbot = gr.Chatbot(height=400, type="messages")
+    message = gr.Textbox(placeholder="Enter message here...")
 
-    with gr.Blocks() as demo:
-        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=150, width=150)
+    artist_dropdown.change(fn=return_image, inputs=artist_dropdown, outputs=artist_image)
+    message.submit(fn=respond, inputs=[message, artist_dropdown, chatbot], outputs=chatbot).then(lambda: "", None, message)
 
-        chatbot = gr.Chatbot(height=400, type="messages")
-        message = gr.Textbox(placeholder="Enter message here...")
 
-        artist_dropdown.change(fn=return_image, inputs=artist_dropdown, outputs=artist_image)
-        message.submit(fn=respond, inputs=[message, artist_dropdown, chatbot], outputs=chatbot).then(lambda: "", None, message)
+# Launch app
 
+if __name__ == "__main__":
+    #demo.launch()
+    
     demo.launch()