Update app.py

app.py

@@ -3,7 +3,7 @@ import requests
 import chromadb
 from sentence_transformers import SentenceTransformer
 from bs4 import BeautifulSoup
-import os
+import json  # Import json to convert dictionary to string
 
 # Initialize embedding model
 model = SentenceTransformer("sentence-transformers/all-MiniLM-L6-v2")
@@ -26,9 +26,9 @@ if not collection.count():
     for recipe in sample_recipes:
         embedding = model.encode(recipe["city"]).tolist()
         collection.add(
-            ids=[recipe["name"]],
+            ids=[recipe["name"]],  # IDs should be unique
             embeddings=[embedding],
-            documents=[recipe]
+            documents=[json.dumps(recipe)]  # Convert dictionary to string
         )
     print("Sample data added to ChromaDB")
 
@@ -66,7 +66,8 @@ if st.button("Find Recipes & Restaurants"):
             if results and "documents" in results and results["documents"]:
                 st.subheader(f"Famous {query} in {city}")
                 for doc in results["documents"]:
-                    for recipe in doc:
+                    for recipe_json in doc:
+                        recipe = json.loads(recipe_json)  # Convert back to dictionary
                         st.write(f"**Recipe:** {recipe['name']}")
                         st.image(recipe["image"], caption=recipe["name"], use_column_width=True)
                         st.write(f"Price: {recipe['price']} PKR")
@@ -90,7 +91,8 @@ if st.button("Find Recipes & Restaurants"):
             if results and "documents" in results and results["documents"]:
                 st.subheader(f"Famous Recipes in {city}")
                 for doc in results["documents"]:
-                    for recipe in doc:
+                    for recipe_json in doc:
+                        recipe = json.loads(recipe_json)  # Convert back to dictionary
                         st.write(f"**Recipe:** {recipe['name']}")
                         st.image(recipe["image"], caption=recipe["name"], use_column_width=True)
                         st.write(f"Price: {recipe['price']} PKR")