Update app.py
Browse files
app.py
CHANGED
@@ -181,36 +181,6 @@ st.markdown('<div class="chat-message">👋 سلام! چطور میتونم کم
|
|
181 |
# st.markdown(f'<div class="chat-message">📩 شما: {user_input}</div>', unsafe_allow_html=True)
|
182 |
|
183 |
|
184 |
-
""""
|
185 |
-
# ----------------- لود PDF و ساخت ایندکس -----------------
|
186 |
-
class TogetherEmbeddings(Embeddings):
|
187 |
-
def __init__(self, model_name: str, api_key: str):
|
188 |
-
self.model_name = model_name
|
189 |
-
self.client = Together(api_key=api_key)
|
190 |
-
|
191 |
-
def embed_documents(self, texts: List[str]) -> List[List[float]]:
|
192 |
-
response = self.client.embeddings.create(model=self.model_name, input=texts)
|
193 |
-
return [item.embedding for item in response.data]
|
194 |
-
|
195 |
-
def embed_query(self, text: str) -> List[float]:
|
196 |
-
return self.embed_documents([text])[0]
|
197 |
-
|
198 |
-
@st.cache_resource
|
199 |
-
def get_pdf_index():
|
200 |
-
with st.spinner('📄 در حال پردازش فایل PDF...'):
|
201 |
-
loader = [PyPDFLoader('test1.pdf')]
|
202 |
-
embeddings = TogetherEmbeddings(
|
203 |
-
model_name="togethercomputer/m2-bert-80M-8k-retrieval",
|
204 |
-
api_key="0291f33aee03412a47fa5d8e562e515182dcc5d9aac5a7fb5eefdd1759005979"
|
205 |
-
)
|
206 |
-
return VectorstoreIndexCreator(
|
207 |
-
embedding=embeddings,
|
208 |
-
text_splitter=RecursiveCharacterTextSplitter(chunk_size=300, chunk_overlap=0)
|
209 |
-
).from_loaders(loader)
|
210 |
-
|
211 |
-
index = get_pdf_index()
|
212 |
-
"""
|
213 |
-
|
214 |
|
215 |
# ----------------- لود csv و ساخت ایندکس -----------------
|
216 |
class TogetherEmbeddings(Embeddings):
|
|
|
181 |
# st.markdown(f'<div class="chat-message">📩 شما: {user_input}</div>', unsafe_allow_html=True)
|
182 |
|
183 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
184 |
|
185 |
# ----------------- لود csv و ساخت ایندکس -----------------
|
186 |
class TogetherEmbeddings(Embeddings):
|