Update app.py

app.py

@@ -45,7 +45,7 @@ def respond(
     selected_references['similarity'] = scores.tolist()[0]
     sorted_references = selected_references.sort_values(by='similarity', ascending=False)
     print(sorted_references.shape[0])
-    sorted_references = sorted_references.iloc[:1]
+    sorted_references = sorted_references.iloc[:3]
     sorted_references = sorted_references.sort_values(by='similarity', ascending=True)
     print(sorted_references.shape[0])
     print(sorted_references['similarity'].tolist())
@@ -54,17 +54,20 @@ def respond(
     translator = Translator()
     
     for index, row in sorted_references.iterrows():
-        print(index)
-        print(f'{row["user"]}')
-        user = translator.translate(f'{row["user"]}', src='ar', dest='en').text
-        print(user)
-        #print(row['assistant'])
-        assistant = translator.translate(row['assistant']).text
-        #print(assistant)
-        messages.append({"role": "user", "content":user })
-        messages.append({"role": "assistant", "content": assistant})
+        if(type(row["user"]) is str and type(row['assistant']) is str):
+            try:
+                print(index)
+                print(f'{row["user"]}')
+                user = translator.translate(f'{row["user"]}', src='ar', dest='en').text
+                print(user)
+                #print(row['assistant'])
+                assistant = translator.translate(row['assistant']).text
+                #print(assistant)
+                messages.append({"role": "user", "content":user })
+                messages.append({"role": "assistant", "content": assistant})
+            except:
+                    print("An exception occurred")
     
-
     #adding more references
     df = pd.read_csv("moslem-bot-reference.csv")
     for index, row in df.iterrows():