Update app.py

app.py

@@ -214,26 +214,26 @@ def process_movies():
         ]
 
         print(f"Создаются эмбеддинги для фильмов: {', '.join(titles)}...")
-        
+
         with db_lock:
-          for movie, embedding_string in zip(batch, embedding_strings):
-              movie_id = movie['id']
-              embedding_crc32 = calculate_crc32(str(embedding.tolist()))
-              string_crc32 = calculate_crc32(embedding_string)
-              
-              # Проверяем, есть ли уже эмбеддинг для этого фильма в базе данных
-              existing_embedding = get_embedding_from_db(conn, embeddings_table, "embedding_crc32", embedding_crc32, model_name)
-              
-              if existing_embedding is None:
-                  # Создаем эмбеддинг, только если его нет в базе данных
-                  embedding = encode_string(embedding_string)
-                  
-                  if insert_embedding(conn, embeddings_table, "embedding_crc32", embedding_crc32, {"movie_id": movie_id, "string_crc32": string_crc32}, embedding):
-                      print(f"Эмбеддинг для фильма '{movie['name']}' сохранен в базе данных.")
-                  else:
-                      print(f"Ошибка сохранения эмбеддинга для фильма '{movie['name']}'.")
-              else:
-                  print(f"Эмбеддинг для фильма '{movie['name']}' уже существует в базе данных.")
+            for movie, embedding_string in zip(batch, embedding_strings):
+                movie_id = movie['id']
+                string_crc32 = calculate_crc32(embedding_string)
+
+                # Проверяем, есть ли уже эмбеддинг для этого фильма в базе данных
+                existing_embedding = get_embedding_from_db(conn, embeddings_table, "string_crc32", string_crc32, model_name)
+
+                if existing_embedding is None:
+                    # Создаем эмбеддинг, только если его нет в базе данных
+                    embedding = encode_string(embedding_string)
+                    embedding_crc32 = calculate_crc32(str(embedding.tolist()))
+
+                    if insert_embedding(conn, embeddings_table, "embedding_crc32", embedding_crc32, {"movie_id": movie_id, "string_crc32": string_crc32}, embedding):
+                        print(f"Эмбеддинг для фильма '{movie['name']}' сохранен в базе данных.")
+                    else:
+                        print(f"Ошибка сохранения эмбеддинга для фильма '{movie['name']}'.")
+                else:
+                    print(f"Эмбеддинг для фильма '{movie['name']}' уже существует в базе данных.")
 
     conn.close()
     print("Обработка фильмов завершена.")