Update app.py

app.py

@@ -24,8 +24,22 @@ theme = gr.themes.Soft(
 with open("essay_writing.txt", "r", encoding="utf-8") as f: 
     essay_writing = f.read()
 
+with open("STAAR_essay.txt", "r", encoding="utf-8") as file:
+    staar_essay = file.read()
+
 # split the text into chunks 
-cleaned_chunks = [chunk.strip() for chunk in essay_writing.split("\n\n") if chunk.strip()]
+def preprocess_text(text): 
+    cleaned_text = text.strip()
+    chunks = cleaned_text.split("\n")
+    cleaned_chunks = []
+    for chunk in chunks: 
+        stripped_chunk = chunk.strip()
+        cleaned_chunks.append(stripped_chunk)
+    return cleaned_chunks
+
+essay_chunks = preprocess_text(essay_writing)
+staar_chunks = preprocess_text(staar_essay)
+essay_chunks.extend(staar_chunks)
 
 # load an embedding model 
 model = SentenceTransformer('all-MiniLM-L6-v2')