Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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
|
| 8 |
from sentence_transformers import SentenceTransformer
|
| 9 |
|
| 10 |
# مدل برای embeddings
|
|
@@ -30,8 +30,7 @@ def load_database():
|
|
| 30 |
|
| 31 |
print("⏳ در حال مقداردهی FAISS...")
|
| 32 |
try:
|
| 33 |
-
|
| 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 =
|
| 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()
|