Reza-galaxy21 commited on
Commit
3be32e1
·
verified ·
1 Parent(s): 212643f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -5
app.py CHANGED
@@ -4,7 +4,7 @@ import faiss
4
  os.system("pip install pymupdf")
5
  import fitz # PyMuPDF
6
  from langchain_community.vectorstores import FAISS
7
- from langchain.embeddings.openai import OpenAIEmbeddings
8
  from sentence_transformers import SentenceTransformer
9
 
10
  # مدل برای embeddings
@@ -30,8 +30,7 @@ def load_database():
30
 
31
  print("⏳ در حال مقداردهی FAISS...")
32
  try:
33
- index = faiss.IndexFlatL2(len(embeddings[0]))
34
- faiss_db = FAISS(index)
35
  print("✅ FAISS مقداردهی شد.")
36
  return faiss_db
37
  except Exception as e:
@@ -48,7 +47,7 @@ else:
48
  # تابع پردازش فایل
49
  def process_pdf(file):
50
  try:
51
- doc = PyMuPDF.open(file.name)
52
  text = "\n".join([page.get_text() for page in doc])
53
  print(f"✅ فایل پردازش شد. {len(text)} کاراکتر استخراج شد.")
54
  return text
@@ -67,4 +66,4 @@ with gr.Blocks() as demo:
67
  # اجرای برنامه
68
  if __name__ == "__main__":
69
  print("🚀 برنامه در حال اجرا است...")
70
- demo.launch()
 
4
  os.system("pip install pymupdf")
5
  import fitz # PyMuPDF
6
  from langchain_community.vectorstores import FAISS
7
+ from langchain_community.embeddings import OpenAIEmbeddings
8
  from sentence_transformers import SentenceTransformer
9
 
10
  # مدل برای embeddings
 
30
 
31
  print("⏳ در حال مقداردهی FAISS...")
32
  try:
33
+ faiss_db = FAISS.from_embeddings(embeddings, model)
 
34
  print("✅ FAISS مقداردهی شد.")
35
  return faiss_db
36
  except Exception as e:
 
47
  # تابع پردازش فایل
48
  def process_pdf(file):
49
  try:
50
+ doc = fitz.open(file.name)
51
  text = "\n".join([page.get_text() for page in doc])
52
  print(f"✅ فایل پردازش شد. {len(text)} کاراکتر استخراج شد.")
53
  return text
 
66
  # اجرای برنامه
67
  if __name__ == "__main__":
68
  print("🚀 برنامه در حال اجرا است...")
69
+ demo.launch()