import numpy as np
from pydub import AudioSegment

def sine_wave(frequency, duration_ms):
    """Generate a sine wave tone at a given frequency and duration."""
    sample_rate = 44100  # Sample rate in Hz
    t = np.linspace(0, duration_ms / 1000, int(sample_rate * (duration_ms / 1000)), False)
    wave = 0.5 * np.sin(2 * np.pi * frequency * t)  # Sine wave formula
    audio_segment = AudioSegment(
        wave.tobytes(), 
        frame_rate=sample_rate,
        sample_width=2,  # 16-bit audio
        channels=1  # Mono audio
    )
    return audio_segment

def generate_kids_music(theme):
    # Create a simple melody based on themes
    notes = {
        "birthday": ["C4", "E4", "G4", "C5"],
        "ocean": ["C4", "D4", "E4", "F4"],
        "adventure": ["E4", "G4", "A4", "C5"]
    }
    
    selected_notes = notes.get(theme, ["C4", "E4", "G4"])  # Default notes if theme not found
    music = AudioSegment.silent(duration=0)  # Start with silence
    
    # Define frequencies for notes
    frequencies = {
        "C4": 261.63,
        "D4": 293.66,
        "E4": 329.63,
        "F4": 349.23,
        "G4": 392.00,
        "A4": 440.00,
        "C5": 523.25
    }

    for note in selected_notes:
        frequency = frequencies[note]
        sine = sine_wave(frequency, duration_ms=500)  # 0.5 second for each note
        music += sine
        
    # Export the generated music
    output_file = f"{theme}_kids_music.wav"
    music.export(output_file, format="wav")
    return output_file


from transformers import pipeline

# Load the GPT-2 model for text generation
comedy_generator = pipeline('text-generation', model='gpt2')

def generate_comedy_script(prompt):
    result = comedy_generator(prompt, max_length=150, num_return_sequences=1)
    return result[0]['generated_text']

from gtts import gTTS

def script_to_audio(script):
    tts = gTTS(script, lang='en')
    audio_file = "comedy_script.mp3"
    tts.save(audio_file)
    return audio_file

import gradio as gr

def generate_music_and_comedy(theme, prompt):
    # Generate music
    music_file = generate_kids_music(theme)
    
    # Generate comedy script
    comedy_script = generate_comedy_script(prompt)
    
    # Convert the comedy script to audio
    audio_file = script_to_audio(comedy_script)
    
    return music_file, comedy_script, audio_file

# Gradio Interface
iface = gr.Interface(
    fn=generate_music_and_comedy, 
    inputs=[
        gr.Dropdown(choices=["birthday", "ocean", "adventure"], label="Select Theme for Kids' Music"), 
        gr.Textbox(lines=2, placeholder="Enter a prompt for comedy...", label="Comedy Prompt")
    ],
    outputs=[
        gr.Audio(label="Generated Kids' Music"), 
        gr.Textbox(label="Generated Comedy Script"), 
        gr.Audio(label="Comedy Audio")
    ],
    title="AI Music and Comedy Generator",
    description="Generate unique music for kids based on themes and comedic scripts from your prompts."
)

# Launch the app
iface.launch()