Update app.py

app.py

@@ -4,6 +4,7 @@ import numpy as np
 from PIL import Image
 import pandas as pd
 import matplotlib.pyplot as plt
+import json
 
 # Load the saved model 
 model_path = "fahrnphi_exam_project.keras"
@@ -33,6 +34,10 @@ def predict_ingredient(image):
 st.title("Ingredient Classifier")
 st.write("A simple MLP classification model for image classification using a pretrained model.")
 
+# Initialize session state for storing ingredients
+if 'ingredients' not in st.session_state:
+    st.session_state['ingredients'] = []
+
 # Upload image
 uploaded_image = st.file_uploader("Choose an image...", type=["jpg", "png"])
 
@@ -46,12 +51,51 @@ if uploaded_image is not None:
 
     # Display predictions as a DataFrame
     st.write("### Prediction Probabilities")
-    df = pd.DataFrame(predictions.items(), columns=["ingredient", "Probability"])
+    df = pd.DataFrame(predictions.items(), columns=["Ingredient", "Probability"])
     st.dataframe(df)
     
     # Display predictions as a pie chart
     st.write("### Prediction Chart")
     fig, ax = plt.subplots()
-    ax.pie(df["Probability"], labels=df["ingredient"], autopct='%1.1f%%', colors=plt.cm.Paired.colors)
+    ax.pie(df["Probability"], labels=df["Ingredient"], autopct='%1.1f%%', colors=plt.cm.Paired.colors)
     ax.set_title('Prediction Probabilities')
-    st.pyplot(fig)
+    st.pyplot(fig)
+    
+    # Allow user to select the best guess
+    best_guess = st.selectbox("Select the best guess", df["Ingredient"])
+    if st.button("Add Ingredient"):
+        st.session_state.ingredients.append(best_guess)
+        st.write(f"Added {best_guess} to ingredients list")
+
+# Display the ingredients added so far
+st.write("### Selected Ingredients")
+st.write(st.session_state.ingredients)
+
+# Finish button to finalize ingredient selection and find recipes
+if st.button("Finish"):
+    st.write("Finding recipes...")
+    # Placeholder: Replace with actual recipe finding logic
+    def find_recipes(ingredients):
+        # This is a mock function, replace with actual recipe finding logic
+        sample_recipes = [
+            {"name": "Vegetable Stir Fry", "ingredients": ["Peperoni", "Carrot", "Onion"]},
+            {"name": "Tomato Garlic Pasta", "ingredients": ["Tomato", "Garlic"]},
+            {"name": "Ginger Potato Soup", "ingredients": ["Ginger", "Potato"]}
+        ]
+        matching_recipes = [recipe for recipe in sample_recipes if all(item in recipe["ingredients"] for item in ingredients)]
+        return matching_recipes
+    
+    matching_recipes = find_recipes(st.session_state.ingredients)
+    
+    if matching_recipes:
+        st.write("### Matching Recipes")
+        for recipe in matching_recipes:
+            st.write(f"**{recipe['name']}**")
+            st.write(", ".join(recipe["ingredients"]))
+    else:
+        st.write("No matching recipes found.")
+
+# Reset button to start over
+if st.button("Reset"):
+    st.session_state.ingredients = []
+    st.write("Ingredients list has been reset.")