Update app.py

app.py

@@ -10,13 +10,6 @@ llm = ChatOpenAI(
     api_key='0291f33aee03412a47fa5d8e562e515182dcc5d9aac5a7fb5eefdd1759005979',
     model="deepseek-ai/DeepSeek-R1"
 )
-def rewrite_answer_with_llm(answer, user_input):
-    prompt = f"پاسخی که باید بازنویسی شود:\n{answer}\n\nلطفاً این پاسخ را با در نظر گرفتن محتوای سوال زیر و لحن آن بازنویسی کن:\n\nسوال: {user_input}"
-    
-    response = llm([{"role": "system", "content": "You are a helpful assistant."},
-                    {"role": "user", "content": prompt}])
-    
-    return response.strip()  
 
 folder_path = '46'
 texts = []
@@ -45,21 +38,27 @@ if query:
     found = False
     for idx, sentence in enumerate(all_sentences):
         if query in sentence:
-            st.success("✅ جمله یافت شد:")
-            st.write(sentence)
 
             next_sentences = []
-            st.markdown("📌 پنج جمله بعدی:")
             for i in range(1, 6):
                 if idx + i < len(all_sentences):
-                    st.write(all_sentences[idx + i])
                     next_sentences.append(all_sentences[idx + i])
 
-            # ↪️ آماده‌سازی برای بازنویسی
+            # ↪️ آماده‌سازی پرامپت برای ارسال به مدل
             total_text = sentence + " " + " ".join(next_sentences)
-            rewritten = rewrite_answer_with_llm(total_text, query)
-            st.markdown("🎨 **بازنویسی شده با LLM:**")
-            st.write(rewritten)
+            prompt = f"پاسخی که باید بازنویسی شود:\n{total_text}\n\nلطفاً این پاسخ را با در نظر گرفتن محتوای سوال زیر و لحن آن بازنویسی کن:\n\nسوال: {query}"
+
+            # ارسال پرامپت به مدل و دریافت پاسخ
+            response = llm([{"role": "system", "content": "You are a helpful assistant."},
+                            {"role": "user", "content": prompt}])
+
+            # استخراج پاسخ مدل
+            if isinstance(response, dict) and 'choices' in response:
+                rewritten = response['choices'][0]['message']['content'].strip()
+                st.markdown("🎨 **بازنویسی شده با LLM:**")
+                st.write(rewritten)
+            else:
+                st.warning("پاسخ دریافتی معتبر نیست.")
 
             found = True
             break