Spaces:
Running
Running
File size: 2,602 Bytes
2adba1d 8cfb148 2adba1d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 |
import io
import requests
import os
import re
import uuid
from PIL import Image, ImageEnhance
from fastapi import APIRouter, Depends
from fastapi.responses import StreamingResponse
from fastapi import UploadFile
from fastapi.responses import JSONResponse
from fastapi import HTTPException
from dotenv import load_dotenv
from pydantic import BaseModel
from pymongo import MongoClient
from models import *
from driver import Driver
from huggingface_hub import InferenceClient
from selenium.webdriver.common.by import By
from selenium.webdriver.support.expected_conditions import (
presence_of_element_located, visibility_of_element_located)
from selenium.webdriver.support.wait import WebDriverWait
class InstagramX(BaseModel):
link: str
router = APIRouter()
def obtain_ids(user: str):
response = requests.get("https://www.instagram.com/" + user)
appid = re.search(r'appId":"(\d*)', response.text)[1]
serverid = re.search(r'server_revision":(\d*)', response.text)[1]
return appid, serverid
@router.post("/akeno/instagram/reels", response_model=SuccessResponse, responses={422: {"model": SuccessResponse}})
async def instagram_reels(payload: InstagramX):
isInstagramLink = lambda link: bool(
re.match(r"^https?://(?:www\.)?instagram\.com/reel/", link)
)
if not isInstagramLink(payload.link):
return SuccessResponse(
status="False",
randydev={"error": "Give a valid Instagram reels link."}
)
try:
driver, error_message = Driver.get()
if not driver:
return SuccessResponse(
status="False",
randydev={"error": error_message}
)
driver.get(payload.link)
wait = WebDriverWait(driver, 10)
element = wait.until(presence_of_element_located((By.TAG_NAME, "video")))
reels_url = element.get_attribute("src")
driver.quit()
if reels_url:
binary_content = requests.get(reels_url).content
file_name = f"reels_{int(time.time())}.mp4"
with open(file_name, "wb") as file:
file.write(binary_content)
with open(file_name, "rb") as video:
encoded_string = base64.b64encode(video.read()).decode('utf-8')
os.remove(file_name)
return SuccessResponse(
status="True",
randydev={"video_data": encoded_string}
)
except Exception as e:
return SuccessResponse(
status="False",
randydev={"error": f"An error occurred: {str(e)}"}
)
|