first commit
This commit is contained in:
commit
0c4ceec221
65
groq_module.py
Normal file
65
groq_module.py
Normal file
|
|
@ -0,0 +1,65 @@
|
|||
from dotenv import load_dotenv
|
||||
import os
|
||||
from groq import Groq
|
||||
import json
|
||||
|
||||
# Load environment variables
|
||||
load_dotenv()
|
||||
|
||||
# Initialize Groq client
|
||||
client = Groq(
|
||||
api_key=os.getenv("GROQ_API_KEY"),
|
||||
)
|
||||
|
||||
|
||||
def chat_with_groq(prompt: str, model: str = "llama-3.3-70b-versatile") -> str:
|
||||
try:
|
||||
# Create a chat completion
|
||||
chat_completion = client.chat.completions.create(
|
||||
messages=[
|
||||
{
|
||||
"role": "system",
|
||||
"content": "you are a helpful AI engineer assistant."
|
||||
},
|
||||
{
|
||||
"role": "user",
|
||||
"content": prompt
|
||||
}
|
||||
],
|
||||
model=model,
|
||||
)
|
||||
# Return the response content
|
||||
return chat_completion.choices[0].message.content
|
||||
except Exception as e:
|
||||
return f"An error occurred: {e}"
|
||||
|
||||
def groq_module_fun(resume_prompt, base64_image):
|
||||
completion = client.chat.completions.create(
|
||||
model="llama-3.2-90b-vision-preview",
|
||||
messages=[
|
||||
{
|
||||
"role": "user",
|
||||
"content": [
|
||||
{
|
||||
"type": "text",
|
||||
"text": resume_prompt
|
||||
},
|
||||
{
|
||||
"type": "image_url",
|
||||
"image_url": {
|
||||
"url": f"data:image/jpeg;base64,{base64_image}",
|
||||
},
|
||||
}
|
||||
]
|
||||
}
|
||||
],
|
||||
temperature=1,
|
||||
max_completion_tokens=1024,
|
||||
top_p=1,
|
||||
stream=False,
|
||||
response_format={"type": "json_object"},
|
||||
)
|
||||
|
||||
# Access the response here inside the function
|
||||
api_response = completion.choices[0].message.content
|
||||
return api_response
|
||||
26
helpers.py
Normal file
26
helpers.py
Normal file
|
|
@ -0,0 +1,26 @@
|
|||
import base64
|
||||
import json
|
||||
import re
|
||||
|
||||
def jpeg_to_base64(file_path):
|
||||
with open(file_path, "rb") as image_file:
|
||||
encoded_string = base64.b64encode(image_file.read())
|
||||
return encoded_string.decode('utf-8')
|
||||
|
||||
def extract_json_to_dict(input_string):
|
||||
# Use regex to extract JSON content
|
||||
json_pattern = re.compile(r"\{.*\}", re.DOTALL) # DOTALL allows matching across multiple lines
|
||||
match = json_pattern.search(input_string)
|
||||
|
||||
if match:
|
||||
json_content = match.group(0) # Extract the matched JSON content
|
||||
try:
|
||||
# Convert the JSON string to a dictionary
|
||||
json_dict = json.loads(json_content)
|
||||
return json_dict
|
||||
except json.JSONDecodeError:
|
||||
print("Error: Extracted content is not valid JSON.")
|
||||
return None
|
||||
else:
|
||||
print("No JSON content found.")
|
||||
return None
|
||||
200
main.py
Normal file
200
main.py
Normal file
|
|
@ -0,0 +1,200 @@
|
|||
from groq_module import groq_module_fun, chat_with_groq
|
||||
from helpers import jpeg_to_base64, extract_json_to_dict
|
||||
from flask import Flask, request, jsonify
|
||||
import json
|
||||
import os
|
||||
|
||||
app = Flask(__name__)
|
||||
|
||||
@app.route('/recipe-maker', methods=['POST'])
|
||||
def recipe_maker():
|
||||
data = request.get_json()
|
||||
if not data:
|
||||
return jsonify({"error": "JSON body is required"}), 400
|
||||
|
||||
query = data.get('query')
|
||||
|
||||
|
||||
prompt = f''' based on the following query:
|
||||
query: {query}
|
||||
generate the recipe of the best 5 dishes in the following JSON format:
|
||||
|
||||
|
||||
{{
|
||||
"message": "<message response>"
|
||||
"Recipe":{{
|
||||
"dish1": {{
|
||||
"name": "<dish name>",
|
||||
"ingredients": "<dish ingredients>",
|
||||
"instructions": "<dish instructions>"
|
||||
}},
|
||||
"dish2": {{
|
||||
"name": "<dish name>",
|
||||
"ingredients": "<dish ingredients>",
|
||||
"instructions": "<dish instructions>"
|
||||
}},
|
||||
"dish3": {{
|
||||
"name": "<dish name>",
|
||||
"ingredients": "<dish ingredients>",
|
||||
"instructions": "<dish instructions>"
|
||||
}},
|
||||
"dish4": {{
|
||||
"name": "<dish name>",
|
||||
"ingredients": "<dish ingredients>",
|
||||
"instructions": "<dish instructions>"
|
||||
}},
|
||||
"dish5": {{
|
||||
"name": "<dish name>",
|
||||
"ingredients": "<dish ingredients>",
|
||||
"instructions": "<dish instructions>"
|
||||
}}
|
||||
}}
|
||||
|
||||
}}
|
||||
'''
|
||||
|
||||
response = chat_with_groq(prompt)
|
||||
print(response)
|
||||
recipe = extract_json_to_dict(response)
|
||||
|
||||
|
||||
|
||||
result = {
|
||||
"recipe": recipe,
|
||||
"query": query,
|
||||
}
|
||||
|
||||
return jsonify(result), 200
|
||||
|
||||
@app.route('/diet-plan', methods=['POST'])
|
||||
def meal_diet_plan():
|
||||
data = request.get_json()
|
||||
if not data:
|
||||
return jsonify({"error": "JSON body is required"}), 400
|
||||
|
||||
weight = data.get('weight')
|
||||
height = data.get('height')
|
||||
target_weight = data.get('target_weight')
|
||||
|
||||
if weight is None or height is None or target_weight is None:
|
||||
return jsonify({"error": "Weight, height, and target weight are required"}), 400
|
||||
|
||||
prompt = f''' based on the following information:
|
||||
weight: {weight}
|
||||
height: {height}
|
||||
target weight: {target_weight}
|
||||
|
||||
Please provide a weekly diet plan for the user to achieve their target weight in the following JSON format:
|
||||
{{
|
||||
"day1": {{
|
||||
"breakfast": "<food item>",
|
||||
"lunch": "<food item>",
|
||||
"snack": "<food item>",
|
||||
"dinner": "<food item>"
|
||||
}},
|
||||
"day2": {{
|
||||
"breakfast": "<food item>",
|
||||
"lunch": "<food item>",
|
||||
"snack": "<food item>",
|
||||
"dinner": "<food item>"
|
||||
}},
|
||||
"day3": {{
|
||||
"breakfast": "<food item>",
|
||||
"lunch": "<food item>",
|
||||
"snack": "<food item>",
|
||||
"dinner": "<food item>"
|
||||
}},
|
||||
"day4": {{
|
||||
"breakfast": "<food item>",
|
||||
"lunch": "<food item>",
|
||||
"snack": "<food item>",
|
||||
"dinner": "<food item>"
|
||||
}},
|
||||
"day5": {{
|
||||
"breakfast": "<food item>",
|
||||
"lunch": "<food item>",
|
||||
"snack": "<food item>",
|
||||
"dinner": "<food item>"
|
||||
}},
|
||||
"day6": {{
|
||||
"breakfast": "<food item>",
|
||||
"lunch": "<food item>",
|
||||
"snack": "<food item>",
|
||||
"dinner": "<food item>"
|
||||
}},
|
||||
"day7": {{
|
||||
"breakfast": "<food item>",
|
||||
"lunch": "<food item>",
|
||||
"snack": "<food item>",
|
||||
"dinner": "<food item>"
|
||||
}}
|
||||
}}
|
||||
'''
|
||||
|
||||
response = chat_with_groq(prompt)
|
||||
print(response)
|
||||
week_diet_plan = extract_json_to_dict(response)
|
||||
|
||||
|
||||
|
||||
result = {
|
||||
"meal_diet_plan": week_diet_plan,
|
||||
"weight": weight,
|
||||
"height": height,
|
||||
"target_weight": target_weight,
|
||||
}
|
||||
|
||||
return jsonify(result), 200
|
||||
|
||||
@app.route('/food-scan', methods=['POST'])
|
||||
def food_scan():
|
||||
if 'file' not in request.files:
|
||||
return jsonify({"error": "File is required"}), 400
|
||||
|
||||
file = request.files['file']
|
||||
if file.filename == '':
|
||||
return jsonify({"error": "Invalid file"}), 400
|
||||
|
||||
# Secure file saving
|
||||
temp_dir = "/tmp"
|
||||
os.makedirs(temp_dir, exist_ok=True) # Ensure temp directory exists
|
||||
jpeg_file_path = os.path.join(temp_dir, file.filename)
|
||||
|
||||
try:
|
||||
file.save(jpeg_file_path)
|
||||
except Exception as e:
|
||||
return jsonify({"error": f"Failed to save file: {str(e)}"}), 500
|
||||
|
||||
# Convert image to Base64
|
||||
base64_string = jpeg_to_base64(jpeg_file_path)
|
||||
if not base64_string:
|
||||
return jsonify({"error": "Failed to encode image"}), 500
|
||||
|
||||
# Construct prompt correctly
|
||||
prompt = """
|
||||
This is a food image. Classify the food name and category in the following JSON format:
|
||||
{
|
||||
"food_name": "<detected food name>",
|
||||
"category": "<food category>",
|
||||
"calories": <calories in food>,
|
||||
"fat": <fat in food>,
|
||||
"carbohydrates": <carbohydrates in food>,
|
||||
"protein": <protein in food>,
|
||||
"sugar": <sugar in food>,
|
||||
"fiber": <fiber in food>
|
||||
}
|
||||
"""
|
||||
|
||||
# Call the function (ensure this is implemented correctly)
|
||||
response = groq_module_fun(prompt, base64_string)
|
||||
food_data = json.loads(response)
|
||||
result = {
|
||||
"result": food_data
|
||||
}
|
||||
|
||||
return jsonify(result), 200
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
app.run(host='0.0.0.0', port=5000, debug=True)
|
||||
|
||||
Loading…
Reference in a new issue