Update app.py

app.py

@@ -13,6 +13,8 @@ import pickle
 from langchain.chat_models import ChatOpenAI
 from langchain.prompts.chat import ChatPromptTemplate
 
+from uz_translit import to_latin
+
 # Загружаем переменные окружения
 load_dotenv()
 hf_key = os.getenv("HF_KEY")
@@ -38,7 +40,7 @@ def update_faiss_index():
     ]
     
     dataset = concatenate_datasets([train_dataset, test_dataset])
-    dataset = dataset.select(range(5))  # Для тестирования на небольшом количестве данных
+    # dataset = dataset.select(range(5))  # Для тестирования на небольшом количестве данных
     docs = []
     unique_chunks = set()
     for row in tqdm(dataset, desc="Загрузка документов..."):
@@ -107,7 +109,7 @@ def retrieve_articles(query, language):
     if language == "Russian":
         translated_query = translate_ru_uz(query)
     else:
-        translated_query = query
+        translated_query = to_lati(query)
 
     results = db.similarity_search(translated_query, k=3)