requirements.txt : added

app.py

@@ -1,22 +1,39 @@
 import gradio as gr
+import laion_clap
+from qdrant_client import QdrantClient
 
+from settings import QDRANT_KEY, QDRANT_URL, ENVIRONMENT
 
-class Hit():
-    def __init__(self):
-        self.score = 1
-        self.payload = {
-            "audio_path": "https://synthia-research.s3.amazonaws.com/music_db/7fbb0c4de0e4bdf5e1dec8f3e803174b.mp3"
-        }
+# Loading the Qdrant DB in local ###################################################################
+if ENVIRONMENT == "PROD":
+    qdrant_client = QdrantClient(url=QDRANT_URL, api_key=QDRANT_KEY)
+else:
+    qdrant_client = QdrantClient("localhost", port=6333)
+print("[INFO] Client created...")
+
+# loading the model
+print("[INFO] Loading the model...")
+model_name = "laion/larger_clap_music"
+model = laion_clap.CLAP_Module(enable_fusion=False)
+model.load_ckpt()  # download the default pretrained checkpoint.
+
+# Gradio Interface #################################################################################
+max_results = 10
 
 
 def sound_search(query):
-    hits = [Hit() for _ in range(3)]
+    text_embed = model.get_text_embedding([query, ''])[0]  # trick because can't accept singleton
+    hits = qdrant_client.search(
+        collection_name="music_db",
+        query_vector=text_embed,
+        limit=max_results,
+    )
     return [
         gr.Audio(
-            hit.payload['audio_path'],
+            hit.payload['s3_url'],
             label=f"score: {hit.score}")
         for hit in hits
-    ]
+    ] * 3
 
 
 with gr.Blocks() as demo:

requirements.txt

@@ -0,0 +1,7 @@
+python-dotenv
+torch
+gradio
+laion_clap
+qdrant_client
+torchvision
+transformers==4.30.0

settings.py

@@ -0,0 +1,16 @@
+import os
+import os.path as op
+from pathlib import Path
+
+from dotenv import load_dotenv
+
+BASE_DIR_PATH = Path(op.dirname(Path(__file__).resolve()))
+load_dotenv(BASE_DIR_PATH / ".env")
+
+# QDrant
+QDRANT_KEY = os.environ.get('QDRANT_KEY')
+QDRANT_URL = os.environ.get('QDRANT_URL')
+QDRANT_COLLECTION_NAME = 'music_db'
+
+# Environment
+ENVIRONMENT = "PROD" if os.environ.get('USERNAME') == "arthur" else "PROD"