Update app.py

app.py

@@ -36,15 +36,14 @@ dataframes = {
 
 #df = pd.read_csv("TA_embeddings.csv")
 #df["embedding"]=df["embedding"].apply(eval).apply(np.array)
-def reply(input,dataframes):
-#, dataset_name):
+def reply(input, dataset_name):
     global messages
     try:
         #if dataset_name not in dataframes:
          #   return "Invalid dataset selected. Please select a valid dataset."
         if not input:
             return "Please Enter a Question to get an Answer"
-        df = dataframes
+        df = dataframes[dataset_name]
         input = input
         input_vector = generate_embeddings(input, model_id,hf_token)
         df["similarities"]=df["embedding"].apply(lambda x: cosine_similarity([x],[input_vector])[0][0])