winwithakash's picture
commit
8add151
import streamlit as st
import numpy as np
import time
import PIL
import PIL.Image as Image
from utils import make_pred_outside_india
from utils import getmodel_outside_india
from utils import getmodel_india
from utils import load_prepare_img
from utils import food_nofood_pred
import sys
from RecipeData import fetchRecipeData
IMG_SIZE = (224, 224)
model_V2 = 'efficientnet_b0.pt'
model_V1 = 'indian_efficientnet_b0.pt'
@st.cache()
def model_prediction(model_path, img_file, rescale,selected_location):
input_img, device = load_prepare_img(img_file)
if(selected_location=='Outside India'):
model = getmodel_outside_india(model_path)
prediction = make_pred_outside_india(input_img, model, device, selected_location)
elif(selected_location=='India'):
model = getmodel_india(model_path)
prediction = make_pred_outside_india(input_img, model, device, selected_location)
print(prediction)
sorceCode, recipe_data = fetchRecipeData(prediction)
return prediction, sorceCode, recipe_data
def main():
st.set_page_config(
page_title="SeeFood",
page_icon="🍔 Know Your Receipe",
layout="wide",
initial_sidebar_state="expanded"
)
st.title('SeeFood🍔')
st.write('Upload a food image and get the recipe for that food and other details of that food')
col1, col2 = st.columns(2)
with col1:
# image uploading button
uploaded_file = st.file_uploader("Choose a file")
selected_location = st.selectbox('Select loaction',('India', 'Outside India'), index=1)
if uploaded_file is not None:
display_img = uploaded_file.read()
uploaded_img = Image.open(uploaded_file)
col2.image(display_img, width=500)
predict = st.button('Get Recipe!')
if predict:
if uploaded_file is not None:
with st.spinner('getting image type'):
img_type=food_nofood_pred(uploaded_img)
print(img_type)
if(img_type=='food'):
with st.spinner('Please Wait 👩‍🍳'):
# setting model and rescalling
if selected_location == 'India':
pred_model = model_V1
pred_rescale = True
if selected_location == 'Outside India':
pred_model = model_V2
pred_rescale =True
# makeing prediction and fetching food recipe form api
food, source_code, recipe_data = model_prediction(pred_model, uploaded_img, pred_rescale,selected_location)
# asssigning caleoric breakdown data
percent_Protein = recipe_data['percentProtein']
percent_fat = recipe_data['percentFat']
percent_carbs = recipe_data['percentCarbs']
# food name message
col1.success(f"It's an {food}")
if source_code == 200:
# desplay food recipe
st.header(recipe_data['title']+" Recipe")
col3, col4 = st.columns(2)
with col3:
# Ingridents of recipie
st.subheader('Ingredients')
# st.info(recipe_data['ingridents'])
for i in recipe_data['ingridents']:
st.info(f"{i}")
# Inctuction for recipe
with col4:
st.subheader('Instructions')
st.info(recipe_data['instructions'])
# st.subheader('Caloric Breakdown')
'''
## Caloric Breakdown
'''
st.success(f'''
* Protien: {percent_Protein}%
* Fat: {percent_fat}%
* Carbohydrates: {percent_carbs}%
''')
else:
st.error('Something went wrong please try again :(')
elif(img_type=='not food'):
# Ingridents of recipie
st.warning('This is not food image Please try again!!')
else:
st.warning('Please Upload Image')
if __name__=='__main__':
main()