import gCalendar, chatAI, weather
import gradio as gr
import requests 
#import config
import os

API_KEY = os.getenv("HF_API_KEY")
API_URL = "https://api-inference.huggingface.co/models/microsoft/resnet-50"
headers = {"Authorization": f"Bearer {API_KEY}"}

clothing_list = [] # Utilized in chatAI.py for more personalized suggestions

def getOutfit(events, gender):
    # event_summaries = gCalendar.getCalendarEvents()
    # if isinstance(event_summaries, list):
    #     event_summaries = ", ".join(event_summaries)
    
    weather_details = weather.getWeather()

    generated_outfit = chatAI.outfitRecommendation(events, ", ".join(clothing_list), weather_details, gender)

    return (
        "Your Event(s): \n"
        f'{events} \n'

        "\n"

        "Suggested Outfit: \n"
        f'{generated_outfit}'
    )

def classifyClothing(filepath):
    def query(filename):
        with open(filename, "rb") as f:
            data = f.read()
        response = requests.post(API_URL, headers=headers, data=data)
        return response.json()

    output = query(filepath)

    if output[0]['label']:
        clothing_list.append(f"[{output[0]['label']}]")
        return [output[0]["label"], ", ".join(clothing_list)]
    else:
        return ["Nothing detected", ", ".join(clothing_list)]

with gr.Blocks() as demo:
    with gr.Tab("Get Outfit(s) Based on Today's Events"):
        with gr.Row():
            gr.Markdown(
                """
                <h1> <center> IntelliStyle </center> </h1>
                <h3> <center> 
                IntelliStyle takes in today's events, allowing it to suggest optimized outfits for your upcoming day. 
                </br>
                It also takes into account weather conditions, making sure you're looking your best — rain or shine. 
                </center> </h3>
                """
            )

        with gr.Row():
            events_textbox = gr.Textbox(lines = 2, label="Type in Today's Events")

        with gr.Row():
            gender_selector = gr.Radio(["Male", "Female", "Non-binary"], label="Gender Identity", info="If gender identity is not provided, results may be less fine-tuned.")

        with gr.Row():
            btn = gr.Button("Suggest an Outfit")

        with gr.Row():
            outputs=gr.Textbox(lines=5, label="Generated Outfit")

        btn.click(fn=getOutfit, inputs=[events_textbox, gender_selector], outputs=outputs)

        with gr.Row():
            with gr.Accordion(label="Additional Notes", open=False):
                gr.Markdown(
                    """
                    <h4 style="text-align:center"> <i><u> This is a "lite" version without integration to Google Calendar. </u></i> </h4>
                    """
                )

    with gr.Tab("Upload Clothing Items"):
        gr.Markdown(
            """
            <h1> <center> Upload Clothes to IntelliStyle </center> </h1>
            <h3> <center> 
            Adding images of your clothing allows Intellistyle to make even better recommendations — tailored just for you. 
            </br>
            Clothing items are best uploaded on their own, preferably on a clean backdrop while <b>not</b> being worn.
            </center> </h3>
            """
        )

        with gr.Row():
            inp = gr.Image(type="filepath")
            out = gr.Textbox(label="What article of clothing we think you uploaded")

        btn = gr.Button("Run")

        with gr.Row():
            display_clothing_list = gr.Textbox(label="Uploaded Clothes:")

        btn.click(fn=classifyClothing, inputs=inp, outputs=[out, display_clothing_list])

        with gr.Row():
            with gr.Accordion(label="Additional Notes", open=False):
                gr.Markdown(
                    """
                    <h4 style="text-align:center"> <i><u> Please do not spam the button as errors will arise. </br> Additionally, errors may occasionally occur even when all user input is correct. If you believe this to be the case, please refresh the page and try using the demo again. </u></i> </h4>
                    """
                )

        with gr.Row():
            gr.Markdown("""<h1><center>Demo Example ↓</center><h1>""")
        
        import os
        gr.Examples(
            examples=[
                [os.path.join(os.path.dirname(__file__), "example_tshirt.jpg")],
            ],            
            inputs = gr.Image(type="filepath", height=400),
            outputs = gr.Textbox(label="What article of clothing IntelliStyle thinks was uploaded"),
            fn=classifyClothing,
            cache_examples=True,
        )

    with gr.Tab("[Not Live] Get Outfit(s) Based on Your Google Calendar"):
        with gr.Row():
            gr.Markdown(
                """
                <h1> <center> IntelliStyle </center> </h1>
                <h3> <center> 
                IntelliStyle is integrated with your Google Calendar, allowing it to suggest optimized outfits for your upcoming day. 
                </br>
                It also takes into account weather conditions, making sure you're looking your best — rain or shine. 
                </center> </h3>
                """
            )

        with gr.Row():
            outputs=gr.Textbox(lines=5, label="Generated Outfit")
            
        with gr.Row():
            gender_selector = gr.Radio(["Male", "Female", "Non-binary"], label="Gender Identity", info="If gender identity is not provided, results may be less fine-tuned.")

        with gr.Row():
            btn = gr.Button("Connect to Google Calendar and Run", interactive=False)
            btn.click(fn=getOutfit, inputs=[btn, gender_selector], outputs=outputs)

        with gr.Row():
            with gr.Accordion(label="Additional Notes", open=False):
                gr.Markdown(
                    """
                    <h4 style="text-align:center"> <i><u> Unfortunately, I can't figure out how to integrate Google Calendar API into gradio, so this tab is currently non-functioning. </u></i> </h4>
                    """
                )
demo.launch()