Update app.py

app.py

@@ -19,7 +19,6 @@ st.set_page_config(page_title="رزم یار ارتش", page_icon="🪖", layout
 st.markdown("""
     <style>
     @import url('https://fonts.googleapis.com/css2?family=Vazirmatn:wght@400;700&display=swap');
-
     html, body, [class*="css"] {
         font-family: 'Vazirmatn', Tahoma, sans-serif;
         direction: rtl;
@@ -215,8 +214,8 @@ def get_csv_index(csv_file):
         
         # تقسیم متن‌های طولانی به بخش‌های کوچکتر
         text_splitter = RecursiveCharacterTextSplitter(
-            chunk_size=2048,
-            chunk_overlap=128,
+            chunk_size=300,
+            chunk_overlap=50,
             length_function=len,
             separators=["\n\n", "\n", " ", ""]
         )
@@ -227,7 +226,7 @@ def get_csv_index(csv_file):
 
         # ایجاد embeddings
         embeddings = TogetherEmbeddings(
-            model_name="togethercomputer/m2-bert-80M-32k-retrieval",
+            model_name="togethercomputer/m2-bert-80M-8k-retrieval",
             api_key="0291f33aee03412a47fa5d8e562e515182dcc5d9aac5a7fb5eefdd1759005979"
         )
 
@@ -249,7 +248,7 @@ csv_file_path = 'output (1).csv'
 try:
     # ساخت ایندکس
     csv_index = get_csv_index(csv_file_path)
-    # st.success("ایندکس فایل CSV با موفقیت ساخته شد!")
+    st.success("ایندکس فایل CSV با موفقیت ساخته شد!")
 except Exception as e:
     st.error(f"خطا در ساخت ایندکس: {str(e)}")
 
@@ -259,7 +258,7 @@ except Exception as e:
 llm = ChatOpenAI(
     base_url="https://api.together.xyz/v1",
     api_key='0291f33aee03412a47fa5d8e562e515182dcc5d9aac5a7fb5eefdd1759005979',
-    model="togethercomputer/m2-bert-80M-32k-retrieval"
+    model="togethercomputer/m2-bert-80M-8k-retrieval"
 )
 
 chain = RetrievalQA.from_chain_type(
@@ -291,7 +290,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'پاسخ را فقط به زبان فارسی جواب بده و به هیچ عنوان از زبان غیر از فارسی در پاسخ استفاده نکن: {st.session_state.pending_prompt}')
         answer = response.split("Helpful Answer:")[-1].strip() if "Helpful Answer:" in response else response.strip()
         if not answer:
             answer = "متأسفم، اطلاعات دقیقی در این مورد ندارم."