Update app.py

app.py

@@ -8,6 +8,21 @@ 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')
@@ -21,26 +36,9 @@ def web_scrape(url):
         wd.implicitly_wait(10)
         page_title = wd.title.replace("Stock Photo | Adobe Stock", "").strip()
         #content_value = meta_element.get_attribute("name")
-        prompts = [
-            f"Please make 3 best microstock titles with {page_title}",
-        ]
-
-        titles = []
-        for prompt in prompts:
-            response = openai.Completion.create(
-                engine="text-davinci-002",
-                prompt=prompt,
-                max_tokens=30,  # Adjust max_tokens as needed
-                n=3,  # Generate 3 titles
-                stop=None,  # You can specify stop words to end the titles if needed
-            )
-
-            generated_titles = [choice['text'] for choice in response['choices']]
-            titles.extend(generated_titles)
-
-        return "\n".join(titles)
+        
 
-        #return meta_element
+        return page_title
     except WebDriverException as e:
         return "error handle website"
     finally:
@@ -50,7 +48,7 @@ def web_scrape(url):
     return Image.open(BytesIO(screenshot))
 
 iface = gr.Interface(
-    fn=web_scrape,
+    fn=generate_response,
     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 with Selenium (Body Tag)",