Update app.py

app.py

@@ -1,8 +1,7 @@
 import streamlit as st
 import requests
-import json
-from sentence_transformers import SentenceTransformer
 import chromadb
+from sentence_transformers import SentenceTransformer
 from bs4 import BeautifulSoup
 
 # Initialize embedding model (Open-Source)
@@ -12,7 +11,7 @@ model = SentenceTransformer("sentence-transformers/all-MiniLM-L6-v2")
 client = chromadb.PersistentClient(path="./recipe_db")
 collection = client.get_or_create_collection("recipes")
 
-# Function to scrape restaurant data (Alternative to API)
+# Function to scrape restaurant data
 def scrape_restaurant_info(city, recipe):
     search_url = f"https://www.foodpanda.pk/restaurants?search={recipe}+{city}"
     headers = {"User-Agent": "Mozilla/5.0"}
@@ -28,33 +27,51 @@ def scrape_restaurant_info(city, recipe):
     return restaurant_list
 
 # Streamlit UI
-st.title("Pakistani City Recipe Finder 🍛")
-
-city = st.selectbox("Select a City", ["Lahore", "Karachi", "Islamabad", "Peshawar", "Quetta"])
-query = st.text_input("Enter a Recipe Name")
-
-if st.button("Find Recipe"):
-    if query:
-        # Retrieve recipe info from vector DB
-        query_embedding = model.encode(query).tolist()
-        results = collection.query(query_embedding, n_results=3)
-
-        if results["documents"]:
-            st.subheader(f"Famous {query} 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")
-
-            # Fetch restaurant data via scraping
-            restaurants = scrape_restaurant_info(city, query)
-            if restaurants:
-                st.subheader("Available at These Restaurants:")
-                for r in restaurants:
-                    st.write(f"- {r}")
+st.title("Pakistani Famous Recipes Finder 🍛")
+
+# User inputs city
+city = st.text_input("Enter a Pakistani City (e.g., Lahore, Karachi, Islamabad)")
+
+# Optional: User inputs recipe (not mandatory)
+query = st.text_input("Enter a Recipe Name (Optional)")
+
+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)
+
+            if 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")
+
+                # Fetch restaurant data via scraping
+                restaurants = scrape_restaurant_info(city, query)
+                if restaurants:
+                    st.subheader("Available at These Restaurants:")
+                    for r in restaurants:
+                        st.write(f"- {r}")
+                else:
+                    st.write("No restaurant data found.")
             else:
-                st.write("No restaurant data found.")
+                st.write("No matching recipes found.")
+
         else:
-            st.write("No matching recipes found.")
+            # Retrieve all famous recipes in the city
+            city_embedding = model.encode(city).tolist()
+            results = collection.query(city_embedding, n_results=5)
+
+            if 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")
+            else:
+                st.write("No famous recipes found for this city.")
     else:
-        st.warning("Please enter a recipe name.")
+        st.warning("Please enter a city name.")