import streamlit as st
import requests
import re
import openai
from pytube import YouTube
import os

# Remove previous audio file
audio_filename = "temp.mp3"
if os.path.exists(audio_filename):
    os.remove(audio_filename)

# Function to check if the provided OpenAI API key is valid
def is_valid_openai_key(api_key):
    try:
        openai.api_key = api_key
        openai.Engine.list()
        return True
    except Exception as e:
        st.write(f"Error: {e}")
        return False

# Function to check if a provided link is a valid YouTube video link
def is_valid_youtube_link(url):
    youtube_regex = re.compile(
        r'(https?://)?(www\.)?'
        r'(youtube|youtu|youtube-nocookie)\.(com|be)/'
        r'(watch\?v=|embed/|v/|.+\?v=)?([^&=%\?]{11})'
    )
    return youtube_regex.match(url) is not None

# Define the application layout
st.set_page_config(page_title="YouTube Video to Text", layout="wide")

# Add the application header
st.title("YouTube Video to Text")

# Add the side panel
with st.sidebar:
    st.header("Settings")
    openai_api_key = st.text_input("OpenAI API Key", type="password")
    youtube_video_link = st.text_input("YouTube Video Link")
    st.caption('Note: Only videos in english language are currently supported.')

# In the main section of the app, add the following code:
if openai_api_key and youtube_video_link and is_valid_youtube_link(youtube_video_link):
    if is_valid_openai_key(openai_api_key):
        # Your code to process the YouTube video using the OpenAI API goes here
        print(youtube_video_link)
        youtube_video = YouTube(youtube_video_link, use_oauth=True, allow_oauth_cache=True)
        st.markdown(f"Title: **{youtube_video.title}**")
        streams=youtube_video.streams.filter(only_audio=True)
        stream=streams.last()
        with st.spinner('Downloading Audio'):
            stream.download(filename='temp.mp3')
        audio_file= open("temp.mp3", "rb")
        with st.spinner('Transcribing Text'):
            transcript_json = openai.Audio.transcribe("whisper-1", audio_file)
            transcript = transcript_json["text"]
            st.markdown("**Text:**")
            st.markdown(transcript)
    else:
        st.warning("Invalid OpenAI API Key. Please enter a valid API key.")
else:
    st.warning("Please enter a valid YouTube video link and OpenAI API key.")

# Add the footer with the creator's name and GitHub link
st.markdown(
    """
    <div style="position: fixed; bottom: 10px; right: 10px; width: auto; height: auto; background-color: white; border: 1px solid #f0f2f6; border-radius: 5px; padding: 1rem;">
        Created by trhacknon - <a href="https://github.com/tucommenceapousser/youtube-video-to-text-generation" target="_blank">Source Code</a>
    </div>
    """,
    unsafe_allow_html=True
)