Update app.py

app.py

@@ -36,26 +36,27 @@ dataframes = {
 #df = pd.read_csv("TA_embeddings.csv")
 #df["embedding"]=df["embedding"].apply(eval).apply(np.array)
 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])
-    data = df.sort_values("similarities", ascending=False).head(10)
-    data.to_csv("sorted.csv")
-    context = []
-    for i, row in data.iterrows():
-        context.append(row['text'])
-    context
-    text = "\n".join(context)
-    context = text
-    prompt = f"""
+    try:
+        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])
+        data = df.sort_values("similarities", ascending=False).head(10)
+        data.to_csv("sorted.csv")
+        context = []
+        for i, row in data.iterrows():
+            context.append(row['text'])
+        context
+        text = "\n".join(context)
+        context = text
+        prompt = f"""
                 Answer the following question using the context given below.If you don't know the answer for certain, say I don't know.
                 Context: {context}
 
                 Q: {input}
 
                 """      
-    return openai.Completion.create(
+        response= openai.Completion.create(
                     prompt=prompt,
                     temperature=1,
                     max_tokens=500,
@@ -64,6 +65,10 @@ def reply(input, dataframe_name):
                     presence_penalty=0,
                     model="text-davinci-003"
                 )["choices"][0]["text"].strip(" \n")
+        return response
+    except Exception as e:
+        return f"An error occurred: {e}"
+