chatgpt rewrites app for gradio > 4.0

app.py

@@ -206,57 +206,68 @@ all_phrases_grateful_dead = generate_cumulative_phrases(songs_from_text(grateful
 size = 350 #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. 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
-Artist_dropdown = gr.Dropdown(choices=["Radiohead", "Kendrick Lamar","Grateful Dead","Google Gemma"],
-                        value="Radiohead",
-                        label="Select artist",
-                        #info="More coming soon"
-                        )
-Arist_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,
-)
-#Chatbot = gr.Chatbot()
+client = InferenceClient(token=HF_API_KEY)
+
+system_message = "Don't be too repetitive. Please limit your response to only a few sentences."
+
+artist_history = [""]  # If you want to track previous artist selection
+
+# Size for the image thumbnail (set your size)
+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:
-    #   gr.Markdown("### Select your flavor.")
-    #with gr.Row(equal_height=True):
     with gr.Row():
         with gr.Column(scale=1):
-            gr.Interface(
-                fn = return_image,
-                outputs = Arist_image,
-                inputs = Artist_dropdown,
-                live=True,
-                clear_btn=None,
+            artist_dropdown = gr.Dropdown(
+                choices=["Radiohead", "Kendrick Lamar", "Grateful Dead", "Google Gemma"],
+                value="Radiohead",
+                label="Select artist",
             )
-        with gr.Column(scale=3):
-            chatbot = gr.Chatbot(height=400,
-                                 type='messages',)
-            gr.ChatInterface(
-                chat_with_musician,
-                additional_inputs=[Artist_dropdown],
-                type="messages",
-                title="Lyrical Language Model",
-                description="Enter a message, receive vibes",
-                fill_height=False,
-                chatbot=chatbot,
+            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,
             )
-    
-# Launch the chatbot
-if __name__ == "__main__":
-    demo.launch()
+        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
+
+# Launch app
+if __name__ == "__main__":
+    demo.launch()