Update main.py

main.py

@@ -1,7 +1,32 @@
-from fastapi import FastAPI
+from fastapi import FastAPI, HTTPException
+import requests
+from bs4 import BeautifulSoup
+import pandas as pd
 
 app = FastAPI()
 
-@app.get("/")
-def read_root():
-    return {"Hello": "World"}
+def get_zillow_data(address):
+    headers = {
+        'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/58.0.3029.110 Safari/537.3'}
+    search_query = requests.utils.quote(address)
+    url = f'https://www.zillow.com/homes/{search_query}_rb/'
+    response = requests.get(url, headers=headers)
+    soup = BeautifulSoup(response.content, 'html.parser')
+
+    square_footage = 'N/A'
+
+    # Adjust the selectors based on Zillow's HTML structure
+    sqft_element = soup.find('span', {'data-testid': 'bed-bath-item'}, text=lambda x: 'sqft' in x)
+    if sqft_element:
+        square_footage = sqft_element.text.split('sqft')[0].strip()
+
+    return square_footage
+
+@app.get("/zillow")
+def read_zillow(address: str):
+    try:
+        sqft = get_zillow_data(address)
+        return {"address": address, "square_footage": sqft}
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))
+