Update app.py

app.py

@@ -87,7 +87,7 @@ def setup_database():
             cur.execute("CREATE EXTENSION IF NOT EXISTS vector;")
 
             # Удаляем существующие таблицы если они есть
-            cur.execute(f"DROP TABLE IF EXISTS {embeddings_table}, {query_cache_table};")
+            # cur.execute(f"DROP TABLE IF EXISTS {embeddings_table}, {query_cache_table};")
 
             # Создаем таблицу для хранения эмбеддингов фильмов
             cur.execute(f"""
@@ -299,6 +299,7 @@ def search_movies(query, top_k=10):
                         ON CONFLICT (query_crc32) DO NOTHING
                     """, (query_crc32, query, model_name, query_embedding.tolist()))
                 conn.commit()
+                logging.info(f"Сохранен новый эмбеддинг запроса: {query}")
             except Exception as e:
                 logging.error(f"Ошибка при сохранении эмбеддинга запроса: {e}")
                 conn.rollback()
@@ -307,13 +308,19 @@ def search_movies(query, top_k=10):
         try:
             with conn.cursor() as cur:
                 cur.execute(f"""
-                    SELECT m.movie_id, m.embedding <=> %s as distance
-                    FROM {embeddings_table} m
+                    WITH query_embedding AS (
+                        SELECT embedding
+                        FROM {query_cache_table}
+                        WHERE query_crc32 = %s
+                    )
+                    SELECT m.movie_id, m.embedding <=> (SELECT embedding FROM query_embedding) as distance
+                    FROM {embeddings_table} m, query_embedding
                     ORDER BY distance ASC
                     LIMIT %s
-                """, (query_embedding.tolist(), top_k))
+                """, (query_crc32, top_k))
 
                 results = cur.fetchall()
+            logging.info(f"Найдено {len(results)} результатов поиска.")
         except Exception as e:
             logging.error(f"Ошибка при выполнении поискового запроса: {e}")
             results = []