Update app.py

app.py

@@ -34,8 +34,8 @@ dataframes = {
 }
 #df = pd.read_csv("TA_embeddings.csv")
 #df["embedding"]=df["embedding"].apply(eval).apply(np.array)
-def reply(input):
-    
+def reply(input, dataframe_name):
+    df = dataframes[dataframe_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])