Update app.py

app.py

@@ -10,6 +10,9 @@ multilingual_embeddings = embedding_functions.SentenceTransformerEmbeddingFuncti
 def predict(api_key, user_input):
     client = chromadb.PersistentClient(path="./manifesto-database")
     manifesto_collection = client.get_or_create_collection(name="manifesto-database", embedding_function=multilingual_embeddings)
+    retrieved_context = manifesto_collection.query(query_texts=[user_input], n_results=3, where={"ideology": "Authoritarian-right"})
+    contexts = [context for context in retrieved_context['documents']]
+    print(contexts[0])
     
     prompt = f"""[INST] {user_input} [/INST]"""