Анализ расширенных данных событий Facebook с использованием векторной базы данных

     import pandas as pd
from apify_client import ApifyClient
# Инициализируйте ApifyClient с вашим токеном API
client = ApifyClient("Enter_your_apify_key_here")

# Подготовьте входные данные для Актора
run_input = {
    "searchQueries": ["Sport New York"],
    "startUrls": [],
    "maxEvents": 50,
}

# Запустите Актора и дождитесь его завершения
run = client.actor("UZBnerCFBo5FgGouO").call(run_input=run_input)

df_columns = ['Name', 'Datetime', 'Description', 'Users_Going', 'Users_Interested', 'Users_Responded', 'City', 'Organized_By', 'Street_Address']
dataframe1 = pd.DataFrame(columns=df_columns)

for item in client.dataset(run["defaultDatasetId"]).iterate_items():
    # Используйте понимание словаря для замены значений None на пустую строку
    row = {
        'Name': item.get('name', ''),
        'Datetime': item.get('dateTimeSentence', ''),
        'Description': item.get('description', ''),
        'Users_Going': item.get('usersGoing', ''),
        'Users_Interested': item.get('usersInterested', ''),
        'Users_Responded': item.get('usersResponded', ''),
        'City': item.get('location', {}).get('city', '') if item.get('location') else '',
        'Organized_By': item.get('organizedBy', ''),
        'Street_Address': item.get('location', {}).get('streetAddress', '') if item.get('location') else ''
    }
    # Убедитесь, что все значения None заменены на пустую строку
    row = {k: (v if v is not None else '') for k, v in row.items()}
    dataframe1 = dataframe1._append(row, ignore_index=True)

# Очистка данных
dataframe1['Description'] = dataframe1['Description'].replace('\\n', '', regex=True)

     # Импортировать необходимые библиотеки для манипуляций с данными и парсинга дат
import pandas as pd
import numpy as np
from datetime import datetime
from dateutil import parser

# Функция для парсинга строк дат, которые могут представлять диапазон или одну дату
def parse_dates(date_str):
    # Проверить, содержит ли строка даты дефис, указывающий на диапазон
    if '-' in date_str:
        parts = date_str.split('-')
        # Если строка разделяется на две части, это действительный диапазон
        if len(parts) == 2:
            try:
                # Распарсить начальную и конечную даты, форматируя их в читаемый формат
                start_date = parser.parse(parts[0], fuzzy=True).strftime('%a, %b %d')
                end_date = parser.parse(parts[1], fuzzy=True).strftime('%a, %b %d')
                return start_date, end_date
            except ValueError:
                # В случае ошибки парсинга, ничего не делать (будет обработано ниже)
                pass  
    # Если это не диапазон или если парсинг диапазона не удался, попытаться распарсить как одну дату
    try:
        parsed_date = parser.parse(date_str, fuzzy=True)
        # Форматировать одну дату для start_date и по-другому для end_date
        start_date = parsed_date.strftime('%a, %b %d AT %I:%M %p EDT')
        end_date = parsed_date.strftime('%a, %b %d')  # Omitting time for end_date
        return start_date, end_date
    except ValueError:
        # Вернуть NaN для обеих дат, если парсинг не удался
        return np.nan, np.nan  

# Функция для извлечения подробной даты, времени и дня из строки даты
def extract_date_time_day(date_str):
    try:
        # Распарсить строку даты, разрешая некоторую гибкость в формате ввода
        parsed_date = parser.parse(date_str, fuzzy=True)
        # Извлечь и отформатировать части даты, времени и дня
        date = parsed_date.strftime('%Y-%m-%d')
        day = parsed_date.strftime('%a')
        # Определить, включала ли исходная строка компонент времени
        time_component = parsed_date.strftime('%I:%M %p') not in date_str
        time = parsed_date.strftime('%H:%M:%S') if not time_component else np.nan
    except ValueError:
        # Если парсинг не удался, установить дату, время и день в NaN
        date, time, day = np.nan, np.nan, np.nan
    
    return date, time, day

# Применить функцию parse_dates к датафрейму, создавая новые столбцы для начальной и конечной дат
dataframe1[['Start_Date', 'End_Date']] = dataframe1.apply(lambda row: pd.Series(parse_dates(row['Datetime'])), axis=1)

# Удалить строки, где Start_Date является NaN, указывая на неудачный парсинг
dataframe = dataframe1.dropna(subset=['Start_Date'])

