main.py

import settings
import typer
from transcript.audio import get_audio
from transcript.whisper import get_transcribe
from chains.summary_chain import summary_chain
from chains.youtube_chain import youtube_chain
from chains.thread_chain import thread_chain
from chains.tweet_chain import tweet_chain
from chains.linkedin_chain import linkedin_chain
from chains.blog_chain import blog_chain
from rich import print


app = typer.Typer()


@app.command()
def audio():
    video_path = f"./input/video.mp4"
    path = get_audio(video_path)
    print(path)


@app.command()
def transcribe(path: str):
    file_path = "./input/audio.wav"
    transcript = get_transcribe(file_path, "large")
    path = f"./{path}/transcript.md"
    with open(path, "w") as file:
        file.write(transcript)
        print(transcript)


@app.command()
def summary(path: str):
    file = open(f"./{path}/transcript.md", "r+")
    transcript = file.read()
    summary = summary_chain.invoke({
        "transcript": transcript,
        "language": "Spanish"
    })
    path = f"./{path}/summarize.md"
    with open(path, "w") as file:
        file.write(summary)
        print(summary)


@app.command()
def youtube(path: str):
    file = open(f"./{path}/summarize.md", "r+")
    summary = file.read()
    result = youtube_chain.invoke({"summary": summary, "language": "Spanish"})
    path = f"./{path}/youtube.md"
    with open(path, "w") as file:
        file.write(result)
        print(result)


@app.command()
def tweet(path: str, link: str, target: str):
    file = open(f"./{path}/summarize.md", "r+")
    text = file.read()
    result = tweet_chain.invoke(
        {
            "word_count": 140,
            "target_audience": target,
            "language": "LATAM Spanish",
            "link": link,
            "text": text,
        }
    )
    path = f"./{path}/tweet.md"
    with open(path, "w") as file:
        file.write(result)
        print(result)

@app.command()
def thread(path: str, link: str, target: str):
    file = open(f"./{path}/summarize.md", "r+")
    text = file.read()
    paragraphs = list(filter(lambda x: x != "", text.split("\n")))
    result = thread_chain.invoke(
        {
            "word_count": 140,
            "target_audience": target,
            "language": "Latam Spanish",
            "link": link,
            "number_of_tweets": len(paragraphs),
            "text": text,
        }
    )
    path = f"./{path}/thread.md"
    with open(path, "w") as file:
        file.write(result)
        print(result)


@app.command()
def linkedin(path: str, link: str, target: str, format: str, type: str):
    file = open(f"./{path}/summarize.md", "r+")
    summary = file.read()
    result = linkedin_chain.invoke(
        {
            "type_content": type,
            "counter_p": 3,
            "language": "Latam Spanish",
            "summary": summary,
            "format": format,
            "link": link,
            "target_audience": target,
        }
    )
    path = f"./{path}/linkedin.md"
    with open(path, "w") as file:
        file.write(result)
        print(result)


@app.command()
def blog(path: str):
    file = open(f"./{path}/summarize.md", "r+")
    summary = file.read()
    result = blog_chain.invoke(
        {
            "language": "Latam Spanish",
            "keyword": "JavaScript, TypeScript, propiedades privadas, métodos privados, privacidad en JavaScript, hashtag en JavaScript, clases en JavaScript, novedades JavaScript, programación, desarrollo web, seguridad en código, ProductService, evolución JavaScript, privacidad nativa, tutorial JavaScript, ejemplo práctico, programación web, JavaScript avanzado",
            "context": summary,
        }
    )
    path = f"./{path}/blog.md"
    with open(path, "w") as file:
        file.write(result)
        print(result)





if __name__ == "__main__":
    app()