Update app.py

app.py

@@ -22,9 +22,9 @@ def return_image(artist):
     if artist == "Radiohead":
         return "radiohead.png"
     if artist == "Kendrick Lamar":
-        return base64.b64encode(open("kendrick.png", "rb").read()).decode()
+        return "kendrick.png"
     if artist == "Google Gemma":
-        return base64.b64encode(open("gemma.png", "rb").read()).decode()
+        return "gemma.png"
     
 def find_most_relevant_lyric(lyrics, user_input):
     user_doc = nlp(user_input)
@@ -49,46 +49,51 @@ client = InferenceClient(model="google/gemma-2-2b-it", token=HF_API_KEY)
 
 system_message = "Please limit your response to only a few sentences."
 # Function to generate chatbot responses
-def chat_with_musician(user_input, history, artist):
-    if history is None:
-        history = []
-    messages = []
-    #messages.append({"role": "user", "content": system_message})
-    
-    for dict in history[-5*2:]:  # Keep only the last 5 exchanges, #I think this must be even to ensure that the last user message is included
-        messages.append(dict)
-    # Add the latest user message
-    messages.append({"role": "user", "content": system_message + "\n\n" + user_input})
-    try:
-        response = client.chat_completion(
-            messages=messages,
-            model="google/gemma-2-2b-it",
-            max_tokens=256,
-            temperature=0.7,
-            top_p=0.9
-        )
-        gemma_response= response["choices"][0]["message"]["content"]
-    except Exception as e:
-        return f"Error: {str(e)}"
-    
-    history.append({"role": "user", "content": user_input})
-    history.append({"role": "assistant", "content": gemma_response})
-    if artist == "Radiohead":
-        lyric_response = find_most_relevant_lyric(stitched_lyrics, 
-                        gemma_response)
-    if artist == "Google Gemma":
-        lyric_response = gemma_response
-    return lyric_response
-
-with gr.Blocks() as demo:
-    gr.Markdown("Start typing below and then click **Run** to see the output.")
-    with gr.Row():
-        inp =gr.Dropdown(choices=["Radiohead", "Kendrick Lamar","Google Gemma"],
+#def chat_with_musician(user_input, history, artist):
+#    if history is None:
+#        history = []
+#    messages = []
+#    #messages.append({"role": "user", "content": system_message})
+#    
+#    for dict in history[-5*2:]:  # Keep only the last 5 exchanges, #I think this must be even to ensure that the last user message is included
+#        messages.append(dict)
+#    # Add the latest user message
+#    messages.append({"role": "user", "content": system_message + "\n\n" + user_input})
+#    try:
+#        response = client.chat_completion(
+#            messages=messages,
+#            model="google/gemma-2-2b-it",
+#            max_tokens=256,
+#            temperature=0.7,
+#            top_p=0.9
+#        )
+#        gemma_response= response["choices"][0]["message"]["content"]
+#    except Exception as e:
+#        return f"Error: {str(e)}"
+#    
+#    history.append({"role": "user", "content": user_input})
+#    history.append({"role": "assistant", "content": gemma_response})
+#    if artist == "Radiohead":
+#        lyric_response = find_most_relevant_lyric(stitched_lyrics, 
+#                        gemma_response)
+#    if artist == "Google Gemma":
+#        lyric_response = gemma_response
+#    return lyric_response
+Artist_dropdown = gr.Dropdown(choices=["Radiohead", "Kendrick Lamar","Google Gemma"],
                         value="Radiohead",
                         label="Select artist",
-                        info="More coming soon"),
-        out = gr.Image(return_image, inputs=inp, label="Thumbnail",height=size, width=size)
-    # Connect the input to the output using the defined functions.
+                        info="More coming soon")
+with gr.Blocks() as demo:
+    gr.Markdown("Select someone to chat with.")
+    with gr.Row():
+        gr.Interface(
+            fn = return_image,
+            inputs = Artist_dropdown,
+            outputs = gr.Image(label="Thumbnail",height=size, width=size),
+            live=True,
+        )
+
+   # Connect the input to the output using the defined functions.
     #inp.change(fn=return_image, inputs=inp, outputs=out)
     #gr.ChatInterface(
     #    chat_with_musician,