adding retrieval system

app.py

@@ -4,9 +4,13 @@ from sentence_transformers import SentenceTransformer
 import torch
 from huggingface_hub import InferenceClient
 
+file_names = [
+    "financial_literacy_text.txt",
+    "financial_aid_text.txt"
+]
 
-# with open("insert name here.txt", "r", encoding = "utf-8") as file:
-    # finlitText = file.read()
+with open(file_names, "r", encoding = "utf-8") as file:
+    financialText = file.read()
 
 model = SentenceTransformer('all-MiniLM-L6-v2')
 
@@ -29,8 +33,8 @@ def getTopChunks(query, chunkEmbeddings, textChunks):
     topChunks = [textChunks[i] for i in topIndices]
     return topChunks
 
-# cleanedChunks = preprocessText(finlitText)
-# chunkEmbeddings = createEmbeddings(cleanedChunks)
+cleanedChunks = preprocessText(financialText)
+chunkEmbeddings = createEmbeddings(cleanedChunks)
 
 client = InferenceClient("deepseek-ai/DeepSeek-R1", token = os.environ.get("SF_TOKEN"))
 
@@ -45,11 +49,11 @@ def respond(message, history):
     if history:
         messages.extend(history)
 
-    # topResults = getTopChunks(message, chunkEmbeddings, cleanedChunks)
-    # context = "\n\n".join(topResults)
+    topResults = getTopChunks(message, chunkEmbeddings, cleanedChunks)
+    context = "\n\n".join(topResults)
 
-    # messages.append({"role": "system",
-    #   "content": context})
+    messages.append({"role": "system",
+                    "content": context})
 
     messages.append({"role": "user",
                      "content": message})