|
import pytesseract |
|
import os |
|
from dotenv import load_dotenv, find_dotenv |
|
|
|
|
|
load_dotenv(find_dotenv()) |
|
import google.generativeai as palm |
|
api_key = os.environ["GOOGLE_API_KEY"] |
|
palm.configure(api_key=api_key) |
|
|
|
models = [m for m in palm.list_models() if 'generateText' in m.supported_generation_methods] |
|
model = models[0].name |
|
|
|
def llm(img): |
|
text = pytesseract.image_to_string(img, lang='eng') |
|
|
|
|
|
prompt = { "0" : "take this piece of information and give all the information in point wise better format also give some recommendation related to them, \ |
|
if you don't get any nutrition content simply reply 'I don't seem to have any knowledge of the particular Nutrition Content' " + text, |
|
|
|
"1" : "Take this Nutrition facts information and give all the contents and proportion in point wise Markdown format also give some recommendation related to them, \ |
|
Make sure the Recommendations are given in bulleted format under the heading Recommendations \ |
|
if you don't get any nutrition content simply reply 'I don't seem to have any knowledge of the particular Nutrition Content' " + text, |
|
|
|
"2" : "I've given you this piece of information it contains Nutrition facts, and I want you to give all the information in point-wise Markdown format also give some \ |
|
recommendation related to them like how the consumption of the content may affect your health and what kind of people will be benefited or harmed from a particular, \ |
|
content and its percentage if you don't get any nutrition content simply reply 'I don't seem to have any knowledge of the particular Nutrition Content' " + text |
|
} |
|
|
|
text = palm.generate_text( |
|
prompt=prompt["0"], |
|
model=model, |
|
temperature=0.3, |
|
max_output_tokens=2000, |
|
top_p=0.8, |
|
top_k=40, |
|
) |
|
return text.result |
