import streamlit as st
from instagrapi import Client
import requests
from pathlib import Path
import os
from moviepy.editor import VideoFileClip
from PIL import Image

# Your Instagram credentials
username = "droidcv1"
password = ";ya_#GkM,KND6?$"

# Initialize the client
cl = Client()

# Add a delay before login
import time
time.sleep(5)

# Login to Instagram
try:
    cl.login(username, password)
except Exception as e:
    st.error(f'Login failed: {e}')
    st.stop()

# Function to download video from Instagram Reel URL
def download_instagram_reel(url, filename):
    shortcode = url.split('/')[-2]
    media_info = cl.media_info_from_shortcode(shortcode)
    video_url = media_info.video_url
    response = requests.get(video_url, stream=True)
    if response.status_code == 200:
        with open(filename, 'wb') as f:
            for chunk in response.iter_content(1024):
                f.write(chunk)
    return filename

# Function to generate thumbnail
def generate_thumbnail(video_path):
    clip = VideoFileClip(str(video_path))
    frame = clip.get_frame(1)  # Get a frame at 1 second
    thumbnail_path = str(video_path) + ".jpg"
    im = Image.fromarray(frame)
    im.save(thumbnail_path)
    return thumbnail_path

# Function to upload video and return the reel URL
def upload_video(video_path, caption):
    thumbnail_path = generate_thumbnail(video_path)
    media = cl.clip_upload(video_path, caption=caption, thumbnail=thumbnail_path)
    return f"https://www.instagram.com/reel/{media.pk}/"

# Directory to save downloaded videos
download_dir = Path("downloads")
download_dir.mkdir(exist_ok=True)

st.title('Instagram Reels Re-uploader')

urls = st.text_area('Enter Instagram Reel URLs (one per line):')

if st.button('Re-upload Reels'):
    if urls:
        reel_urls = urls.splitlines()
        for idx, reel_url in enumerate(reel_urls, start=1):
            with st.spinner(f'Downloading Reel {idx}...'):
                video_path = download_instagram_reel(reel_url, download_dir / f"reel_{idx}.mp4")
            caption = f"Re-uploaded Reel {idx}"
            with st.spinner(f'Uploading Reel {idx}...'):
                try:
                    reel_url = upload_video(video_path, caption)
                    st.success(f'Uploaded Reel {idx} URL: {reel_url}')
                except Exception as e:
                    st.error(f'Failed to upload Reel {idx}: {e}')
            os.remove(video_path)
            os.remove(video_path.with_suffix('.mp4.jpg'))
    else:
        st.warning('Please enter at least one Instagram Reel URL.')