Update main.py

main.py

@@ -139,5 +139,34 @@ async def root(site: str = 'abcnews.go'):
         return {"RESULTS": ii_list}
     except requests.RequestException as e:
         return {"error": str(e), "status_code": 500}
+
+
+@app.get("/extract-images-livemint")
+async def root(site: str = 'livemint'):
+    
+    try:
+        ii_list = []
+        response = supabase.table('news').select("*").eq('source', f'www.livemint.com').is_('image_url', 'null').order('published_date', desc=True).limit(15).execute()
+
+        
+        for i in range(len(response.data)):
+        
+            url = response.data[i]['article_url']
+            try:
+                res = requests.get(url)
+                soup = BeautifulSoup(res.text, 'html.parser')
+                results = soup.find_all(['img'])
+                url = results[18]['src']
+                ii_list.append(url)
+
+                insert_image(url, response.data[i]['id'] )
+        
+            except Exception as e:
+                print(e)
+                results = None
+
+        return {"RESULTS": ii_list}
+    except requests.RequestException as e:
+        return {"error": str(e), "status_code": 500}