Update app.py

app.py

@@ -640,10 +640,13 @@ if query:
         """
     
         # ارسال پیام به مدل به صورت صحیح
-        response = llm(prompt)  # بدون پارامتر اضافی
+        response = llm([
+            SystemMessage(content="تو رزم یار ارتش هستی و  از کتاب و دیتای موجود به سوالات پاسخ میدی."),
+            HumanMessage(content=prompt)
+        ])
         
         # فرض بر این است که خروجی مدل به صورت دیکشنری است
-        rewritten = clean_text(response[0]['generated_text'].strip())  # اصلاح شده برای خروجی دیکشنری
+        rewritten = clean_text(response.content.strip())
     
         # نمایش نتیجه
         st.markdown(f'<div class="chat-message">{rewritten}</div>', unsafe_allow_html=True)