flatindo commited on
Commit
f60388c
·
1 Parent(s): 68d00ca

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +24 -2
app.py CHANGED
@@ -3,6 +3,10 @@ from selenium import webdriver
3
  from selenium.common.exceptions import WebDriverException
4
  from PIL import Image
5
  from io import BytesIO
 
 
 
 
6
 
7
  def web_scrape(url):
8
  options = webdriver.ChromeOptions()
@@ -15,10 +19,28 @@ def web_scrape(url):
15
  wd.set_window_size(1080, 720) # Adjust the window size here
16
  wd.get(url)
17
  wd.implicitly_wait(10)
18
- meta_element = wd.title.replace("Stock Photo | Adobe Stock", "").strip()
19
  #content_value = meta_element.get_attribute("name")
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
20
 
21
- return meta_element
22
  except WebDriverException as e:
23
  return "error handle website"
24
  finally:
 
3
  from selenium.common.exceptions import WebDriverException
4
  from PIL import Image
5
  from io import BytesIO
6
+ import openai
7
+
8
+ api_key = "YOUR_API_KEY"
9
+ openai.api_key = api_key
10
 
11
  def web_scrape(url):
12
  options = webdriver.ChromeOptions()
 
19
  wd.set_window_size(1080, 720) # Adjust the window size here
20
  wd.get(url)
21
  wd.implicitly_wait(10)
22
+ page_title = wd.title.replace("Stock Photo | Adobe Stock", "").strip()
23
  #content_value = meta_element.get_attribute("name")
24
+ prompts = [
25
+ f"Please make 3 best microstock titles with {page_title}",
26
+ ]
27
+
28
+ titles = []
29
+ for prompt in prompts:
30
+ response = openai.Completion.create(
31
+ engine="text-davinci-002",
32
+ prompt=prompt,
33
+ max_tokens=30, # Adjust max_tokens as needed
34
+ n=3, # Generate 3 titles
35
+ stop=None, # You can specify stop words to end the titles if needed
36
+ )
37
+
38
+ generated_titles = [choice['text'] for choice in response['choices']]
39
+ titles.extend(generated_titles)
40
+
41
+ return "\n".join(titles)
42
 
43
+ #return meta_element
44
  except WebDriverException as e:
45
  return "error handle website"
46
  finally: