Update app.py

app.py

@@ -122,6 +122,9 @@ def get_titan_embedding(bedrock, doc_name, text, attempt=0, cutoff=10000):
 def ask_ds(message, history):
 
 
+    if len(message) == 0:
+        return
+        
     question = message
 
     # RAG
@@ -129,7 +132,7 @@ def ask_ds(message, history):
 
     similar_documents = []
     for file, data in extractions.items():
-        similarity = cosine_similarity(question_embedding, data['embedding'])
+        similarity = cosine_similarity(question_embedding, np.array(data['embedding']))
         similar_documents.append((file, similarity))
 
     similar_documents.sort(key=lambda x: x[1], reverse=False)