Update app.py

app.py

@@ -138,6 +138,14 @@ def rag(input_text, history, jezik, file):
 #        metadatas=[{"state": "s0", "next": "s0", "used": False, "source": 'None', "page": -1, "lang": jezik } for i in range(len(documents)) ]
 #    )
 ##    else:
+    query_results = collection.query(
+        query_texts = [ input_text ],
+        n_results = max_results,
+        where = { "lang": jezik },
+        #where = { "$and": [ {"lang": jezik}, {"page": { "$nin": [ -1 ]}}]},
+        #where = { "$and": [ {"$and": [ { "$or": [ {"state": self.cs },  { "page": { "$nin": [ -1 ] } } ] } , { "used": False } ] } ,
+        #        {"lang": jezik } ] },
+    )
     o_jezik = "N/A"
     match jezik:
         case 'hrvatski':