import atproto
import requests
import os
from PIL import Image

def upload_voice_to_bluesky(handle, password, voice_clip_path, image_path="generic_image.png"):
    """
    Uploads a voice clip to Bluesky as an mp4 with a generic image.

    Args:
        handle: Your Bluesky handle (e.g., molly.bsky.social).
        password: Your Bluesky app password.
        voice_clip_path: Path to your voice clip (mp3, wav, etc.).
        image_path: Path to a generic image (png, jpg).
    """
    try:
        # Create a Bluesky client
        client = atproto.Client()
        client.login(handle, password)

        # Create a generic image if one doesn't exist
        if not os.path.exists(image_path):
            img = Image.new('RGB', (100, 100), color=(73, 109, 137))
            img.save(image_path)

        # Upload the image
        with open(image_path, "rb") as image_file:
            upload_image_result = client.upload_blob(image_file.read())

        # Convert the audio to an mp4 with the image
        mp4_path = "voice_with_image.mp4"
        os.system(f"ffmpeg -loop 1 -i {image_path} -i {voice_clip_path} -c:v libx264 -tune stillimage -c:a aac -b:a 192k -shortest {mp4_path}")

        # Upload the mp4
        with open(mp4_path, "rb") as mp4_file:
            upload_video_result = client.upload_blob(mp4_file.read())

        # Create the post with the video
        post_text = "Here's my voice clip!"  # Or whatever you want to say!
        post = client.send_post(
            text=post_text,
            embed=atproto.models.AppBskyEmbedExternal.Main(
                 external=atproto.models.AppBskyEmbedExternal.External(
                    uri=upload_video_result.blob,
                    title="Voice Clip",
                    description="Listen to my voice!",
                    thumb=upload_image_result.blob,
                 )
            )
        )

        print(f"Post created! {post.uri}")

        #Clean up the temporary files
        os.remove(mp4_path)
        if not os.path.exists("generic_image.png"):
            os.remove(image_path)

    except Exception as e:
        print(f"Error uploading voice clip: {e}")

# Example usage (replace with your actual credentials and file paths)
handle = "yourhandle.bsky.social"
password = "your_app_password" #make sure to create an app password!
voice_clip_path = "your_voice_clip.mp3"

upload_voice_to_bluesky(handle, password, voice_clip_path)