Update app.py

app.py

@@ -38,7 +38,7 @@ def get_matches(text_query):
     matches = compare_text_to_audio_embeddings(text_query, "audio_embeddings_v3.pkl")
 
     # Format the output
-    output = f"<h2>Top 5 matches for '{text_query}</h2>'\n\n"
+    output = f"<h2>Top 5 matches for '{text_query}'</h2>\n\n"
 
     for filename, url, similarity in matches[:3]:
         output += f"{similarity:.4f}\n"
@@ -48,11 +48,13 @@ def get_matches(text_query):
             # Extract video ID from the URL
             video_id = url.split("v=")[-1] if "v=" in url else url.split("/")[-1]
             print(f"https://www.youtube.com/embed/{video_id}")
+            song_name = filename.rsplit('.', 1)
             # Create iframe for YouTube video
-            output += f'<iframe width="560" height="315" src="https://www.youtube.com/embed/{video_id}" frameborder="0" allow="autoplay; encrypted-media" allowfullscreen></iframe>\n\n'
-        else:
+            output += f'{song_name}: <a href="{url}">{url}</a>\n\n'
+            #output += f'<iframe width="560" height="315" src="https://www.youtube.com/embed/{video_id}" frameborder="0" allow="autoplay; encrypted-media" allowfullscreen></iframe>\n\n'
+        #else:
             # Use regular link for non-YouTube URLs
-            output += f'<a href="{url}">{url}</a>\n\n'
+            
 
     return output