init

app.py

@@ -1,5 +1,5 @@
 import streamlit as st
-from sentence_transformers import SentenceTransformer, torch_util
+from sentence_transformers import SentenceTransformer, util
 
 def app():
     # 创建Streamlit应用程序
@@ -15,20 +15,21 @@ def app():
         st.session_state.inputs = []
         st.session_state.inputs_index = 0
 
-    # 创建一个添加输入框的按钮
-    add_input_button = st.button("添加一个待比较句子")
 
     with st.container():
         # 在容器中渲染已经存在的输入框列表
         for i in range(0, st.session_state.inputs_index):
             st.session_state.inputs[i]= st.text_input(f"请输入第 {i+1} 个句子", "", key=i)
 
-	    # 当用户点击按钮时往容器中添加新的输入框
-        if add_input_button:
-            i = st.session_state.inputs_index
-            st.session_state.inputs.append(st.text_input(f"请输入第 {i+1} 个句子", "", key=i))
-            # 自增输入框的key
-            st.session_state.inputs_index += 1
+    # 创建一个添加输入框的按钮
+    add_input_button = st.button("添加一个待比较句子")
+
+	# 当用户点击按钮时往容器中添加新的输入框
+    if add_input_button:
+        i = st.session_state.inputs_index
+        st.session_state.inputs.append(st.text_input(f"请输入第 {i+1} 个句子", "", key=i))
+        # 自增输入框的key
+        st.session_state.inputs_index += 1
     
     button_generate = st.button("计算")
     button_clear = st.button("清空")
@@ -36,32 +37,34 @@ def app():
     def transformer(sentences):
         model = SentenceTransformer('sentence-transformers/all-MiniLM-L6-v2', device="cpu")
         source_emb = model.encode(source_text, convert_to_tensor=True)
+        sent_embs = model.encode(sentences, convert_to_tensor=True)
+        cos_sim = util.cos_sim(source_emb, sent_embs)
 
         cosin_dict = []
 
-        for sent in sentences:
-            sent_emb = model.encode(sent, convert_to_tensor=True)
-            cosine_scores = torch_util.pairwise_cosine_similarity(source_emb, sent_emb)
-            cosin_dict[sent] = cosine_scores
+        for i, cos in cos_sim:
+            cosin_dict[sentences[i]] = cos
         
         sorted_dict = dict(sorted(cosin_dict.items(), key=lambda item: item[1],reverse = True))
         return sorted_dict
 
 
     if button_generate:
+        st.write(st.session_state["inputs"])
+        pass
         # embeddings
-        embeddings = transformer(st.session_state.inputs)
+        #embeddings = transformer(st.session_state.inputs)
 
         # 显示生成的文本
-        st.write(embeddings)
+        #st.write(embeddings)
         #output_text.success(generated_text)
 
     if button_clear:
-        for i in st.session_state.inputs:
-            del st.session_state.inputs[i]
         st.session_state.inputs.clear()
         del st.session_state["inputs"]
         st.session_state.inputs_index = 0
+        source_text = ''
+        st.experimental_rerun()
 
 if __name__ == "__main__":
     # 运行应用程序