Update main.py

main.py

@@ -5,6 +5,12 @@ from fastapi.responses import StreamingResponse
 from pydantic import BaseModel
 import re
 import replicate
+import OSError
+from supabase import create_client, Client
+
+url: str = os.environ.get("DB_URL")
+key: str = os.environ.get("DB_KEY")
+supabase: Client = create_client(url, key)
 
 
 class Item(BaseModel):
@@ -86,8 +92,6 @@ async def root(item: Item):
 
 
 
-
-
 @app.post("/extract-content")
 async def root(item: Item):
     
@@ -102,5 +106,41 @@ async def root(item: Item):
         
     except requests.RequestException as e:
         return {"error": str(e), "status_code": 500}
+
+
+async def insert_image(url , id):
+    data, count = await supabase.table('news')
+  .update({'image_url': url})
+  .eq('id', id)
+  .execute()
+    
+
+@app.get("/extract-images")
+async def root(site: str = 'abcnews.go'):
+    
+    try:
+
+        response = supabase.table('news').select("*")
+        .eq('source', 'www.'+ site + '.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[1]['src']
+
+                await insert_image(url, response.data[i]['id'] )
+        
+            except Exception as e:
+                results = None
+
+        
+    except requests.RequestException as e:
+        return {"error": str(e), "status_code": 500}