Upload 4 files

llm_api/connect.py

@@ -0,0 +1,15 @@
+from meal_planner_api import utils
+import google.generativeai as genai
+from meal_planner_api import config
+
+
+def generate_content(text: str):
+    model = connect()
+    response = model.generate_content(text)
+    return response.text
+
+
+def connect():
+    genai.configure(api_key=config.API_KEY)
+    model = genai.GenerativeModel('gemini-pro')
+    return model

meal_planner_api/config.py

@@ -0,0 +1,2 @@
+import os
+API_KEY = os.environ.get("GENAI_API_KEY")

meal_planner_api/create_meal_plan.py

@@ -0,0 +1,15 @@
+from llm_api import connect
+
+
+def generate(age, gender,current_weight, desired_weight, goal, restrictions, preferences):
+    vegetarian_text = "vegetarian" if vegetarian == "Yes" else "non-vegetarian"
+    prompt = (
+        f"Create a personalized one-week meal plan for someone whose goal is to {goal}, "
+        f"with an activity level of '{activity_level}', targeting {daily_calories} calories daily. "
+        f"The person prefers {vegetarian_text} meals and follows a '{meal_preference}' diet with {meal_count} meals per day. "
+        f"Each day should include breakfast, lunch, dinner, and snacks if applicable."
+    )
+
+    meal_plan = connect.generate_content(prompt)
+
+    return meal_plan

meal_planner_api/utils.py

@@ -0,0 +1,7 @@
+import textwrap
+from IPython.display import Markdown
+
+
+def to_markdown(text):
+    text = text.replace('•', '  *')
+    return Markdown(textwrap.indent(text, '> ', predicate=lambda _: True))