import os
from google.generativeai import configure, generate_text
from duckduckgo_search import DDG

def generate_script(prompt, video_length, creativity, api_key):
    # Configure Google Gemini API
    configure(api_key=api_key)

    # Create search tool
    search_tool = DDG()

    # Fetch search data
    search_results = search_tool.text(prompt, max_results=5)
    search_data = "\n".join([res["snippet"] for res in search_results])

    # Generate title using Gemini-Flash
    title_prompt = f"Generate a creative and engaging title for a YouTube video about '{prompt}'."
    title_response = generate_text(title_prompt, temperature=creativity)
    title = title_response["text"]

    # Generate script using Gemini-Flash
    script_prompt = (
        f"Write a {video_length}-minute YouTube script with the title '{title}'. "
        f"Use the following data to enrich the content: {search_data}."
    )
    script_response = generate_text(script_prompt, temperature=creativity)
    script = script_response["text"]

    return title.strip(), script.strip(), search_data