import gradio as gr
import openai
import requests
from gtts import gTTS
from moviepy.editor import *
from io import BytesIO
from tempfile import NamedTemporaryFile
from base64 import b64encode

input_gif_path = "https://i.imgur.com/ork8hoP.gif"
#Lucy https://i.imgur.com/RLMkj1P.gif
#Lisa "https://i.imgur.com/ork8hoP.gif"

def get_text_response(prompt,openAI_key):
  openai.api_key = openAI_key
  completions = openai.Completion.create(engine="text-davinci-003",prompt=prompt,max_tokens=600,n=1,stop=None,temperature=0.5,)
  message = completions.choices[0].text.strip()
  return message

def text_to_speech(text, output_file):
    tts = gTTS(text=text, lang="en")
    tts.save(output_file)

def chat_and_animate(output_file, user_prompt,openAI_key):
    # Get audio response from OpenAI API
    text_response = get_text_response(user_prompt,openAI_key)
    text_to_speech(text_response, "response.mp3")

    # Get audio duration in milliseconds
    audio = AudioSegment.from_file("response.mp3")
    audio_duration = len(audio)

    # Download the input GIF
    response = requests.get(input_gif_path)

    # Save the input GIF to a temporary file
    with NamedTemporaryFile(delete=False, suffix=".gif") as temp_gif:
        temp_gif.write(response.content)

        # Load the input GIF
        gif_clip = VideoFileClip(temp_gif.name)

    # Calculate the number of loops required to match the audio duration
    num_loops = audio_duration / (gif_clip.duration * 1000)

    # Duplicate the animated GIF to match the audio duration
    final_gif = gif_clip.loop(n=int(num_loops))

    # Set the audio to the animated GIF
    final_video = final_gif.set_audio(AudioFileClip("response.mp3"))

    # Save the final video
    final_video.write_videofile(output_file, codec="libx264", audio_codec="aac")

    # Clean up the temporary GIF file
    os.unlink(temp_gif.name)

# Set up the input and output components for the Gradio app
user_prompt = gr.inputs.Textbox(label="Ask me anything", default="Welcome")
output_video = gr.outputs.Video()

def chatbot(user_prompt,openAI_key):
    if user_prompt.strip() == '':
        return '[ERROR]: Blank input not allowed.'
    if openAI_key.strip() == '':
      return '[ERROR]: Please enter your Open AI Key. Get your key here : https://platform.openai.com/account/api-keys'
    if user_prompt.lower() == "welcome":
        output_file = "preloaded.mp4"
    else:
        output_file = "output.mp4"
        chat_and_animate(output_file, user_prompt,openAI_key)
    return output_file

'''
gr.Interface(
    fn=chatbot,
    inputs=gr.inputs.Textbox(label="Ask me anything", placeholder="Type your question and press enter"),
    outputs=gr.outputs.Video(),
    allow_flagging=False,
    flagging_options=[],
    title="Deep talks with Lisa !",
    description="Ask me anything, and I'll give you an animated response! Type 'Welcome' to load the chat assistant.",
    theme="compact"
).launch()
'''

with gr.Blocks() as demo:
    gr.Markdown(f'<center><h1>Ask Lisa !</h1></center>')
    gr.Markdown(f'<span><img src="https://i.imgur.com/ork8hoP.gif" width="100" height="100"></span>')
    with gr.Row():
        
        with gr.Group():
            gr.Markdown(f'<p style="text-align:center">Get your Open AI API key <a href="https://platform.openai.com/account/api-keys">here</a></p>')
            openAI_key=gr.Textbox(label='Enter your OpenAI API key here')
            question = gr.Textbox(label='Ask me anything, and I will give you an animated response! Enteryour question here')
            btn = gr.Button(value='Talk')
            
            btn.style(full_width=True)
        with gr.Group():
            output_video = gr.outputs.Video()
        btn.click(chatbot, inputs=[question,openAI_key], outputs=[output_video])
#openai.api_key = os.getenv('Your_Key_Here') 

demo.launch()