Update app.py

app.py

@@ -1,4 +1,5 @@
 import streamlit as st
+import random
 from PIL import Image
 import requests
 import json
@@ -104,6 +105,9 @@ def main():
         probs = outputs.logits.softmax(dim=1)
         percentage = round(probs[0, label_num].item() * 100, 2)
 
+        st.markdown("### Your Image:")
+        st.image(img, width=260)
+
         st.write("The Predicted Classification is:")
 
         if label_num==0:
@@ -134,14 +138,16 @@ def main():
         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.markdown("### Your Image:")
         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")
+        max_calories = calories
+
+        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" 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}
+        params = {"f": category, "Calories": max_calories}
 
         if diet != "All":
             params["d"] = diet
@@ -149,14 +155,75 @@ def main():
         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.markdown("### No Recipe Found:")
             st.write("**No Recipe Found, please try another option from the dropdown menus.**")
         else:
-            st.write(response_json)
+            st.markdown("### Recommended Recipe:")
+            if len(response_json) > 1:
+                random_recipe = random.choice(response_json)
+                st.write("**Title:** ", random_recipe['Title'])
+                if random_recipe['Image Link'].endswith(".jpg") or random_recipe['Image Link'].endswith(".jpeg") or random_recipe['Image Link'].endswith(".png"):
+                    st.image(random_recipe['Image Link'], width=300)
+                else:
+                    st.write("**Image Link:** ", random_recipe['Image Link'])
+                st.write("**Rating:** ", random_recipe['Rating'])
+                st.write("**Description:** ", random_recipe['Description'])
+                st.write("**Ingredients:** ", random_recipe['Ingredients'])
+                st.write("**Recipe Facts:** ", random_recipe['Recipe Facts'])
+                st.write("**Directions:** ", random_recipe['Directions'])
+                st.write("**Nutrition Facts:** ", random_recipe['Nutrition Facts'])
+                st.write("**Number of Servings:** ", random_recipe['Number of Servings'])
+                st.write("**Calories:** ", random_recipe['Calories'])
+                st.write("**Total Fat:** ", random_recipe['Total Fat'])
+                st.write("**Saturated Fat:** ", random_recipe['Saturated Fat'])
+                st.write("**Cholesterol:** ", random_recipe['Cholesterol'])
+                st.write("**Sodium:** ", random_recipe['Sodium'])
+                st.write("**Total Carbohydrate:** ", random_recipe['Total Carbohydrate'])
+                st.write("**Dietary Fiber:** ", random_recipe['Dietary Fiber'])
+                st.write("**Total Sugars:** ", random_recipe['Total Sugars'])
+                st.write("**Protein:** ", random_recipe['Protein'])
+                st.write("**Vitamin C:** ", random_recipe['Vitamin C'])
+                st.write("**Calcium:** ", random_recipe['Calcium'])
+                st.write("**Iron:** ", random_recipe['Iron'])
+                st.write("**Potassium:** ", random_recipe['Potassium'])
+                st.write("**Tags:** ", random_recipe['Tags'])
+                st.write("**Recipe URL:** ", random_recipe['Recipe URLs'])
+                st.markdown("### JSON:")
+                st.write(response_json)                
+            else:
+                st.write("**Title:** ", response_json[0]['Title'])
+                if response_json[0]['Image Link'].endswith(".jpg") or response_json[0]['Image Link'].endswith(".jpeg") or response_json[0]['Image Link'].endswith(".png"):
+                    st.image(response_json[0]['Image Link'], width=300)
+                else:
+                    st.write("**Image Link:** ", response_json[0]['Image Link'])
+                st.write("**Rating:** ", response_json[0]['Rating'])
+                st.write("**Description:** ", response_json[0]['Description'])
+                st.write("**Ingredients:** ", response_json[0]['Ingredients'])
+                st.write("**Recipe Facts:** ", response_json[0]['Recipe Facts'])
+                st.write("**Directions:** ", response_json[0]['Directions'])
+                st.write("**Nutrition Facts:** ", response_json[0]['Nutrition Facts'])
+                st.write("**Number of Servings:** ", response_json[0]['Number of Servings'])
+                st.write("**Calories:** ", response_json[0]['Calories'])
+                st.write("**Total Fat:** ", response_json[0]['Total Fat'])
+                st.write("**Saturated Fat:** ", response_json[0]['Saturated Fat'])
+                st.write("**Cholesterol:** ", response_json[0]['Cholesterol'])
+                st.write("**Sodium:** ", response_json[0]['Sodium'])
+                st.write("**Total Carbohydrate:** ", response_json[0]['Total Carbohydrate'])
+                st.write("**Dietary Fiber:** ", response_json[0]['Dietary Fiber'])
+                st.write("**Total Sugars:** ", response_json[0]['Total Sugars'])
+                st.write("**Protein:** ", response_json[0]['Protein'])
+                st.write("**Vitamin C:** ", response_json[0]['Vitamin C'])
+                st.write("**Calcium:** ", response_json[0]['Calcium'])
+                st.write("**Iron:** ", response_json[0]['Iron'])
+                st.write("**Potassium:** ", response_json[0]['Potassium'])
+                st.write("**Tags:** ", response_json[0]['Tags'])
+                st.write("**Recipe URL:** ", response_json[0]['Recipe URLs'])
+                st.markdown("### JSON:")
+                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)