import requests
import json
import gradio as gr
import os

def callSearchAPI(query: str) -> dict:
    """
        Call search API hosted on AWS
    """

    # hardcode public url of api
    url = os.getenv('GCR_API_URL')

    # make get request
    params = {"query": query}
    response = requests.get(url + "/search", params=params)

    # return response as python dict
    return json.loads(response.text)

def formatResultText(title: str, video_id: str):
    """
        Function to format video title and id as markdown
    """
    
    text = f"""<br> <br>
# {title}<br>

🔗 [Video Link](https://youtu.be/{video_id})"""

    return text

def formatVideoEmbed(video_id: str):
    """
        Function to generate video embed from video_id
    """
    
    return '<iframe width="576" height="324" src="https://www.youtube.com/embed/'+ video_id +'"></iframe>'

def searchResults(query):
    """
        Function to update search outputs for gradio ui
    """
    
    # API call
    response = callSearchAPI(query)

    # format search results

    # initialize list of outputs
    output_list = []

    # compute number of null search results (out of 5)
    num_empty_results = 5-len(response['title'])

    # display search results
    for i in range(len(response['title'])):
        video_id = response['video_id'][i]
        title = response['title'][i]

        embed = gr.HTML(value = formatVideoEmbed(video_id), visible=True)
        text = gr.Markdown(value = formatResultText(title, video_id), visible=True)

        output_list.append(embed)
        output_list.append(text)

    # make null search result slots invisible
    for i in range(num_empty_results):
        
        # if no search results display "No results." text
        if num_empty_results==5 and i==0:
            embed = gr.HTML(visible=False)
            text = gr.Markdown(value = "No results. Try rephrasing your query.", visible=True)

            output_list.append(embed)
            output_list.append(text)
            continue

        embed = gr.HTML(visible=False)
        text = gr.Markdown(visible=False)

        output_list.append(embed)
        output_list.append(text)
        
    return output_list