Update src/app.py

src/app.py

@@ -23,11 +23,29 @@ class Kokiku:
         self.cooking_tools = ["Pan", "Oven", "Air Fryer", "Microwave", "Griller", "Steamer", "Rice Cooker", "Blender"]
         self.ingredients = []
         self.include_shopping_list = True
+        
+        # Main Prompt
         self._main_prompt = ""
         self._health_condition_prompt = ""
         self._intolerances_prompt = ""
         self._include_facts_prompt = ""
         self._final_prompt = ""
+
+        # Meal Planner Prompt
+        self._meal_planner_prompt = ""
+        self._meal_planner_intolerances_prompt = ""
+        self._meal_planner_include_facts_prompt = ""
+        self._meal_planner_shopping_list_prompt = ""
+        self._meal_planner_final_prompt = ""
+
+        # Order Recommendation Prompt
+        self._order_recommendation_prompt = ""
+        self._order_recommendation_health_condition_prompt = ""
+        self._order_recommendation_intolerances_prompt = ""
+        self._order_recommendation_include_facts_prompt = ""
+        self._order_recommendation_shopping_list_prompt = ""
+        self._order_recommendation_final_prompt = ""
+
         self._openai_key = os.getenv("OPENAI_KEY")
         # ai.api_key = self._openai_key
 
@@ -87,9 +105,60 @@ class Kokiku:
         except AssertionError:
             return "","**Please complete the information first!**"
         
-    def prompt_meal_plan(self, ingredients, plan_duration, meals_per_day, cuisine, cooking_skill, cooking_time, intolerances, nutrition_facts, shopping_list, servings):
+    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"
-        return "", prompt
+         # Check if the input is complete
+        try:
+            assert ingredients != None
+            assert cuisine != None
+            assert cooking_skill != None
+            assert cooking_time != None
+            assert servings != None
+            assert intolerances != None                   
+        
+            # Preprocess the input
+            plan_duration = plan_duration.lower()
+            meals_per_day = meals_per_day.lower()
+            cuisine = cuisine.lower()
+            cooking_skill = cooking_skill.lower()
+            cooking_time = cooking_time.lower()
+            intolerances = intolerances.lower()
+            
+            self._meal_planner_intolerances_prompt = f"Exclude {intolerances}. "
+            self._meal_planner_include_facts_prompt = "Provide the nutritions fact for recipe's one portion. "
+            self._meal_planner_shopping_list_prompt = f"Make a shopping list for {servings} according to it. "
+
+
+            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._meal_planner_prompt = f"Make {plan_duration} meal plan ({meals_per_day}/day) consist of {cuisine} food using {self.ingredients}, which can be cooked with {cooking_skill} skill during {cooking_time} of cooking time. "
+            else:
+                return "","**Please input the ingredients first!**"
+
+            # Initialize the final prompt
+            self._final_prompt = self._meal_planner_prompt
+
+            if intolerances != "none":
+                self._meal_planner_final_prompt += self._meal_planner_intolerances_prompt
+            if include_facts:
+                self._meal_planner_final_prompt += self._meal_planner_shopping_list_prompt
+            if shopping_list:
+                self._meal_planner_final_prompt += self._meal_planner_shopping_list_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 prompt_order_recommendation(self):
         return