SearchGPT / app.py
Serg4451D's picture
Create app.py
3909f76
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)