# Применить extract_date_time_day для разделения начальной и конечной дат на отдельные столбцы даты, времени и дня
dataframe['Start_Date'], dataframe['Start_Time'], dataframe['Start_Day'] = zip(*dataframe['Start_Date'].apply(extract_date_time_day))
dataframe['End_Date'], _, dataframe['End_Day'] = zip(*dataframe['End_Date'].apply(extract_date_time_day))

# Удалить исходный столбец 'Datetime', так как он больше не нужен
dataframe=dataframe.drop(['Datetime'], axis=1)

# Преобразовать 'Start_Date' и 'End_Date' в формат даты, извлекая только часть даты
dataframe['Start_Date'] = pd.to_datetime(dataframe['Start_Date']).dt.date
dataframe['End_Date'] = pd.to_datetime(dataframe['End_Date']).dt.date

# Преобразовать 'Start_Time' в формат даты, сохраняя информацию о времени
dataframe['Start_Time'] = pd.to_datetime(dataframe['Start_Time'])

     # Импортируем библиотеку OpenAI для доступа к API.
from openai import OpenAI 
# Инициализируем клиент OpenAI с помощью ключа API.
openai_client = OpenAI(api_key="your_openai_api_key_here")
# Функция для получения текстовых эмбеддингов 
def get_embedding(text, model="text-embedding-3-small"):
    return openai_client.embeddings.create(input=text, model=model).data
embeddings = get_embedding(dataframe["Description"].tolist())
# Извлекаем векторы эмбеддингов из объекта эмбеддингов
vectors = [embedding.embedding for embedding in embeddings]
array = np.array(vectors)
embeddings_series = pd.Series(list(array))
# Добавляем эмбеддинги как новый столбец в DataFrame.
dataframe['Description_Embeddings'] = embeddings_series

     import clickhouse_connect
client = clickhouse_connect.get_client(
    host='host_name_here',
    port=443,
    username='username_here',
    password='passwd_here'
)

     client.command("""
    CREATE TABLE default.Events (
    Name String,
    Description String,
    Users_Going Int64,
    Users_Interested Int64,
    Users_Responded Int64,
    City String,
    Organized_By String,
    Street_Address String,
    Start_Date Date32,
    End_Date Nullable(Date32),
    Start_Time Nullable(DateTime64),
    Start_Day String,
    End_Day String,
    Description_Embeddings Array(Float32),
    CONSTRAINT check_data_length CHECK length(Description_Embeddings) = 1536
    ) ENGINE = MergeTree()
    ORDER BY (Name);
    """)

     batch_size = 10  # Adjust based on your needs

num_batches = len(dataframe) // batch_size

for i in range(num_batches):
    start_idx = i * batch_size
    end_idx = start_idx + batch_size
    batch_data = dataframe[start_idx:end_idx]
    # print(batch_data["Description_Embeddings"])
    client.insert("default.Events", batch_data.to_records(index=False).tolist(), column_names=batch_data.columns.tolist())
    print(f"Batch {i+1}/{num_batches} inserted.")

client.command("""
ALTER TABLE default.Events
    ADD VECTOR INDEX vector_index Description_Embeddings
    TYPE MSTG
""")

     results=client.query("""
        SELECT Name,Description FROM default.Events LIMIT 10
    """)
for row in results.named_results():
        print(row["Name"])
        print(row['Description'])

     results = client.query(f"""
        SELECT Name, City, Users_Going, Users_Interested, Users_Responded
        FROM default.Events
        ORDER BY Users_Going DESC, Users_Interested DESC
        LIMIT 10
    """)
for row in results.named_results():
        print("Event Name  ", row["Name"])
        print("City ", row["City"])
        print("Users Going ", row["Users_Going"])
        print("Interested Users ", row["Users_Interested"])

     embeddings=get_embedding(["One of the Longest Running Shows in the Country - Operating since 1974 ...NOW our 50th YEAR !!!Our Schenectady"])
embeddi=embeddings[0].embedding
results = client.query(f"""
        SELECT Name,City, Description, Users_Going,distance(Description_Embeddings, {embeddi}) as dist
        FROM default.Events
        WHERE City LIKE '%New York%' and dist < 1.5
        ORDER BY Users_Going DESC,dist
        LIMIT 10
    """)
for row in results.named_results():
        print("Event Name  ", row["Name"])
        print("Description ", row["Description"])
        print("Users Going ", row["Users_Going"])