import os
import gradio as gr
from utils import get_text_from_audio, get_video_from_yt
from logger import logging
import config


<<<<<<< HEAD
def get_text(audio_path):
    if not audio_path:
        return "No audio file chosen..."
    model = whisper.load_model(name='base', download_root=config.MODEL_DIR)
    results = model.transcribe(audio_path)
    return results['text']
=======
def get_results(video_url :str) -> str:

    logging.info(f">>>Getting predictions for : {video_url}")

    try :
        video_path = get_video_from_yt(video_url=video_url, save_file_dir=config.AUDIO_FILES_DIR)
        
        if not video_path:
            return "Problem while downloading the video. Please check the logs."
        
        text = get_text_from_audio(video_path)

        if not text:
            return "Problem generating the text. Please check the logs."
        
        return text

    except Exception as e:
        logging.exception(e)
        return str(e)
>>>>>>> 0740c74 (input through youtube added)



iface = gr.Interface(fn=get_results, inputs="text", outputs="text")
iface.launch()