Update app2.py

app2.py

@@ -1557,6 +1557,23 @@ def automated_legal_brief_generation_ui():
             mime="text/plain"
         )
 
+PRACTICE_AREAS = [
+    "Personal Injury", "Medical Malpractice", "Criminal Law", "DUI & DWI", "Family Law",
+    "Divorce", "Bankruptcy", "Business Law", "Consumer Law", "Employment Law",
+    "Estate Planning", "Foreclosure Defense", "Immigration Law", "Intellectual Property",
+    "Nursing Home Abuse", "Probate", "Products Liability", "Real Estate Law", "Tax Law",
+    "Traffic Tickets", "Workers' Compensation", "Agricultural Law", "Animal & Dog Law",
+    "Antitrust Law", "Appeals & Appellate", "Arbitration & Mediation", "Asbestos & Mesothelioma",
+    "Cannabis & Marijuana Law", "Civil Rights", "Collections", "Communications & Internet Law",
+    "Construction Law", "Domestic Violence", "Education Law", "Elder Law",
+    "Energy, Oil & Gas Law", "Entertainment & Sports Law", "Environmental Law",
+    "Gov & Administrative Law", "Health Care Law", "Insurance Claims", "Insurance Defense",
+    "International Law", "Juvenile Law", "Landlord Tenant", "Legal Malpractice",
+    "Maritime Law", "Military Law", "Municipal Law", "Native American Law", "Patents",
+    "Securities Law", "Social Security Disability", "Stockbroker & Investment Fraud",
+    "Trademarks", "White Collar Crime"
+]
+
 STATES = [
     "Alabama", "Alaska", "Arizona", "Arkansas", "California", "Colorado", "Connecticut", "Delaware", "Florida", "Georgia",
     "Hawaii", "Idaho", "Illinois", "Indiana", "Iowa", "Kansas", "Kentucky", "Louisiana", "Maine", "Maryland",
@@ -1618,9 +1635,17 @@ CITIES_BY_STATE = {
     "Wyoming": ["Cheyenne", "Casper", "Laramie"]
 }
 
-def find_lawyers(state, city, pages=1):
+def find_lawyers(state, city=None, practice_area=None, pages=1):
     base_url = "https://www.justia.com/lawyers/"
-    url = f"{base_url}{state.lower()}/{city.lower().replace(' ', '-')}"
+    url = base_url
+    
+    if practice_area:
+        url += f"{practice_area.lower().replace(' ', '-')}/"
+    
+    url += state.lower()
+    
+    if city:
+        url += f"/{city.lower().replace(' ', '-')}"
     
     headers = {
         'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36'
@@ -1713,7 +1738,7 @@ def find_lawyers(state, city, pages=1):
 def lawyer_finder_ui():
     st.title("Find Lawyers in Your Area")
     
-    col1, col2 = st.columns(2)
+    col1, col2, col3 = st.columns(3)
     with col1:
         state = st.selectbox("Select a State:", STATES)
     
@@ -1721,6 +1746,9 @@ def lawyer_finder_ui():
         cities = CITIES_BY_STATE.get(state, [])
         city = st.selectbox("Select a City:", cities)
     
+    with col3:
+        practice_area = st.selectbox("Select a Practice Area:", [""] + PRACTICE_AREAS)
+    
     if not city:
         st.warning("Please select a city to continue.")
         return
@@ -1729,7 +1757,7 @@ def lawyer_finder_ui():
     
     if st.button("Find Lawyers", type="primary"):
         with st.spinner("Searching for lawyers in your area..."):
-            df_lawyers = find_lawyers(state, city, pages)
+            df_lawyers = find_lawyers(state, city, practice_area, pages)
             
             if not df_lawyers.empty:
                 st.success(f"Found {len(df_lawyers)} lawyers in {city}, {state}.")