Update app.py

app.py

@@ -9,9 +9,11 @@ from langchain.chains import RetrievalQA
 from langchain.chat_models import ChatOpenAI
 from typing import List
 from together import Together
-from PIL import Image
 
 
+import streamlit as st
+from PIL import Image
+
 st.set_page_config(page_title="چت‌ بات توانا", page_icon="🪖", layout="wide")
 
 # استایل
@@ -98,7 +100,8 @@ st.markdown("""
         <div class="subtitle">دستیار هوشمند برای تصمیم‌گیری در میدان نبرد</div>
     </div>
 """, unsafe_allow_html=True)
-embeddings = OpenAIEmbeddings()
+
+
 class TogetherEmbeddings(Embeddings):
     def __init__(self, model_name: str, api_key: str):
         self.model_name = model_name
@@ -118,6 +121,10 @@ def get_pdf_index():
         embeddings = TogetherEmbeddings(
             model_name="togethercomputer/m2-bert-80M-8k-retrieval",
             api_key="0291f33aee03412a47fa5d8e562e515182dcc5d9aac5a7fb5eefdd1759005979"
+
+
+
+
         )
         return VectorstoreIndexCreator(
             embedding=embeddings,
@@ -161,7 +168,7 @@ if st.session_state.pending_prompt:
         thinking = st.empty()
         thinking.markdown("🤖 در حال فکر کردن...")
 
-        response = chain.run(f'پاسخ را فقط به زبان فارسی جواب بده. سوال: {st.session_state.pending_prompt}')
+        response = chain.run(f'question:پاسخ را فقط به زبان فارسی جواب بده {st.session_state.pending_prompt}')
         answer = response.split("Helpful Answer:")[-1].strip()
         if not answer:
             answer = "متأسفم، اطلاعات دقیقی در این مورد ندارم."
@@ -177,3 +184,4 @@ if st.session_state.pending_prompt:
         placeholder.markdown(full_response)
         st.session_state.messages.append({'role': 'ai', 'content': full_response})
         st.session_state.pending_prompt = None
+