Update app.py

app.py

@@ -30,24 +30,25 @@ def scrape_restaurant_info(city, recipe):
 st.title("Pakistani Famous Recipes Finder 🍛")
 
 # User inputs city
-city = st.text_input("Enter a Pakistani City (e.g., Lahore, Karachi, Islamabad)")
+city = st.text_input("Enter a Pakistani City (e.g., Lahore, Karachi, Islamabad)").strip()
 
 # Optional: User inputs recipe (not mandatory)
-query = st.text_input("Enter a Recipe Name (Optional)")
+query = st.text_input("Enter a Recipe Name (Optional)").strip()
 
 if st.button("Find Recipes & Restaurants"):
     if city:
         if query:
             # Retrieve specific recipe info from vector DB
             query_embedding = model.encode(query).tolist()
-            results = collection.query(query_embedding, n_results=3)
+            results = collection.query(query_embedding, n_results=5)
 
-            if results["documents"]:
+            if results and "documents" in results and results["documents"]:
                 st.subheader(f"Famous {query} in {city}")
-                for recipe in results["documents"][0]:
-                    st.write(f"**Recipe:** {recipe['name']}")
-                    st.image(recipe["image"], caption=recipe["name"], use_column_width=True)
-                    st.write(f"Price: {recipe['price']} PKR")
+                for doc in results["documents"]:
+                    for recipe in doc:
+                        st.write(f"**Recipe:** {recipe['name']}")
+                        st.image(recipe["image"], caption=recipe["name"], use_column_width=True)
+                        st.write(f"Price: {recipe['price']} PKR")
 
                 # Fetch restaurant data via scraping
                 restaurants = scrape_restaurant_info(city, query)
@@ -58,20 +59,21 @@ if st.button("Find Recipes & Restaurants"):
                 else:
                     st.write("No restaurant data found.")
             else:
-                st.write("No matching recipes found.")
+                st.write(f"No matching recipes found for '{query}' in {city}.")
 
         else:
             # Retrieve all famous recipes in the city
             city_embedding = model.encode(city).tolist()
             results = collection.query(city_embedding, n_results=5)
 
-            if results["documents"]:
+            if results and "documents" in results and results["documents"]:
                 st.subheader(f"Famous Recipes in {city}")
-                for recipe in results["documents"][0]:
-                    st.write(f"**Recipe:** {recipe['name']}")
-                    st.image(recipe["image"], caption=recipe["name"], use_column_width=True)
-                    st.write(f"Price: {recipe['price']} PKR")
+                for doc in results["documents"]:
+                    for recipe in doc:
+                        st.write(f"**Recipe:** {recipe['name']}")
+                        st.image(recipe["image"], caption=recipe["name"], use_column_width=True)
+                        st.write(f"Price: {recipe['price']} PKR")
             else:
-                st.write("No famous recipes found for this city.")
+                st.write(f"No famous recipes found for {city}.")
     else:
         st.warning("Please enter a city name.")