import streamlit as st
from transformers import pipeline
from gtts import gTTS
import moviepy.editor as mp
from PIL import Image, ImageDraw, ImageFont
import os

# Streamlit setup
st.title("AI Video Learning Chatbot")

# User input for the learning topic
prompt = st.text_input("Enter a learning topic:")

if st.button("Generate Video"):
    try:
        # Step 1: Generate Text from Prompt using GPT-2
        st.write("Generating text response...")
        text_generator = pipeline('text-generation', model='gpt2')
        generated_text = text_generator(prompt, max_length=50)[0]['generated_text']
        st.write(f"Generated Text: {generated_text}")
        
        # Step 2: Convert Generated Text to Speech using gTTS
        st.write("Converting text to speech...")
        tts = gTTS(generated_text)
        tts.save("audio.mp3")
        
        # Step 3: Create Multiple Frames with Text Animation
        st.write("Generating animated frames for video...")
        num_frames = 30  # Number of frames in the video
        frame_duration = 0.1  # Duration per frame in seconds
        image_size = (640, 480)
        frames = []

        for i in range(num_frames):
            image = Image.new('RGB', image_size, color='black')
            draw = ImageDraw.Draw(image)
            font = ImageFont.load_default()
            
            # Animate text position
            text = generated_text
            text_position = (0, 20 + i * 10 % 400)  # Move text down over frames
            draw.text(text_position, text, fill="white", font=font)
            frame_path = f"frame_{i:03d}.png"
            image.save(frame_path)
            frames.append(mp.ImageClip(frame_path).set_duration(frame_duration))
        
        # Combine frames into a video
        video_clip = mp.concatenate_videoclips(frames, method="compose")
        audio_clip = mp.AudioFileClip("audio.mp3")
        video_clip = video_clip.set_audio(audio_clip)
        
        # Save the video with fps set to 24
        video_clip.write_videofile("output.mp4", codec="libx264", fps=24)
        
        # Display the video
        st.write("Video generated successfully!")
        st.video("output.mp4")
        
        # Clean up temporary files
        for frame_path in [f"frame_{i:03d}.png" for i in range(num_frames)]:
            os.remove(frame_path)
        os.remove("audio.mp3")
        os.remove("output.mp4")
        
    except Exception as e:
        st.error(f"An error occurred: {e}")