Update app.py

app.py

@@ -21,6 +21,7 @@ def extract_website_domain(url):
 def google_address(address):
         
     address_number = re.findall(r'\b\d+\b', address)[0]
+    address_zip =re.search(r'(\d{5})$', address).group()
     
     search_query = quote(address)
     url=f'https://www.google.com/search?q={search_query}'
@@ -55,16 +56,17 @@ def google_address(address):
     
     df['Square Footage']=df['Description'].str.extract(r"((\d+) Square Feet|(\d+) sq. ft.|(\d+) sqft|(\d+) Sq. Ft.|(\d+) sq|(\d+(?:,\d+)?) Sq\. Ft\.|(\d+(?:,\d+)?) sq)")[0]
     df['Square Footage']=df['Square Footage'].replace({',':''},regex=True).str.replace(r'\D', '')
-  
+    
     df['Beds']=df['Description'].replace({'-':' ','total':''},regex=True).str.extract(r"(\d+) bed")
     
+    
     df['Baths']=df['Description'].replace({'-':' ','total':''},regex=True).str.extract(r"((\d+) bath|(\d+(?:\.\d+)?) bath)")[0]
     df['Baths']=df['Baths'].str.extract(r'([\d.]+)').astype(float)
     
     df['Year Built']=df['Description'].str.extract(r"built in (\d{4})")
     
     df_final=df[df['Address'].notnull()]
-    df_final=df_final[df_final['Address'].str.contains(str(address_number))]
+    df_final=df_final[(df_final['Address'].str.contains(str(address_number))| df_final['Address'].str.contains(str(address_zip)))]
     df_final.insert(0,'Address Input',address)
     return df_final