Update app.py

app.py

@@ -36,12 +36,24 @@ def compare_text_to_audio_embeddings(text, pickle_file):
 
 def get_matches(text_query):
     matches = compare_text_to_audio_embeddings(text_query, "audio_embeddings_v3.pkl")
-    
+
     # Format the output
-    output = f"Top 5 matches for '{text_query}':\n\n"
-    for filename, url, similarity in matches[:5]:
-        output += f"{filename}, {url}: {similarity:.4f}\n"
-    
+    output = f"## Top 5 matches for '{text_query}'\n\n"
+
+    for filename, url, similarity in matches[:3]:
+        output += f"{similarity:.4f}\n"
+        
+        # Check if the URL is a YouTube URL
+        if "youtube.com" in url or "youtu.be" in url:
+            # Extract video ID from the URL
+            video_id = url.split("v=")[-1] if "v=" in url else url.split("/")[-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:
+            # Use regular link for non-YouTube URLs
+            output += f'<a href="{url}">{url}</a>\n\n'
+
     return output
 
 # Create the Gradio interface