Update app.py

app.py

@@ -45,6 +45,7 @@ if JINA_API_KEY is None:
     raise ValueError("JINA_API_KEY environment variable not set.")
 JINA_RERANKER_MODEL = "jina-reranker-v2-base-multilingual"
 
+
 # Имена таблиц
 embeddings_table = "movie_embeddings"
 query_cache_table = "query_cache"
@@ -297,7 +298,7 @@ def get_movie_embeddings(conn):
         logging.error(f"Ошибка при загрузке эмбеддингов фильмов: {e}")
     return movie_embeddings
 
-def rerank_with_api(query, results):
+def rerank_with_api(query, results, top_k):
     """Переранжирует результаты с помощью Jina AI Reranker API."""
     logging.info(f"Начало переранжирования для запроса: '{query}'")
     documents = []
@@ -305,46 +306,43 @@ def rerank_with_api(query, results):
         movie = next((m for m in movies_data if m['id'] == movie_id), None)
         if movie:
             movie_info = f"Название: {movie['name']}\nГод: {movie['year']}\nЖанры: {movie['genreslist']}\nОписание: {movie['description']}"
-            documents.append({"text": movie_info})
+            documents.append(movie_info)
 
+    headers = {
+        'Content-Type': 'application/json',
+        'Authorization': f'Bearer {JINA_API_KEY}'
+    }
     data = {
         "model": JINA_RERANKER_MODEL,
         "query": query,
-        "top_n": len(results),
+        "top_n": top_k,
         "documents": documents
     }
-    logging.info(f"Отправка данных на реранк (Jina AI API):\n{data}")
-
-    headers = {
-        'Content-Type': 'application/json',
-        'Authorization': f'Bearer {JINA_API_KEY}'
-    }
+    logging.info(f"Отправка данных на реранжировку: {data}")
 
     try:
         response = requests.post(JINA_API_URL, headers=headers, json=data)
         response.raise_for_status()  # Проверка на ошибки HTTP
-        response_json = response.json()
-        logging.info(f"Ответ от Jina AI API: {response_json}")
+        result = response.json()
+        logging.info(f"Ответ от API реранжировщика: {result}")
 
         reranked_results = []
-        for result in response_json.get("results", []):
-            index = result.get("index")
-            score = result.get("relevance_score")
-            if index is not None and score is not None:
-              reranked_results.append((results[index][0], score))
+        if 'results' in result:
+            for item in result['results']:
+              
+                index = item['index']
+                movie_id = results[index][0]
+                reranked_results.append((movie_id, item['relevance_score']))
+        else:
+            logging.warning("Ответ от API не содержит ключа 'results'.")
 
-        reranked_results.sort(key=lambda x: x[1], reverse=True)
         logging.info("Переранжирование завершено.")
+        time.sleep(0.5)
         return reranked_results
-    
+
     except requests.exceptions.RequestException as e:
-        logging.error(f"Ошибка при запросе к Jina AI Reranker API: {e}")
+        logging.error(f"Ошибка при запросе к API реранжировщика: {e}")
         return []
-    except (KeyError, ValueError) as e:
-        logging.error(f"Ошибка при обработке ответа от Jina AI Reranker API: {e}")
-        return []
-    finally:
-        time.sleep(0.5)
 
 def search_movies(query, top_k=20):
     """Выполняет поиск фильмов по запросу."""
@@ -397,11 +395,11 @@ def search_movies(query, top_k=20):
             logging.error(f"Ошибка при выполнении поискового запроса: {e}")
             results = []
 
-        # Переранжируем результаты
-        reranked_results = rerank_with_api(query, results)
+        # Переранжируем результаты с помощью API
+        reranked_results = rerank_with_api(query, results, top_k)
 
         output = ""
-        for movie_id, score in reranked_results[:top_k]:
+        for movie_id, score in reranked_results:
             # Находим фильм по ID
             movie = next((m for m in movies_data if m['id'] == movie_id), None)
             if movie: