Update app.py

app.py

@@ -540,13 +540,24 @@ st.markdown("""
 
 
 
+from sentence_transformers import SentenceTransformer, util
+
+# بارگذاری مدل برای جستجوی معنایی
+model = SentenceTransformer('paraphrase-multilingual-MiniLM-L12-v2')
+
 if query:
     found = False
-    threshold = 62
+    threshold = 0.62 
+
+    query_embedding = model.encode(query, convert_to_tensor=True)
+
+    sentence_embeddings = model.encode(all_sentences, convert_to_tensor=True)
+
+    similarities = util.pytorch_cos_sim(query_embedding, sentence_embeddings)[0]
 
-    for idx, sentence in enumerate(all_sentences):
-        similarity = fuzz.partial_ratio(query, sentence)
+    for idx, similarity in enumerate(similarities):
         if similarity >= threshold:
+            sentence = all_sentences[idx]
             prompt = f"""
                         تعدادی پاسخ برای سوال زیر تولید شده است. لطفاً ابتدا این پاسخ‌ها را بررسی کن، سپس با در نظر گرفتن محتوای سوال و لحن آن، یک پاسخ نهایی حرفه‌ای، دقیق و روان ارائه کن که هم به سوال پاسخ دهد و هم از نظر نگارشی و ساختاری در سطح بالایی باشد. پاسخ نهایی باید حداکثر 1024 کاراکتر و حداقل 512 باشد، خلاصه و واضح نوشته شود و فقط به زبان فارسی باشد. از تکرار اضافی پرهیز کن و محتوای چند پاسخ را در صورت نیاز با هم ترکیب کن.
                         سوال:
@@ -563,8 +574,7 @@ if query:
 
             # بررسی مرتبط بودن پاسخ با سؤال
             review_prompt = f"""
-            لطفاً بررسی کن که آیا پاسخ زیر، حدودا  به سوال زیر پاسخ داده است یا خیر یعنی  کلمات  کلیدی اصلی در سوال در پاسخ هست یا ن . اگر پاسخ به سوال مرتبط، درست و قابل قبول است، فقط بنویس: 'تأیید شدواگر کمی مربوط بود بنویس: 'کمی خوب''.
-            اگر پاسخ اشتباه است یا هیچ ربطی به سوال ندارد، فقط بنویس: 'نیاز به اصلاح دارد'.
+            لطفاً بررسی کن که آیا پاسخ زیر، حدودا  به سوال زیر پاسخ داده است یا خیر یعنی  کلمات  کلیدی اصلی در سوال در پاسخ هست یا ن . اگر پاسخ به سوال مرتبط، درست و قابل قبول است، فقط بنویس: 'تأیید شد'. اگر کمی مربوط بود بنویس: 'کمی خوب'. اگر پاسخ اشتباه است یا هیچ ربطی به سوال ندارد، فقط بنویس: 'نیاز به اصلاح دارد'.
             سوال:
             {query}
             پاسخ: