Update summarizer.py

summarizer.py

@@ -9,7 +9,8 @@ import nltk
 import openai
 
 nltk.download('punkt')
-OPENAI_API_KEY = ""
+OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
+SCRAP_API_KEY = os.getenv("SCRAP_API_KEY")
 
 
 def create_brand_html(brand_link):
@@ -34,3 +35,23 @@ def create_langchain_openai_query(docs):
     map_reduce_chain = load_summarize_chain(llm, chain_type="map_reduce")
     output = map_reduce_chain.run(docs)
     return output
+
+
+def create_screenshot_from_scrap_fly(link_to_fetch):
+    import requests
+    import random
+    try:
+        params = {
+            'key': SCRAP_API_KEY,
+            'url': link_to_fetch,
+            'auto_scroll': True,
+            'capture': 'fullpage',
+            'options': 'block_banners'
+        }
+        response = requests.get('https://api.scrapfly.io/screenshot', params=params)
+        location = f"brand_ss_{random.randint(1, 100000000)}.png"
+        with open('screenshot.jpg', 'wb') as file:
+            file.write(response.content)
+        return {"location": location, "success": True}
+    except Exception as e:
+        return {"success": False, "error": e}