Update main.py

main.py

@@ -2,7 +2,7 @@ import os
 import shutil
 from flask import Flask, render_template, request, jsonify
 from llama_index.core import StorageContext, load_index_from_storage, VectorStoreIndex, SimpleDirectoryReader, ChatPromptTemplate, Settings
-from llama_index.embeddings.huggingface import HuggingFaceEmbedding
+from sentence_transformers import SentenceTransformer  # Fallback for embeddings
 from deep_translator import GoogleTranslator
 import google.generativeai as genai
 
@@ -15,10 +15,16 @@ if not GOOGLE_API_KEY:
 genai.configure(api_key=GOOGLE_API_KEY)
 gemini_model = genai.GenerativeModel('gemini-flash-1.0')
 
-# Configure Llama index settings
-Settings.embed_model = HuggingFaceEmbedding(
-    model_name="sentence-transformers/paraphrase-multilingual-mpnet-base-v2"
-)
+# Custom Embedding Model (Fallback for HuggingFaceEmbedding)
+class CustomEmbedding:
+    def __init__(self, model_name):
+        self.model = SentenceTransformer(model_name)
+
+    def get_text_embedding(self, text):
+        return self.model.encode(text).tolist()
+
+# Configure Llama index settings with Custom Embedding
+Settings.embed_model = CustomEmbedding("sentence-transformers/paraphrase-multilingual-mpnet-base-v2")
 
 PERSIST_DIR = "db"
 PDF_DIRECTORY = 'data'