Update src/app.py

src/app.py

@@ -21,6 +21,11 @@ class Kokiku:
 
         self.servings = ["1 person", "2 people", "3 people", "4 people", "5 people", "6 people", "7 people", "8 people", "9 people", "10 people"]
         self.cooking_tools = ["Pan", "Oven", "Air Fryer", "Microwave", "Griller", "Steamer", "Rice Cooker", "Blender"]
+        self.countries = ["South Korea", "United States", "Indonesia", "United Kingdom", "Singapore", "Japan", "Australia"]
+        self.merchant = ["McDonalds", "Subway", "Starbucks", "KFC", "Pizza Hut", "Burger King", "Dunkin Donuts", "Domino's Pizza", "Pizza Hut", "KFC", "Krispy Kreme", "Papa's John", "Domino Pizza", "Taco Bell"]
+        self.activity_level = ["Sedentary", "Lightly Active", "Moderately Active", "Very Active", "Extra Active"]
+        self.age = 1
+
         self.ingredients = []
         self.include_shopping_list = True
         
@@ -106,7 +111,6 @@ class Kokiku:
             return "","**Please complete the information first!**"
         
     def prompt_meal_plan(self, ingredients, plan_duration, meals_per_day, cuisine, cooking_skill, cooking_time, intolerances, include_facts, shopping_list, servings):
-        prompt = "Make 1 week meal plan (2 meals/day) consist of Korean and western food using chicken, which can be cooked with beginner skill during 30 minutes of cooking time. Exclude seafood and show the nutrition facts for one portion of each meal. Make a shopping list for 2 people according to it. Answer it directly"
          # Check if the input is complete
         try:
             assert ingredients != None
@@ -160,9 +164,62 @@ class Kokiku:
         except AssertionError:
             return "","**Please complete the information first!**"
     
-    def prompt_order_recommendation(self):
-        return
+    def prompt_order_recommendation(self, eating_time, merchant, countries, activity_level, ingredients, age, intolerances):
+         # Check if the input is complete
+        try:
+            assert ingredients != None
+            assert merchant != None
+            assert countries != None
+            assert activity_level != None
+            assert intolerances != None
+            assert age != None
+            assert eating_time != None             
+        
+            # Preprocess the input
+            eating_time = eating_time.lower()
+            merchant = merchant.lower()
+            countries = countries.lower()
+            activity_level = activity_level.lower()
+            intolerances = intolerances.lower()
+            
+            self._order_recommendation_intolerances_prompt = f"Exclude {intolerances}. "
+
+            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._order_recommendation_prompt = f"Suggest me what to order for {eating_time} at {merchant} {countries}, which contain {ingredients} "
+            else:
+                return "","**Please input the ingredients first!**"
+            
+            if age.isnumeric():
+                age = int(age)
+                if self.is_integer(age) and age > 0 and age < 105:
+                    self._order_recommendation_prompt += f"and enough nutrient for age {age} with {activity_level} activity level. "
+                else:
+                    return "","**Please input the age first within the range (1 <= age <= 105 years)!**"
+            else:
+                return "","**Please input the age first with the correct value!**"
 
+            # Initialize the final prompt
+            self._final_prompt = self._order_recommendation_prompt
+
+            if intolerances != "none":
+                self._meal_planner_final_prompt += self._meal_planner_intolerances_prompt
+
+            self._final_prompt += " Give the condensed yet detailed answer"
+        
+            # 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}", 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?"),
@@ -178,16 +235,27 @@ class Kokiku:
                                                                                                                                                                                             gr.Checkbox(include_facts, label="Include Nutrition Facts?")], outputs=[gr.Radio(label="Kokiku's Meal Plan", placeholder="Complete the information to see our meal plans recommendations", default="Complete the information to see our meal plans recommendations", readonly=True, info="Our meal plans recommendation will be shown here!"), gr.Markdown()])
         return interface
     
-    def generate_order_recommendaton(self):
-        interface = gr.Interface(self.prompt_order_recommendation, css="footer{display:none !important}", show_label=True, theme=gr.themes.Soft(primary_hue="orange", secondary_hue="orange"), inputs=[gr.Textbox(input="text")], outputs=[gr.Markdown()])
+    def generate_order_recommendaton(self, eating_time, merchant, countries, activity_level, intolerances):
+        interface = gr.Interface(self.prompt_order_recommendation, css="footer{display:none !important}", show_label=True, theme=gr.themes.Soft(primary_hue="orange", secondary_hue="orange"), inputs=[gr.Radio(eating_time, value=eating_time[0], label="Eating Time", info="When do you want to eat?"), gr.Dropdown(merchant, value=merchant[0], label="Merchant", info="Where are you going to eat?"), gr.Dropdown(countries, value=countries[0], label="Country", info="Which country?"),
+                                                                                                                                                                                                       gr.Dropdown(activity_level, value=activity_level[0], label="Activty Level", info="How active are you?"), gr.Textbox(input="text", label="Ingredients", info="What ingredients to include in your food?", placeholder="Beef, Celery, Meatballs"), gr.Textbox(input="text", label="Age", info="How old are you?", placeholder="20 (numbers only)"), gr.Dropdown(intolerances, value=intolerances[0], label="Intolerances", info="Is there any intolerances?")],
+                                                                                                                                                                                                        outputs=[gr.Radio(label="Kokiku's order suggestion", placeholder="Complete the information to see our order suggestions", default="Complete the information to see our order suggestions", readonly=True, info="Our food order recommendations will be shown here!"), gr.Markdown()])
         return interface
+    
+    @staticmethod
+    def is_integer(n):
+        try:
+            float(n)
+        except ValueError:
+            return False
+        else:
+            return float(n).is_integer()
                                                                                                                                                                                
 
 def main():
     kokiku = Kokiku()
     app_recipe = 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_plan = kokiku.generate_meal_plan(kokiku.plan_duration, kokiku.meals_per_day, kokiku.cuisine, kokiku.cooking_skill, kokiku.cooking_time, kokiku.intolerances, kokiku.servings, kokiku.include_shopping_list, kokiku.include_facts)
-    app_order = kokiku.generate_order_recommendaton()
+    app_order = kokiku.generate_order_recommendaton(kokiku.eating_time, kokiku.merchant, kokiku.countries, kokiku.activity_level, kokiku.intolerances)
     app_entire = gr.TabbedInterface([app_recipe, app_plan, app_order], tab_names=["Recipe Maker", "Meal Planner", "What to Order"], title="Kokiku", theme=gr.themes.Soft(primary_hue="orange", secondary_hue="orange"), css="footer{display:none !important}")
     app_entire.launch(favicon_path="/assets/chef.ico", debug=False)