Update app.py

app.py

@@ -1,6 +1,7 @@
 import streamlit as st
 from PIL import Image
-
+import requests
+import json
 from transformers import pipeline
 import numpy as np
 from transformers import AutoFeatureExtractor
@@ -38,16 +39,6 @@ def main():
     st.markdown("This is a Food Category Image Classifier model that has been trained by [Kaludi](https://huggingface.co/Kaludi) to recognize **12** different categories of foods, which includes **Bread**, **Dairy**, **Dessert**, **Egg**, **Fried Food**, **Fruit**, **Meat**, **Noodles**, **Rice**, **Seafood**, **Soup**, and **Vegetable**. It can accurately classify an image of food into one of these categories by analyzing its visual features. This model can be used by food bloggers, restaurants, and recipe websites to quickly categorize and sort their food images, making it easier to manage their content and provide a better user experience.")  
     st.header("Try it out!")
 
-#    images = ["examples/example_0.jpg",
-#                    "examples/example_1.jpg",
-#                    "examples/example_2.jpg",
-#                    "examples/example_3.jpg",
-#                    "examples/example_4.jpg",
-#                    "examples/example_5.jpg",
-#                    "examples/example_6.jpg",
-#                    "examples/example_7.jpg"]
-#    show_images = False
-
     if st.checkbox("Show/Hide Examples"):
         st.header("Example Images")
 
@@ -76,13 +67,23 @@ def main():
 #            for image in images:
 #                st.image(image, width=260)
 
-    select_health = st.radio("Select One (Not Functional Yet):", ["Regular", "Low-Calorie"], horizontal=True)
-    
+#    select_health = st.radio("Select One (Not Functional Yet):", ["Regular", "Low-Calorie"], horizontal=True)
+
+    # Dropdown for Diet
+    diet_options = ['All', 'Gluten-Free', 'Vegan', 'Vegetarian', 'Dairy-Free']
+    diet = st.selectbox('Diet', diet_options)
+
+    # Dropdown for Cuisine
+    cuisine_options = ['All', 'African', 'Asian', 'Caribbean', 'Central American', 'Europe', 'Middle Eastern', 'North American', 'Oceanic', 'South American']
+
+    cuisine = st.selectbox('Cuisine', cuisine_options)
+
+    # Slider for Calories
     calories = st.slider("Select Max Calories (Per Serving)", 50, 2000)
     
     # print the calories
     st.write("Selected: **{}** Max Calories.".format(calories))
-    
+
     uploaded_file = st.file_uploader("Upload Files", type=['png','jpeg','jpg'])
 
     loading_text = st.empty()
@@ -98,6 +99,7 @@ def main():
         loading_text.empty()
         label_num=outputs.logits.softmax(1).argmax(1)
         label_num=label_num.item()
+        
 
         probs = outputs.logits.softmax(dim=1)
         percentage = round(probs[0, label_num].item() * 100, 2)
@@ -128,18 +130,36 @@ def main():
             st.write("**Soup** (" + f"{percentage}%)")
         else:
             st.write("**Vegetable** (" + f"{percentage}%)")
-
     
-        if select_health == "Regular":
-            st.write("You Selected **_Regular_** With Max", calories, "Calories For", "**Bread**" if label_num==0 else "**Dairy**" if label_num==1 else "**Dessert**" if label_num==2 else "**Egg**" if label_num==3 else "**Fried Food**" if label_num==4 else "**Fruit**" if label_num==5 else "**Meat**" if label_num==6 else "**Noodles**" if label_num==7 else "**Rice**" if label_num==8 else "**Seafood**" if label_num==9 else "**Soup**" if label_num==10 else "**Vegetable**")
-            # Add code to fetch healthy recipe here (line #125-138)
-        elif select_health == "Low-Calorie":
-            st.write("You Selected **_Low-Calorie_** With Max", calories, "Calories For", "**Bread**" if label_num==0 else "**Dairy**" if label_num==1 else "**Dessert**" if label_num==2 else "**Egg**" if label_num==3 else "**Fried Food**" if label_num==4 else "**Fruit**" if label_num==5 else "**Meat**" if label_num==6 else "**Noodles**" if label_num==7 else "**Rice**" if label_num==8 else "**Seafood**" if label_num==9 else "**Soup**" if label_num==10 else "**Vegetable**")
-            # Add code to fetch unhealthy recipe here (line #125-138)
+        st.write("You Selected **{}** For Diet and **{}** For Cuisine with Max".format(diet, cuisine), calories, "Calories For", ( "**Bread**" if label_num==0 else "**Dairy**" if label_num==1 else "**Dessert**" if label_num==2 else "**Egg**" if label_num==3 else "**Fried Food**" if label_num==4 else "**Fruit**" if label_num==5 else "**Meat**" if label_num==6 else "**Noodles**" if label_num==7 else "**Rice**" if label_num==8 else "**Seafood**" if label_num==9 else "**Soup**" if label_num==10 else "**Vegetable**"))
+
+        # Add code to fetch healthy recipe here (line #125-138)
 
         st.image(img, width=260)
+
+        url = "https://alcksyjrmd.execute-api.us-east-2.amazonaws.com/default/nutrients_response"
+
+        category = ("Bread" if label_num==0 else "Dairy" if label_num==1 else "Dessert" if label_num==2 else "Egg" if label_num==3 else "Fried Food" if label_num==4 else "Fruit" if label_num==5 else "Meat" if label_num==6 else "Noodles" if label_num==7 else "Rice" if label_num==8 else "Seafood" if label_num==9 else "**Soup**" if label_num==10 else "Vegetable")
+
+        params = {"f": category}
+
+        if diet != "All":
+            params["d"] = diet
+
+        if cuisine != "All":
+            params["c"] = cuisine
+
+        st.markdown("### Recommended Recipe:") 
+        response = requests.get(url, params=params)
+        response_json = json.loads(response.content)
+
+        if len(response_json) == 0:
+            st.write("**No Recipe Found, please try another option from the dropdown menus.**")
+        else:
+            st.write(response_json)
+
         st.markdown("<hr style='text-align: center;'>", unsafe_allow_html=True)
-        st.markdown("<p style='text-align: center'><a href='https://github.com/Kaludii'>Github</a> | <a href='https://huggingface.co/Kaludi'>HuggingFace</a></p>", unsafe_allow_html=True)
+        st.markdown("<p style='text-align: center'><a href='https://github.com/Kaludii'>Github</a> | <a href='https://huggingface.co/Kaludi'>HuggingFace</a></p>", unsafe_allow_html=True)        
 
 if __name__ == '__main__':
     main()
@@ -158,4 +178,4 @@ if __name__ == '__main__':
 
 ## Add code to display the healthy and unhealthy recipes
 #st.write("Healthy Recipe: ", healthy_recipe['name'], "Calories: ", healthy_recipe['calories'])
-#st.write("Unhealthy Recipe: ", unhealthy_recipe['name'], "Calories: ", unhealthy_recipe['calories'])
+#st.write("Unhealthy Recipe: ", unhealthy_recipe['name'], "Calories: ", unhealthy_recipe['calories'])