Update app.py

app.py

@@ -14,13 +14,10 @@ llm = ChatOpenAI(
 def rewrite_answer_with_llm(answer, user_input):
     prompt = f"پاسخی که باید بازنویسی شود:\n{answer}\n\nلطفاً این پاسخ را با در نظر گرفتن محتوای سوال زیر و لحن آن بازنویسی کن:\n\nسوال: {user_input}"
 
-    response = llm.chat.completions.create(
-        messages=[{"role": "user", "content": prompt}],
-        model=llm.model
-    )
-    return response.choices[0].message.content.strip()
+    response = llm(prompt)
+    
+    return response['choices'][0]['text'].strip()
 
-# 📁 بارگذاری فایل‌های کتاب
 folder_path = '46'
 texts = []
 
@@ -32,7 +29,6 @@ for filename in os.listdir(folder_path):
         if file_text.strip():
             texts.append(file_text)
 
-# 🌀 تبدیل کل کتاب به جملات
 normalizer = Normalizer()
 sentence_tokenizer = SentenceTokenizer()
 
@@ -42,7 +38,6 @@ for text in texts:
     sentences = sentence_tokenizer.tokenize(normalized)
     all_sentences.extend(sentences)
 
-# 📌 دریافت ورودی از کاربر
 query = st.text_input("🔎 کلمه یا عبارت موردنظر خود را وارد کنید:")
 
 # ✅ نمایش جمله و ۵ جمله بعدی + بازنویسی با LLM