Update app.py

app.py

@@ -1,4 +1,3 @@
-
 import time
 import streamlit as st
 from langchain.document_loaders import PyPDFLoader
@@ -10,10 +9,6 @@ from langchain.chains import RetrievalQA
 from langchain.chat_models import ChatOpenAI
 from typing import List
 from together import Together
-from langchain.embeddings import OpenAIEmbeddings, HuggingFaceInstructEmbeddings
-
-import streamlit as st
-from PIL import Image
 
 st.set_page_config(page_title="چت‌ بات توانا", page_icon="🪖", layout="wide")
 
@@ -21,19 +16,16 @@ st.set_page_config(page_title="چت‌ بات توانا", page_icon="🪖", lay
 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;
         text-align: right;
     }
-
     .stApp {
         background: url("military_bg.jpeg") no-repeat center center fixed;
         background-size: cover;
         backdrop-filter: blur(2px);
     }
-
     .stChatMessage {
         background-color: rgba(255,255,255,0.8);
         border: 1px solid #4e8a3e;
@@ -43,7 +35,6 @@ st.markdown("""
         box-shadow: 0 4px 10px rgba(0,0,0,0.2);
         animation: fadeIn 0.4s ease-in-out;
     }
-
     .stTextInput > div > input, .stTextArea textarea {
         background-color: rgba(255,255,255,0.9) !important;
         border-radius: 8px !important;
@@ -51,7 +42,6 @@ st.markdown("""
         text-align: right;
         font-family: 'Vazirmatn', Tahoma;
     }
-
     .stButton>button {
         background-color: #4e8a3e !important;
         color: white !important;
@@ -60,11 +50,9 @@ st.markdown("""
         padding: 8px 20px;
         transition: 0.3s;
     }
-
     .stButton>button:hover {
         background-color: #3c6d30 !important;
     }
-
     .header-text {
         text-align: center;
         margin-top: 20px;
@@ -74,20 +62,17 @@ st.markdown("""
         border-radius: 20px;
         box-shadow: 0 4px 12px rgba(0,0,0,0.2);
     }
-
     .header-text h1 {
         font-size: 42px;
         color: #2c3e50;
         margin: 0;
         font-weight: bold;
     }
-
     .subtitle {
         font-size: 18px;
         color: #34495e;
         margin-top: 8px;
     }
-
     @keyframes fadeIn {
         from { opacity: 0; transform: translateY(10px); }
         to { opacity: 1; transform: translateY(0); }
@@ -112,8 +97,6 @@ st.markdown("""
     </div>
 """, unsafe_allow_html=True)
 embeddings = OpenAIEmbeddings()
-
-
 class TogetherEmbeddings(Embeddings):
     def __init__(self, model_name: str, api_key: str):
         self.model_name = model_name
@@ -130,7 +113,10 @@ class TogetherEmbeddings(Embeddings):
 def get_pdf_index():
     with st.spinner('📄 در حال پردازش فایل PDF...'):
         loader = [PyPDFLoader('test1.pdf')]
-        embeddings = HuggingFaceInstructEmbeddings(model_name="SajjadAyoubi/xlm-roberta-large-fa-qa")
+        embeddings = TogetherEmbeddings(
+            model_name="togethercomputer/m2-bert-80M-8k-retrieval",
+            api_key="0291f33aee03412a47fa5d8e562e515182dcc5d9aac5a7fb5eefdd1759005979"
+        )
         return VectorstoreIndexCreator(
             embedding=embeddings,
             text_splitter=RecursiveCharacterTextSplitter(chunk_size=300, chunk_overlap=0)
@@ -173,7 +159,7 @@ if st.session_state.pending_prompt:
         thinking = st.empty()
         thinking.markdown("🤖 در حال فکر کردن...")
 
-        response = chain.run(f'question:پاسخ را فقط به زبان فارسی جواب بده {st.session_state.pending_prompt}')
+        response = chain.run(f'پاسخ را فقط به زبان فارسی جواب بده. سوال: {st.session_state.pending_prompt}')
         answer = response.split("Helpful Answer:")[-1].strip()
         if not answer:
             answer = "متأسفم، اطلاعات دقیقی در این مورد ندارم."