analyze-data / app.py
flatindo's picture
Update app.py
3b64794
import gradio as gr
from selenium import webdriver
from selenium.common.exceptions import WebDriverException
from PIL import Image
from io import BytesIO
import openai
api_key = "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"
openai.api_key = api_key
def generate_response():
try:
prompts = [
f"Please make 3 best microstock titles with {web_scrape}",
]
response = openai.Completion.create(
engine="davinci", # You can choose different engines
prompt=prompts,
max_tokens=50, # Adjust max_tokens as needed
)
generated_text = response.choices[0].text.strip()
return generated_text
except Exception as e:
return f"Error: {str(e)}"
def web_scrape(url):
options = webdriver.ChromeOptions()
options.add_argument('--headless')
options.add_argument('--no-sandbox')
options.add_argument('--disable-dev-shm-usage')
try:
wd = webdriver.Chrome(options=options)
wd.set_window_size(1080, 720) # Adjust the window size here
wd.get(url)
wd.implicitly_wait(70)
page_title = wd.title.replace("Stock Photo | Adobe Stock", "").strip()
#content_value = meta_element.get_attribute("name")
return page_title
except WebDriverException as e:
return "error handle website"
finally:
if wd:
wd.quit()
return Image.open(BytesIO(screenshot))
iface = gr.Interface(
fn=web_scrape,
inputs=gr.inputs.Textbox(label="Website URL", default="https://stock.adobe.com/stock-photo/id/621214874"),
outputs=gr.outputs.Textbox(label="Web Content"),
title="Web Scraping by KecoakSalto",
description="Scrape the content of a website's titles.",
)
iface.launch()