Update app.py

app.py

@@ -36,13 +36,13 @@ dataframes = {
 
 #df = pd.read_csv("TA_embeddings.csv")
 #df["embedding"]=df["embedding"].apply(eval).apply(np.array)
-def reply(input, csv_dropdown):
+def reply(input, dataset_name):
     try:
-        #if dataset_name not in dataframes:
-         #   return "Invalid dataset selected. Please select a valid dataset."
+        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 = csv_dropdown
+        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])
@@ -69,20 +69,15 @@ def reply(input, csv_dropdown):
                     model="text-davinci-003"
                 )["choices"][0]["text"].strip(" \n")
         return response
-        
     except Exception as e:
         return f"An error occurred: {e}"
-        
-    
-    
-
 
 csv_dropdown = gr.inputs.Dropdown(
-                        label="Select the Book",
-                        choices=["AP_Bio", "AP_Physics"],
-                        default="AP_Bio"
+    label="Select the Book",
+    choices=["AP_Bio", "AP_Physics"],
+    default="AP_Bio"
             
-                        )
+)
 input_text = gr.inputs.Textbox(
     label="Enter your questions here",
     placeholder="E.g. What is DNA?",