Update app.py

app.py

@@ -1,20 +1,6 @@
 import streamlit as st
-from transformers import BertTokenizer, BertForTokenClassification, pipeline
 import random
 
-# Load the NER model and tokenizer
-model_name = "DavidEB2/bert-large-NER-finetuned-Tasteset-spoonacular"
-tokenizer = BertTokenizer.from_pretrained(model_name)
-model = BertForTokenClassification.from_pretrained(model_name)
-
-ner_pipeline = pipeline("ner", model=model, tokenizer=tokenizer)
-
-# Function to extract ingredients
-def extract_ingredients(text):
-    entities = ner_pipeline(text)
-    ingredients = [entity['word'] for entity in entities if entity['entity'] == 'B-ingredient']
-    return ingredients
-
 # Sample dataset for dish suggestions and recipes
 # Ideally, this would be a large dataset or database
 recipe_data = {
@@ -57,17 +43,18 @@ def suggest_dishes(ingredients):
 st.title("Cooking Suggestion App")
 st.write("Enter 1 or 2 ingredients to get recipe suggestions:")
 
-ingredients_input = st.text_input("Ingredients:")
+# Ingredients input
+ingredients_input = st.text_input("Enter ingredients (comma separated):")
 
 if st.button("Suggest Dishes"):
     if ingredients_input:
-        # Extract ingredients from input
-        extracted_ingredients = extract_ingredients(ingredients_input)
-        st.write("Extracted Ingredients:", extracted_ingredients)
+        # Split input into individual ingredients
+        ingredients = [ingredient.strip().lower() for ingredient in ingredients_input.split(',')]
+        st.write("Entered Ingredients:", ingredients)
         
-        if extracted_ingredients:
-            # Suggest dishes based on extracted ingredients
-            suggested_dishes = suggest_dishes(extracted_ingredients)
+        if ingredients:
+            # Suggest dishes based on entered ingredients
+            suggested_dishes = suggest_dishes(ingredients)
             if suggested_dishes:
                 st.write("Suggested Dishes:")
                 selected_dish = st.selectbox("Choose a dish", suggested_dishes)
@@ -76,8 +63,8 @@ if st.button("Suggest Dishes"):
                     st.write(f"Recipe for {selected_dish}:")
                     st.write(recipe_data[selected_dish]["recipe"])
             else:
-                st.write("No dishes found for the extracted ingredients.")
+                st.write("No dishes found for the entered ingredients.")
         else:
-            st.write("No ingredients extracted.")
+            st.write("No valid ingredients entered.")
     else:
         st.write("Please enter some ingredients.")