Update app.py

app.py

@@ -3,15 +3,23 @@ import pandas as pd
 
 df = pd.read_csv('./Mental_Health_FAQ.csv')
 
+import pandas as pd
+from sentence_transformers import SentenceTransformer
+# Assuming your DataFrame is already loaded as 'df'
+
 context_data = []
+
 for i in range(len(df)):
-  context = ""
-  for j in range(3):
-    context += df.columns[j]
-    context += ": "
-    context += df.iloc[i][j]
-    context += " "
+  context = f"Question: {df.iloc[i]['Questions']} Answer: {df.iloc[i]['Answers']}"
   context_data.append(context)
+# print(context_data)
+
+# Embed the contexts
+embedding_model = SentenceTransformer('all-MiniLM-L6-v2')
+context_embeddings = embedding_model.encode(context_data)
+
+print(f"Number of contexts: {len(context_data)}")
+print(f"Shape of embeddings: {context_embeddings.shape}")
 
 
 import os