import os
import requests
import json

import gradio as gr
import openai as ai


class Kokiku:
    def __init__(self):
        self.include_facts = True
        self.cooking_skill = ["Beginner", "Intermediate", "Advanced", "Expert"]
        self.cooking_time = ["< 30 minutes", "30 - 60 minutes", "> 60 minutes"]
        self.cuisine = ["Worldwide", "Korean", "Japanese", "Chinese", "Indonesian", "Indian", "Italian", "Mexican", "French", "American", "Thai", "Mediterranean", "Vietnamese", "Middle Eastern", "Spanish", "Greek", "German", "Nordic", "Caribbean", "British", "Irish", "Moroccan", "Russian", "Jewish", "Cajun", "Portuguese", "Hawaiian", "Hungarian", "Filipino", "Brazilian"]
        self.health_condition = ["None", "Weight Loss", "Diabetes", "High Cholesterol", "High Blood Pressure", "Kidney Disease", "Gout", "Anemia", "Celiac Disease", "GERD", "Crohn's Disease", "IBS", "Diverticulitis", "Pregnancy", "Lactation", "Gallstones", "Hepatitis", "HIV/AIDS", "Cancer", "PCOS"] 
        self.intolerances = ["None", "Dairy", "Egg", "Gluten", "Grain", "Peanut", "Seafood", "Sesame", "Shellfish", "Soy", "Sulfite", "Tree Nut", "Wheat", "Mustard", "Celery", "Lupin", "Mollusk", "Alcohol", "Red Meat", "White Meat", "FODMAP"]
        self.eating_time = ["Breakfast", "Lunch", "Dinner", "Brunch"]
        self.servings = ["1 person", "2 people", "3 people", "4 people", "5 people", "6 people", "7 people", "8 people", "9 people", "10 people"]
        self.ingredients = []
        self._main_prompt = ""
        self._health_condition_prompt = ""
        self._intolerances_prompt = ""
        self._include_facts_prompt = ""
        self._final_prompt = ""
        self._openai_key = os.getenv("OPENAI_KEY")
        # ai.api_key = self._openai_key

    def prompt_additional(self, ingredients, cuisine, cooking_skill, cooking_time, eating_time, servings, health_condition, intolerances, include_facts):
        # Check if the input is complete
        try:
            assert ingredients != None
            assert cuisine != None
            assert cooking_skill != None
            assert cooking_time != None
            assert eating_time != None
            assert servings != None
            assert health_condition != None
            assert intolerances != None                   
        
            # Preprocess the input
            cuisine = cuisine.lower()
            cooking_skill = cooking_skill.lower()
            cooking_time = cooking_time.lower()
            eating_time = eating_time.lower()
            servings = servings.lower()
            health_condition = health_condition.lower()
            intolerances = intolerances.lower()
            
            self._health_condition_prompt = f"Consider the {health_condition} health condition. "
            self._intolerances_prompt = f"Exclude {intolerances}. "
            self._include_facts_prompt = "Provide the nutritions fact for recipe's one portion. "

            if ingredients != "":
                punct = "," + ";"
                self.ingredients = "".join(ch for ch in ingredients if ch not in punct)
                assert self.ingredients != [], "Please input your ingredients first"
                self.ingredients = self.ingredients.lower()
                self._main_prompt = f"Give me 3 ideas on what to cook for {cuisine} dish {eating_time} using {self.ingredients} only, for {servings} with {cooking_skill} cooking skill during {cooking_time} of cooking time. "
            else:
                return "","**Please input your available ingredients first!**"

            # Initialize the final prompt
            self._final_prompt = self._main_prompt

            if health_condition != "none":
                self._final_prompt += self._health_condition_prompt
            if intolerances != "none":
                self._final_prompt += self._intolerances_prompt
            if include_facts:
                self._final_prompt += self._include_facts_prompt
        
            # response = ai.ChatCompletion.create(
            #             model="gpt-3.5-turbo-0613",
            #             temperature=0.9,
            #             messages = [{"role": "user", "content": "What's the weather like in Boston?"}]
            #         )
            # print(response.json())
            return "", self._final_prompt
        except AssertionError:
            return "","**Please complete the information first!**"

    def generate_recipe(self, cuisine, cooking_skill, cooking_time, eating_time, servings, health_condition, intolerances, include_facts):
        interface = gr.Interface(self.prompt_additional, css = "footer{display:none !important}", title="Kokiku", show_label=True, theme=gr.themes.Soft(primary_hue="orange", secondary_hue="orange"), inputs=[gr.Textbox(input="text", label="Ingredients", info="What's inside your fridge?", placeholder="Chicken, Goguma, Chili etc"), gr.Dropdown(cuisine, value=cuisine[0], label="Cuisine", info="What Cuisine that you crave now?"), gr.Radio(cooking_skill, value=cooking_skill[0], label="Cooking Skill", info="What is your cooking skill?"),
                                                                                          gr.Radio(cooking_time, value=cooking_time[0], label="Cooking Time", info="How much time do you have to cook?"), gr.Radio(eating_time, value=eating_time[0], label="Eating Time", info="When do you want to eat?"), gr.Dropdown(servings, value=servings[0], label="Servings", info="How many people will eat?"),
                                                                                          gr.Dropdown(health_condition, value=health_condition[0], label="Health Condition", info="Is there any special health condition?"), gr.Dropdown(intolerances, value=intolerances[0], label="Intolerances", info="Is there any intolerances?"),
                                                                                          gr.Checkbox(include_facts, label="Include Nutrition Facts?")], outputs=[gr.Radio(label="Kokiku's Recipe", placeholder="Complete the information to see our brilliant and tasty recipe recommendations", default="Complete the information to see our brilliant and tasty recipe recommendations", readonly=True, info="Our recipe recommendation will be shown here!"), gr.Markdown()])

        return interface
    

def main():
    kokiku = Kokiku()
    app = kokiku.generate_recipe(kokiku.cuisine, kokiku.cooking_skill, kokiku.cooking_time, kokiku.eating_time, kokiku.servings, kokiku.health_condition, kokiku.intolerances, kokiku.include_facts)
    app.launch(favicon_path="/assets/chef.ico", debug=False, share=True)


if __name__ == "__main__":
    main()