App version 1

Chromadb/6c03f27c-dc64-4ecd-aff5-c13bd101e3f7/data_level0.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a13e72541800c513c73dccea69f79e39cf4baef4fa23f7e117c0d6b0f5f99670
+size 3212000

Chromadb/6c03f27c-dc64-4ecd-aff5-c13bd101e3f7/header.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0ec6df10978b056a10062ed99efeef2702fa4a1301fad702b53dd2517103c746
+size 100

Chromadb/6c03f27c-dc64-4ecd-aff5-c13bd101e3f7/length.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:44466131cc589c2595a6f2320436bda108bcc206f0dfd06f24a57d9726f41232
+size 4000

Dockerfile

@@ -0,0 +1,14 @@
+# read the doc: https://huggingface.co/docs/hub/spaces-sdks-docker
+# you will also find guides on how best to write your Dockerfile
+
+FROM python:3.9
+
+WORKDIR /code
+
+COPY ./requirements.txt /code/requirements.txt
+
+RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
+
+COPY . .
+
+CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]

main.py

@@ -0,0 +1,41 @@
+import fastapi
+from fastapi.middleware.cors import CORSMiddleware
+import chromadb
+from chromadb.utils import embedding_functions
+from utils import GAMES_DICT
+
+app = fastapi.FastAPI()
+
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+@app.get("/")
+def hello_world():
+    return {"message": "Hello, World!"}
+
+@app.get("/{name}/{query}")
+def hello_world(name: str, query: str):
+    
+    if name in GAMES_DICT.keys():
+        chroma_client = chromadb.PersistentClient(path="Chromadb/")
+        SentenceTransformerEmbeddings= embedding_functions.SentenceTransformerEmbeddingFunction(model_name="all-mpnet-base-v2")
+        collection= chroma_client.get_collection("GameMaster", embedding_function=SentenceTransformerEmbeddings)   
+        
+        results= collection.query(
+        query_texts=[query],
+        n_results=10,
+        where= {"source": GAMES_DICT[name]},
+        include= [ "documents" ]
+        )
+        
+        answer= results["documents"][0][0]
+        
+        return {"message": answer}
+    
+    else:
+        return {"message": "Game not found"}
+        

requirements.txt

@@ -0,0 +1,4 @@
+fastapi
+chromadb
+sentence_transformers
+uvicorn

utils.py

@@ -0,0 +1,5 @@
+GAMES_DICT= {
+    "EXPLODING KITTENS":"../rules_md/exploding_kittens.md", 
+    "CARCASSONE":"../rules_md/carcassone.md", 
+    "CATAN":"../rules_md/catan.md"
+}