import requests
from bs4 import BeautifulSoup
import openai
import streamlit as st

# Установка API ключа OpenAI
openai.api_key = "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"

# Заголовок приложения
st.title("Поиск и генерация текста")

# Поле для ввода запроса
query = st.text_input("Введите запрос для поиска:")

# Кнопка для запуска поиска и генерации текста
if st.button("Найти и сгенерировать"):
    # Запрос к Google
    url = f"https://www.google.com/search?q={query}"
    headers = {
        "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/58.0.3029.110 Safari/537.3"}
    response = requests.get(url, headers=headers)

    # Парсинг результатов
    soup = BeautifulSoup(response.content, "html.parser")
    results = soup.find_all("div", class_="g")

    # Сбор текста из результатов
    text = ""
    for result in results:
        text += result.find("a").text + " "

    # Генерация текста с помощью OpenAI API
    response = openai.Completion.create(
        engine="davinci",
        prompt=f"generate text based on the query '{query}' and the search results: {text}",
        max_tokens=1024,
        n=1,
        stop=None,
        temperature=0.5,
    )

    # Вывод сгенерированного текста
    st.write(response.choices[0].text)